Update app.py

app.py

@@ -42,9 +42,9 @@ def submit_answer(user_ipa, user_id):
     if user_ipa.strip() == session['correct_ipa']:
         score = session['score'] + 1
         session['score'] = score
-        return "Correct!", session['score'], session['trials']
+        return f"Correct! The answer was '{session['correct_ipa']}'.", session['score'], session['trials']
     else:
-        return "Try again", session['score'], session['trials']
+        return f"Try again. The correct answer was '{session['correct_ipa']}'.", session['score'], session['trials']
 
 # Function to quit and show results
 def quit_quiz(user_id):
@@ -68,6 +68,9 @@ def gradio_app():
         result_output = gr.Textbox(label="Result", interactive=False)
         quit_button = gr.Button("Quit")
         
+        # State variables for user ID
+        user_id_state = gr.State()
+
         # Initialize a unique session ID for each user
         def init_session():
             user_id = random.randint(1000, 9999)  # Use random number as session ID
@@ -81,13 +84,18 @@ def gradio_app():
         # Start quiz based on selections
         def start(segment_type, description_type):
             user_id = init_session()  # Create session for this user
+            # Reset score and trials to 0 when the quiz starts again
+            user_sessions[user_id]["score"] = 0
+            user_sessions[user_id]["trials"] = 0
             description = generate_question(segment_type, description_type, user_id)
             return description, "", gr.update(visible=True), user_id, user_sessions[user_id]['score'], user_sessions[user_id]['trials']
         
-        # Submit the answer and check
-        def submit(user_ipa, user_id):
+        # Submit the answer and check, and generate new question
+        def submit(user_ipa, segment_type, description_type, user_id):
             result, score, trials = submit_answer(user_ipa, user_id)
-            return result, score, trials
+            # After checking, generate a new description automatically
+            new_description = generate_question(segment_type, description_type, user_id)
+            return new_description, "", result, score, trials
         
         # Quit and show final results
         def quit(user_id):
@@ -95,10 +103,10 @@ def gradio_app():
         
         # Bind actions
         start_button.click(fn=start, inputs=[segment_type, description_type], 
-                           outputs=[description_output, ipa_input, submit_button, gr.State(), gr.State(), gr.State()])
-        submit_button.click(fn=submit, inputs=[ipa_input, gr.State()], 
-                            outputs=[result_output, gr.State(), gr.State()])
-        quit_button.click(fn=quit, inputs=[gr.State()], outputs=[result_output])
+                           outputs=[description_output, ipa_input, submit_button, user_id_state, gr.State(), gr.State()])
+        submit_button.click(fn=submit, inputs=[ipa_input, segment_type, description_type, user_id_state], 
+                            outputs=[description_output, ipa_input, result_output, gr.State(), gr.State()])
+        quit_button.click(fn=quit, inputs=[user_id_state], outputs=[result_output])
     
     return app