Update app.py

app.py

@@ -17,38 +17,39 @@ if not OPENAI_API_KEY:
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
+# ✅ Track the selected method
+selected_method = None  # This will store which method the teacher selects
+
 # ✅ Chatbot Response Function
 def respond(user_message, history):
     """
     Handles user input for the chatbot.
-    - If a method is selected, AI asks for teacher's reasoning first.
-    - AI only provides feedback after hearing the teacher's approach.
+    - Step 1: Teacher selects a method (Bar Model, Double Number Line, or Equation).
+    - Step 2: AI asks teacher to explain their reasoning before giving guidance.
+    - Step 3: AI listens to teacher's explanation and provides feedback.
     """
+    global selected_method  # Store the selected method so AI remembers it
+
     user_message = user_message.strip().lower()
 
-    # ✅ Check if the last message in history was a method selection prompt
-    if history and history[-1][1] in [
-        get_prompt_for_method("bar model"),
-        get_prompt_for_method("double number line"),
-        get_prompt_for_method("equation")
-    ]:
-        # ✅ Retrieve the last selected method
-        last_method = history[-1][1].split("**")[1].split(" ")[0].lower()
-        response = get_feedback_for_method(last_method, user_message)
+    # ✅ Step 1: Check if user selected a method (but hasn't explained yet)
+    if user_message in ["bar model", "double number line", "equation"]:
+        selected_method = user_message  # Store selected method
+        response = get_prompt_for_method(user_message)  # Ask for reasoning
         history.append((user_message, response))
         return "", history
 
-    # ✅ If user selects a method, ask them to explain their reasoning
-    if user_message in ["bar model", "double number line", "equation"]:
-        response = get_prompt_for_method(user_message)
+    # ✅ Step 2: If a method was selected, process teacher's explanation
+    if selected_method:
+        response = get_feedback_for_method(selected_method, user_message)  # Give feedback
         history.append((user_message, response))
         return "", history
 
-    # ✅ Default response if input is unclear and no method has been selected
-    if not history or history[-1][0] not in ["bar model", "double number line", "equation"]:
+    # ✅ If no method is selected yet, ask the user to select one
+    if not selected_method:
         return "I didn’t understand that. Please select a method first (Bar Model, Double Number Line, or Equation).", history
 
-    return "", history  # ✅ Keep the conversation flowing without errors
+    return "", history  # ✅ Maintain conversation flow
 
 # ✅ Gradio UI Setup
 with gr.Blocks() as demo: