Update demo.py

demo.py

@@ -33,13 +33,13 @@ def get_level_mapping(level):
     if level is None:
         raise gr.Error("Please select a level.")
 
-    return {
-        "Primary School": 0,
-        "Middle School": 1,
-        "High School": 2,
-        "College": 3,
-        "Academy": 4,
-    }[level]
+    return levels.index(level)
+
+def get_category_mapping(category):
+    if category is None:
+        raise gr.Error("Please select a level.")
+
+    return categories.index(category)
 
 def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, number_of_distractors, temperature, language):
     """
@@ -85,6 +85,8 @@ def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, nu
         "language": language
     }
 
+    print(input_json)
+
     output = ""
     try:
         response = requests.post(
@@ -141,10 +143,12 @@ def generate_open_question(original_text, level, temperature, language, question
         "level": get_level_mapping(level),
         "temperature": int(temperature) * 0.2,
         "language": language,
-        "question_type": question_type,
-        "question_category": question_category
+        "type": question_type,
+        "category": get_category_mapping(question_category),
     }
 
+    print(input_json)
+
     output = ""
     try:
         response = requests.post(
@@ -206,7 +210,7 @@ def generate_multiplechoice(original_text, level, number_of_easy_distractors, nu
         "text": original_text,
         "type": bool(exercise_type),
         "level": get_level_mapping(level),
-        "question_category": question_category,
+        "category": get_category_mapping(question_category),
         "n_o_ca": int(correct_answers),
         "n_o_d": int(number_of_distractors),
         "nedd": int(number_of_easy_distractors),
@@ -365,7 +369,7 @@ def build_multiplechoice_interface():
                 language = gr.Dropdown(languages, label="Language", value="English")
                 with gr.Row():
                     exercise_type = gr.Radio(["Theoretical", "Practical"], label="Exercise type", value="Theoretical", type="index")
-                    question_category = gr.Dropdown(categories, label="Question Category", value="Factual Knowledge", type="index")
+                    question_category = gr.Dropdown(categories, label="Question Category", value="Factual Knowledge")
                 with gr.Row():
                     correct_answers = gr.Number(value=1, minimum=1, maximum=3, step=1, label="Number of correct answers")
                     easy_distractors = gr.Number(value=1, minimum=0, maximum=8, step=1, label="Number of easy distractors")
@@ -380,12 +384,12 @@ def build_multiplechoice_interface():
                 def update_exercise_type(exercise_type, question_category, number_of_correct_answers):
                     if exercise_type == 0:
                         return (
-                            gr.Dropdown(categories, label="Question Category", value="Analysys And Evaluation", type="index"),
+                            gr.Dropdown(categories, label="Question Category", value="Analysys And Evaluation"),
                             gr.Number(value=number_of_correct_answers, minimum=1, maximum=3, step=1, label="Number of correct answers")
                         )
                     elif exercise_type == 1:
                         return (
-                            gr.Dropdown(["Analysys And Evaluation"], label="Question Category", value="Analysys And Evaluation", type="index"),
+                            gr.Dropdown(["Analysys And Evaluation"], label="Question Category", value="Analysys And Evaluation"),
                             gr.Number(value=1, minimum=1, maximum=1, step=1, label="Number of correct answers")
                         )
 
@@ -423,7 +427,7 @@ def build_open_question_interface():
                 language = gr.Dropdown(languages, label="Language", value="English")
                 with gr.Row():
                     question_type = gr.Dropdown(["Open", "ShortAnswer", "TrueFalse"], label="Question Type", value="Open", type="index")
-                    question_category = gr.Dropdown(categories, label="Question Category", value="Factual Knowledge", type="index")
+                    question_category = gr.Dropdown(categories, label="Question Category", value="Factual Knowledge")
                 temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
         with gr.Row():
             output = gr.TextArea(placeholder="Generated text", label="Output")