Update app.py

app.py

@@ -5,60 +5,52 @@ 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,
     )
-    current_question = response.choices[0].text.strip()
+    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: {current_question}\nAnswer: ",
+      prompt=f"What is the answer to the following question based on the text: \"{text}\"?\nQuestion: {question}\nAnswer: ",
       max_tokens=50,
       n=1,
     )
-    correct_answer = response.choices[0].text.strip()
+    answer = response.choices[0].text.strip()
+    
+    return question, answer
 
-def get_feedback(text, user_answer=None, continue_quiz='Yes'):
-    global current_step, current_question, correct_answer
+def get_feedback(text=None, user_answer=None, continue_quiz=None, state=None):
+    if state is None:
+        state = {}
 
-    if current_step == "question_generation":
+    if 'step' not in state:
+        state['step'] = 'get_text'
+
+    if state['step'] == 'get_text':
         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":
+            state['question'], state['correct_answer'] = generate_question_and_answer(text)
+            state['step'] = 'get_answer'
+            return f"Question: {state['question']}", state
+
+    elif state['step'] == 'get_answer':
         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"
+            feedback = "Correct!" if user_answer.lower() == state['correct_answer'].lower() else f"Incorrect! The correct answer was: {state['correct_answer']}"
+            state['step'] = 'continue_quiz'
+            return f"Feedback: {feedback}\nDo you want to answer another question?", state
+
+    elif state['step'] == 'continue_quiz':
+        if continue_quiz:
+            if continue_quiz.lower() == 'no':
+                state['step'] = 'get_text'
+                return "Quiz ended. Thank you for participating! Please enter new text to start again.", state
+            else:
+                state['step'] = 'get_text'
+                return "Please input text to generate a new question.", state
 
 iface = gr.Interface(
     fn=get_feedback,
@@ -69,7 +61,7 @@ iface = gr.Interface(
     ],
     outputs=[
         gr.outputs.Textbox(label="Model Output"),
-        gr.outputs.Textbox(label="Prompt"),
+        gr.outputs.State(label="State")
     ],
     live=True
 )