def user_input():
    todo = input("Ok, do you want to see them or to edit them [see/edit]?\n")
    if todo == "see":
        print("Which subject do you want to know the information?")
        subject = input()
        return "output", subject
    elif todo == "edit":
        print("Do you want to add a subject, a category or a grade [subject/category/grade]?")
        user_add = input()
        if user_add == "subject":
            subject = input("Which subject do you want to add? \n")
            main_subject = input("Is this subject a main subject [y/n]? \n")
            if main_subject == 'y':
                main_subject = True
            elif main_subject == 'n':
                main_subject = False
            oral_exam = input("Will there be an oral exam in this subject [y/n]? \n")
            if oral_exam == 'y':
                oral_exam = True
            elif oral_exam == 'n':
                oral_exam = False
            return "input", "subject", subject, main_subject, oral_exam
        elif user_add == "category":
            subject = input("To which subject do you want to add the category? \n")
            category = input("Which category do you want to add? \n")
            rating = float(input("Rating? \n"))
            return "input", "category", subject, category, rating
        elif user_add == "grade":
            subject = input("To which subject do you want to add the grade? \n")
            category = input("To which category do you want to add it? \n")
            grade = input("Grade? \n")
            grade_name = input("What's the name of the grade? \n")
            return "input", "grade", subject, category, grade, grade_name


def user_output(subject, categories, average, average_subject):
    print("\n\n In the subject {} are the following categories:".format(subject))
    for counter, category in enumerate(categories):
        print("The average in {} is {} with these grades:".format(category, average[counter]))
        string = ""
        for grade in categories[category]:
            string += str(grade) + ", "
        print(string + "\n")
        print("The average of {} is {}.".format(subject, average_subject))

def password():
    print("What is your password?")
    return input()