Update demo.py

demo.py

@@ -25,21 +25,22 @@ firebase_current_ref = None
 
 
 base_url = "https://skapi.polyglot-edu.com/"
-levels = ["Pre K", "Primary", "Middle School", "High School", "Academic"]
+levels = ["Primary School", "Middle School", "High School", "College", "Academy"]
+languages = ["English", "Italian", "French", "German", "Spanish"]
 
 def get_level_mapping(level):
     if level is None:
         raise gr.Error("Please select a level.")
 
     return {
-        "Academic": 0,
-        "Pre K": 1,
-        "Primary": 2,
-        "Middle School": 3,
-        "High School": 4,
+        "Primary School": 0,
+        "Middle School": 1,
+        "High School": 2,
+        "College": 3,
+        "Academy": 4,
     }[level]
 
-def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, number_of_distractors, temperature):
+def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, number_of_distractors, temperature, language):
     """
     Generate a fill-gaps question from a given text.
 
@@ -56,6 +57,8 @@ def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, nu
         The number of distractors to generate for each gap.
     temperature : float
         The temperature for the generation.
+    language : str
+        The language of the generated text.
 
     Returns
     -------
@@ -77,7 +80,8 @@ def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, nu
         "n_o_w": int(number_of_words),
         "n_o_g": int(number_of_gaps),
         "n_o_d": int(number_of_distractors),
-        "temperature": int(temperature) * 0.2
+        "temperature": int(temperature) * 0.2,
+        "language": language
     }
 
     output = ""
@@ -107,7 +111,7 @@ def generate_fill_gaps(original_text, level, number_of_words, number_of_gaps, nu
             "flagged": False,
         })
 
-def generate_open_question(original_text, level, temperature):
+def generate_open_question(original_text, level, temperature, language, question_type, question_category):
     """
     Generate an open question from a given text.
 
@@ -118,6 +122,12 @@ def generate_open_question(original_text, level, temperature):
         The original text from which to generate the open question.
     temperature : float
         The temperature for the generation.
+    language : str
+        The language of the generated text.
+    question_type : str
+        The type of the question.
+    question_category : str
+        The category of the question.
 
     Returns
     -------
@@ -128,7 +138,10 @@ def generate_open_question(original_text, level, temperature):
     input_json = {
         "text": original_text,
         "level": get_level_mapping(level),
-        "temperature": int(temperature) * 0.2
+        "temperature": int(temperature) * 0.2,
+        "language": language,
+        "question_type": question_type,
+        "question_category": question_category
     }
 
     output = ""
@@ -158,7 +171,7 @@ def generate_open_question(original_text, level, temperature):
             "flagged": False,
         })
 
-def generate_multiplechoice(original_text, level, number_of_options, number_of_easy_distractors, number_of_distractors, temperature):
+def generate_multiplechoice(original_text, level, number_of_easy_distractors, number_of_distractors, temperature, language):
     """
     Generate a multiple-choice question from a given text.
 
@@ -167,14 +180,14 @@ def generate_multiplechoice(original_text, level, number_of_options, number_of_e
     ----------
     original_text : str
         The original text from which to generate the multiple-choice question.
-    number_of_options : int
-        The number of options to generate.
     number_of_easy_distractors : int
         The number of easy distractors to generate for each option.
     number_of_distractors : int
         The number of distractors to generate for each option.
     temperature : float
         The temperature for the generation.
+    language : str
+        The language of the generated text.
 
     Returns
     -------
@@ -188,9 +201,12 @@ def generate_multiplechoice(original_text, level, number_of_options, number_of_e
         "level": get_level_mapping(level),
         "n_o_d": int(number_of_distractors),
         "nedd": int(number_of_easy_distractors),
-        "temperature": int(temperature) * 0.2
+        "temperature": int(temperature) * 0.2,
+        "language": language
     }
 
+    print(input_json)
+
     output = ""
     try:
         response = requests.post(
@@ -278,15 +294,17 @@ def build_fill_gaps_interface():
     with gr.Blocks(title="Fill-gaps") as demo:
         with gr.Row():
             with gr.Column(scale=5):
-                input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)")
+                input_field = gr.TextArea(lines=10, max_lines=10, label="Input Educational resource text or URL",
+                                          info="Note: If the input resource is too long, the API will return an error. If the input is an URL, it must be a valid URL to webpage or a file.")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column(scale=4):
-                output_text_length = gr.Radio(["≈ 150 Words", "≈ 250 Words", "≈ 350 Words"], label="Output text length", value="≈ 150 Words")
                 level = gr.Radio(levels, label="Level")
+                language = gr.Dropdown(languages, label="Language", value="English")
+                output_text_length = gr.Radio(["≈ 150 Words", "≈ 250 Words", "≈ 350 Words"], label="Output text length", value="≈ 150 Words")
                 with gr.Row():
                     blanks = gr.Number(value=5, minimum=4, maximum=8, step=1, label="Number of blanks")
                     distractors = gr.Number(value=2, minimum=0, maximum=5, step=1, label="Number of distractors")
-                temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)")
+                temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
 
                 def update_numeric(output_text_length, blanks, distractors):
                     if output_text_length == "≈ 150 Words":
@@ -314,16 +332,11 @@ def build_fill_gaps_interface():
             remove_preference_btn = gr.Button(value="Remove Preference")
             downvote_btn = gr.Button(value="👎  Downvote")
             flag_btn = gr.Button(value="⚠️  Flag")
-        gr.Markdown("<hr/>")
-        with gr.Row():
-            comment_text = gr.TextArea(placeholder="Comment", label="Leave a comment about the generated text", scale=6)
-            comment_btn = gr.Button(value="💬 Comment", scale=2)
-
-        submit_btn.click(generate_fill_gaps, [input_field, level, output_text_length, blanks, distractors, temperature], [output])
+        
+        submit_btn.click(generate_fill_gaps, [input_field, level, output_text_length, blanks, distractors, temperature, language], [output])
         upvote_btn.click(like)
         remove_preference_btn.click(neutral_like)
         downvote_btn.click(dislike)
-        comment_btn.click(comment, [comment_text])
         flag_btn.click(flag, [flag_btn], [flag_btn])
 
         return demo
@@ -335,31 +348,22 @@ def build_multiplechoice_interface():
     with gr.Blocks(title="Open Question") as demo:
         with gr.Row():
             with gr.Column(scale=5):
-                input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)")
+                input_field = gr.TextArea(lines=10, max_lines=10, label="Input Educational resource text or URL",
+                                          info="Note: If the input resource is too long, the API will return an error. If the input is an URL, it must be a valid URL to webpage or a file.")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column(scale=4):
                 level = gr.Radio(levels, label="Level")
+                language = gr.Dropdown(languages, label="Language", value="English")
                 with gr.Row():
-                    options = gr.Number(value=4, minimum=2, maximum=8, step=1, label="Number of blanks", interactive=False)
                     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)")
-
-                def update_options(options, easy_distractors, distractors):
-                    distractors = clamp(distractors, 0, options)
-                    easy_distractors = clamp(easy_distractors, 0, distractors)
-
-                    return (
-                        gr.Number(value=easy_distractors, minimum=0, maximum=distractors, label="Number of easy distractors"),
-                        gr.Number(value=distractors, minimum=0, maximum=options, label="Number of distractors")
-                    )
+                temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
                 
                 def update_distractors(easy_distractors, distractors):
                     easy_distractors = clamp(easy_distractors, 0, distractors)
 
                     return gr.Number(value=easy_distractors, minimum=0, maximum=distractors, label="Number of easy distractors")
 
-                options.change(update_options, [options, easy_distractors, distractors], [easy_distractors, distractors])
                 distractors.change(update_distractors, [easy_distractors, distractors], [easy_distractors])
 
         with gr.Row():
@@ -369,16 +373,11 @@ def build_multiplechoice_interface():
             remove_preference_btn = gr.Button(value="Remove Preference")
             downvote_btn = gr.Button(value="👎  Downvote")
             flag_btn = gr.Button(value="⚠️  Flag")
-        gr.Markdown("<hr/>")
-        with gr.Row():
-            comment_text = gr.TextArea(placeholder="Comment", label="Comment", scale=6)
-            comment_btn = gr.Button(value="💬 Comment", scale=2)
-
-        submit_btn.click(generate_multiplechoice, [input_field, level, options, easy_distractors, distractors, temperature], [output])
+        
+        submit_btn.click(generate_multiplechoice, [input_field, level, easy_distractors, distractors, temperature, language], [output])
         upvote_btn.click(like)
         remove_preference_btn.click(neutral_like)
         downvote_btn.click(dislike)
-        comment_btn.click(comment, [comment_text])
         flag_btn.click(flag, [flag_btn], [flag_btn])
 
         return demo
@@ -390,13 +389,18 @@ def build_open_question_interface():
     with gr.Blocks(title="Multiple choice") as demo:
         with gr.Row():
             with gr.Column(scale=5):
-                input_field = gr.TextArea(lines=10, max_lines=10, label="Input text (can be a text or a url)")
+                input_field = gr.TextArea(lines=10, max_lines=10, label="Input Educational resource text or URL",
+                                          info="Note: If the input resource is too long, the API will return an error. If the input is an URL, it must be a valid URL to webpage or a file.")
                 submit_btn = gr.Button(value="Submit")
             with gr.Column(scale=4):
                 level = gr.Radio(levels, label="Level")
+                language = gr.Dropdown(languages, label="Language", value="English")
                 with gr.Row():
-                    pass
-                temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)")
+                    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")
+                temperature = gr.Checkbox(value=False, label="Increase creativity (decreases preciseness)", visible=False)
         with gr.Row():
             output = gr.TextArea(placeholder="Generated text", label="Output")
         with gr.Row() as button_row:
@@ -404,26 +408,40 @@ def build_open_question_interface():
             remove_preference_btn = gr.Button(value="Remove Preference")
             downvote_btn = gr.Button(value="👎  Downvote")
             flag_btn = gr.Button(value="⚠️  Flag")
-        gr.Markdown("<hr/>")
-        with gr.Row():
-            comment_text = gr.TextArea(placeholder="Comment", label="Comment", scale=6)
-            comment_btn = gr.Button(value="💬 Comment", scale=2)
 
-        submit_btn.click(generate_open_question, [input_field, level, temperature], [output])
+        submit_btn.click(generate_open_question, [input_field, level, temperature, language, question_type, question_category], [output])
         upvote_btn.click(like)
         remove_preference_btn.click(neutral_like)
         downvote_btn.click(dislike)
-        comment_btn.click(comment, [comment_text])
         flag_btn.click(flag, [flag_btn], [flag_btn])
 
         return demo
 
 def build_demo():
-    return gr.TabbedInterface(
-        [build_fill_gaps_interface(), build_open_question_interface(), build_multiplechoice_interface()],
-        ["Fill-gaps", "Open question", "Multiple-choice"],
-        title="Education AI"
-    )
+    with gr.Blocks(title="Educational AI") as demo:
+        gr.Markdown("<h1 style='text-align: center; margin-bottom: 1rem'>Educational AI</h1>")
+
+
+        with gr.Row():
+            gr.Markdown("<h4>Click on the button on the right to fill up our questionnaire!</h4>")
+            gr.Button(value="📝  Questionnaire", link="https://forms.gle/T8CS5CiQgPbKUdeM9", scale=0.5, interactive=True)
+
+        with gr.Tab("Fill-gaps"):
+            build_fill_gaps_interface()
+
+        with gr.Tab("Open question"):
+            build_open_question_interface()
+
+        with gr.Tab("Multiple-choice"):
+            build_multiplechoice_interface()
+
+        with gr.Blocks():
+            with gr.Row():
+                comment_text = gr.TextArea(placeholder="Comment", label="Comment", scale=6)
+                comment_btn = gr.Button(value="💬 Comment", scale=2)
+                comment_btn.click(comment, [comment_text])
+
+        return demo
 
 if __name__ == "__main__":
     build_demo().launch(share=False)