Update demo.py

demo.py

@@ -27,6 +27,7 @@ firebase_current_ref = None
 base_url = "https://skapi.polyglot-edu.com/"
 levels = ["Primary School", "Middle School", "High School", "College", "Academy"]
 languages = ["English", "Italian", "French", "German", "Spanish"]
+categories = ["Factual Knowledge", "Understanding of Concepts", "Application of Skills", "Analysys And Evaluation"]
 
 def get_level_mapping(level):
     if level is None:
@@ -171,7 +172,7 @@ def generate_open_question(original_text, level, temperature, language, question
             "flagged": False,
         })
 
-def generate_multiplechoice(original_text, level, number_of_easy_distractors, number_of_distractors, temperature, language):
+def generate_multiplechoice(original_text, level, number_of_easy_distractors, number_of_distractors, temperature, language, question_category, correct_answers, exercise_type):
     """
     Generate a multiple-choice question from a given text.
 
@@ -188,6 +189,12 @@ def generate_multiplechoice(original_text, level, number_of_easy_distractors, nu
         The temperature for the generation.
     language : str
         The language of the generated text.
+    question_category : str
+        The category of the question.
+    correct_answers : int
+        The number of correct answers.
+    exercise_type : bool
+        The type of the exercise (theory or practice).
 
     Returns
     -------
@@ -197,8 +204,10 @@ def generate_multiplechoice(original_text, level, number_of_easy_distractors, nu
     
     input_json = {
         "text": original_text,
-        "type": True,
+        "type": bool(exercise_type),
         "level": get_level_mapping(level),
+        "question_category": question_category,
+        "n_o_ca": int(correct_answers),
         "n_o_d": int(number_of_distractors),
         "nedd": int(number_of_easy_distractors),
         "temperature": int(temperature) * 0.2,
@@ -355,6 +364,10 @@ def build_multiplechoice_interface():
                 level = gr.Radio(levels, label="Level")
                 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")
+                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")
                     distractors = gr.Number(value=1, minimum=0, maximum=8, step=1, label="Number of distractors")
                 temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
@@ -364,7 +377,20 @@ def build_multiplechoice_interface():
 
                     return gr.Number(value=easy_distractors, minimum=0, maximum=distractors, label="Number of easy distractors")
 
+                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.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.Number(value=1, minimum=1, maximum=1, step=1, label="Number of correct answers")
+                        )
+
                 distractors.change(update_distractors, [easy_distractors, distractors], [easy_distractors])
+                exercise_type.change(update_exercise_type, [exercise_type, question_category, correct_answers], [question_category, correct_answers])
 
         with gr.Row():
             output = gr.TextArea(placeholder="Generated text", label="Output")
@@ -374,7 +400,7 @@ def build_multiplechoice_interface():
             downvote_btn = gr.Button(value="👎  Downvote")
             flag_btn = gr.Button(value="⚠️  Flag")
         
-        submit_btn.click(generate_multiplechoice, [input_field, level, easy_distractors, distractors, temperature, language], [output])
+        submit_btn.click(generate_multiplechoice, [input_field, level, easy_distractors, distractors, temperature, language, question_category, correct_answers, exercise_type], [output])
         upvote_btn.click(like)
         remove_preference_btn.click(neutral_like)
         downvote_btn.click(dislike)
@@ -397,9 +423,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(
-                        ["Factual Knowledge", "Understanding of Concepts", "Application of Skills", "Analysys And Evaluation"],
-                        label="Question Category", value="Factual Knowledge", type="index")
+                    question_category = gr.Dropdown(categories, label="Question Category", value="Factual Knowledge", type="index")
                 temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
         with gr.Row():
             output = gr.TextArea(placeholder="Generated text", label="Output")