Update app.py

app.py

@@ -7,59 +7,106 @@ import sys
 import openai
 from openai import Completion as complete
 
-mcq_prompt = "Generate French MCQs on the above text. Each MCQ must have four choices. Generate ten questions with answers:"
+qa_gen_prompt = str("""You are a highly intelligent & complex question-answer generative model. 
+  You take a passage as an input and generate 5 high-quality and diverse multiple-choice questions 
+  with four choices each from the given passage by imitating the way a human asks questions and give answers. 
+  Your output format is only 
+    [{\"Q\": Question, \"A1\": Answer, \"A2\": Answer,\"A3\": Answer, \"A4\": Answer}, ...]\") }} form, no other form. \n The passage: """)
+end_prompt = "the QA pairs:"
 
 openai.api_key = os.environ["openai_api_key"]
 access_code = os.environ["access_code"]
 
-def generate_questions(context, credentials, num_questions, mcq_prompt=mcq_prompt):
-    """
-    Generate questions by calling davinci-003.
-    """
-    prompt = context + mcq_prompt
-    try: 
-        if credentials != access_code:
-            return "mauvais code d'accès"
-        else:
-            try:
-                completion = complete.create(model="text-davinci-003", prompt=prompt, max_tokens=1875)
-                return str(completion.choices[0].text)
-            except Exception as e:
-                # return str(e)
-                # return python version
-                return str(sys.version)
-    except:
-        return str(sys.version)
+def convert_questions_to_html(qapairs):
+  qapairs_template = []
+  for qapair in qapairs:
 
+    result = "<problem>\n<choiceresponse>  \n<label>" + qapair['Q'] + "</label>\n<checkboxgroup>"
 
-# def append_completion(selected_text, feedback):
-#     return selected_text, selected_text + feedback
-with gr.Blocks() as demo:
+    for key, value in qapair.items():
+        if key != "Q":
+            result += "\n    <choice correct=\"false\">" + value + "</choice>"
+    result+="\n"
+    result += "  </checkboxgroup>\n</choiceresponse>\n</problem>"
+    qapairs_template.append(result)
+  return "".join([qa + "\n\n" for qa in qapairs_template])
 
-    with gr.Row():
 
-        gr.Markdown("Pour créer des questions à choix multiples, insérez un texte dans la boîte à gauche, puis cliquez sur 'Générer dix questions.' Attention! N'oubliez pas le code d'accès.")
-    
-        credentials = gr.Textbox("Code d'accès.", interactive=True)
+def convert_questions_to_template(qapairs):
+  qapairs_template = []
+  for qapair in qapairs:
 
-    with gr.Row():
+    result = ">>" + qapair['Q'] + " <<\n\n"
+    choices = [qapair['A1'], qapair['A2'], qapair['A3'], qapair['A4']]
+    for i, choice in enumerate(choices):
+      result += "[ ] " + choice + "\n"
+      
+    qapairs_template.append(result)
 
-        with gr.Column():
+  return "".join([qa + "\n\n" for qa in qapairs_template])
 
-            quiz_button = gr.Button("Générer dix questions.")
 
-            context = gr.Textbox(placeholder="Insérez le texte ici.", lines=10)
+def list_of_dicts_to_str(qapairs, indent=0):
+    result = ''
+    for d in qapairs:
+        for key, value in d.items():
+            if key.startswith("A"):
+                result += ' ' * (indent + 4) + str(key) + ': ' + str(value) + '\n'
+            else:
+                result += ' ' * indent + str(key) + ': ' + str(value) + '\n'
+    return result
 
-            slider = gr.Slider(minimum=1, maximum=10, value=5, step=1, label="Nombre de questions.")
 
+def generate_questions(context, credentials):
+    """
+    Generate questions by calling davinci-003.
+    """
+    if credentials != access_code:
+        return "Code d'accès incorrect. Veuillez réessayer.", "Code d'accès incorrect. Veuillez réessayer.", "Code d'accès incorrect. Veuillez réessayer."
 
-        qa_pairs = gr.Textbox(placeholder="Les questions apparaîtront ici.", lines=15)
+    prompt = qa_gen_prompt + context + end_prompt
     
+    try:
+        completion = complete.create(model="text-davinci-003", prompt=prompt, max_tokens=1875)
+        q_dict = eval(completion.choices[0].text)
+        return list_of_dicts_to_str(q_dict), convert_questions_to_html(q_dict), convert_questions_to_template(q_dict)
+    except Exception as e:
+        # return str(e)
+        # return python version
+        return str(sys.version)
+
+
+with gr.Blocks() as demo:
+
+    # with gr.Row():
+        
+    #     with gr.Column():
+
+
+    with gr.Row():
+        
+        with gr.Column():   
+            gr.Markdown("## Générateur de questions à choix multiples")    
+            gr.Markdown("Pour créer des questions à choix multiples, insérez un texte dans la boîte à gauche, puis cliquez sur 'Générer des questions.' Attention! N'oubliez pas le code d'accès.")
+            credentials = gr.Textbox(placeholder="Code d'accès", lines=1)
+            context = gr.Textbox(placeholder="insérez le texte ici.", lines=8)
+
+            quiz_button = gr.Button("Générer cinq questions.", variant="primary")   
+
+        with gr.Tab("brut"):
+            qa_pairs = gr.Textbox(placeholder="Les questions apparaîtront ici.", lines=19)
+        with gr.Tab("format html"):
+            qa_pairs_html = gr.Textbox(placeholder="Les questions apparaîtront ici.", lines=19)
+        with gr.Tab("format edulib"):
+            qa_pairs_edulib = gr.Textbox(placeholder="Les questions apparaîtront ici.", lines=19)
+
         #todo: test generating with just one phrase for one MCQ with smaller models
 
-    quiz_button.click(fn=generate_questions,
-                         inputs=[context, credentials, slider],	
-                         outputs=qa_pairs)
+    quiz_button.click(fn= generate_questions,
+                          inputs=[context, credentials],	
+                          outputs=[qa_pairs, qa_pairs_html, qa_pairs_edulib]
+                     )
+    
 
 if __name__ == "__main__":
-    demo.launch()    
+    demo.launch()