Chem

import csv

def split_uppercase(string):
    x=''
    for i in string:
        if i.isupper(): x+=' %s' %i
        else: x+=i
    return x.strip()

#Read in data and store it in dictionary
elemen_data = { element.lower():weight for element,weight in csv.reader(open("chem.csv", "rb"))}
formul_data = { element.lower():weight for element,weight in csv.reader(open("form.csv", "rb"))}

while True:
    choice = int(raw_input("Element name or Formula? (1/2): "))

    if choice == 1:
        elements = {}
        numberOfPrompts = int(raw_input("Enter total amount of elements: "))
        for i in range(numberOfPrompts):
            # will prompt "Enter Element 1: " on the first iteration
            userInput = raw_input("Enter Element %s: " % (i+1, ))
            elements[userInput] = float(elemen_data.get(userInput.lower()))
        for element in elements.keys():
            weightSum = 0
            for weight in elements.values():
                weightSum += weight

        print "Total weight =" ,weightSum

    if choice == 2:
        print "Not ready, try a later version"
        formula = raw_input("Enter formula: ")


        upper = split_uppercase(formula)
        print upper
        #Pull in data from form.csv
        weight1 = float(formul_data.get(element1.lower()))
        weight2 = float(formul_data.get(element2.lower()))
        weight3 = float(formul_data.get(element3.lower()))


        weightSum = weight1 + weight2 + weight3
        print "Total weight =", weightSum

    if choice == 0:
        break