Update app.py

app.py

@@ -5,36 +5,60 @@ import openai
 # Replace with your OpenAI API key
 openai.api_key = "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"
 
+# Global variables to store the state
+current_step = "question_generation"
+current_question = ""
+correct_answer = ""
+
+def reset_state():
+    global current_step, current_question, correct_answer
+    current_step = "question_generation"
+    current_question = ""
+    correct_answer = ""
+
 def generate_question_and_answer(text):
+    global current_question, correct_answer
+
     response = openai.Completion.create(
       engine="gpt-3.5-turbo-0301",
       prompt=f"Create a question based on the following text: \"{text}\".\nQuestion: ",
       max_tokens=50,
       n=1,
     )
-    question = response.choices[0].text.strip()
+    current_question = response.choices[0].text.strip()
 
     response = openai.Completion.create(
       engine="gpt-3.5-turbo-0301",
-      prompt=f"What is the answer to the following question based on the text: \"{text}\"?\nQuestion: {question}\nAnswer: ",
+      prompt=f"What is the answer to the following question based on the text: \"{text}\"?\nQuestion: {current_question}\nAnswer: ",
       max_tokens=50,
       n=1,
     )
-    answer = response.choices[0].text.strip()
-    
-    return question, answer
+    correct_answer = response.choices[0].text.strip()
 
 def get_feedback(text, user_answer=None, continue_quiz='Yes'):
-    if continue_quiz.lower() == 'no':
-        return "Quiz ended. Thank you for participating!"
+    global current_step, current_question, correct_answer
 
-    question, correct_answer = generate_question_and_answer(text)
-
-    if user_answer is not None:
-        feedback = "Correct!" if user_answer.lower() == correct_answer.lower() else f"Incorrect! The correct answer was: {correct_answer}"
-        return f"Question: {question}\nFeedback: {feedback}"
-    else:
-        return f"Question: {question}"
+    if current_step == "question_generation":
+        if text:
+            generate_question_and_answer(text)
+            current_step = "answer_checking"
+            return f"Question: {current_question}", "", "hidden"
+        else:
+            return "Please input text to generate a question.", "", "hidden"
+    elif current_step == "answer_checking":
+        if user_answer is not None:
+            feedback = "Correct!" if user_answer.lower() == correct_answer.lower() else f"Incorrect! The correct answer was: {correct_answer}"
+            current_step = "continue_prompt"
+            return f"Feedback: {feedback}", "Do you want to answer another question?", "text"
+        else:
+            return f"Question: {current_question}", "", "hidden"
+    elif current_step == "continue_prompt":
+        if continue_quiz.lower() == 'no':
+            reset_state()
+            return "Quiz ended. Thank you for participating!", "", "hidden"
+        else:
+            current_step = "question_generation"
+            return "Please input text to generate a new question.", "", "hidden"
 
 iface = gr.Interface(
     fn=get_feedback,
@@ -43,7 +67,12 @@ iface = gr.Interface(
         gr.inputs.Textbox(lines=1, label="Your Answer"),
         gr.inputs.Radio(choices=["Yes", "No"], label="Continue?")
     ],
-    outputs=gr.outputs.Textbox(label="Model Feedback"),
+    outputs=[
+        gr.outputs.Textbox(label="Model Output"),
+        gr.outputs.Textbox(label="Prompt"),
+        gr.outputs.Textbox(label="Continue?", type="hidden")
+    ],
+    live=True
 )
 
-iface.launch()
+iface.launch()