Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 from dotenv import load_dotenv
 from openai import OpenAI
 from prompts.initial_prompt import INITIAL_PROMPT
-from prompts.main_prompt import get_prompt_for_method, get_feedback_for_method
+from prompts.main_prompt import MAIN_PROMPT, get_prompt_for_method, get_feedback_for_method
 
 # ✅ Load API key from .env file
 if os.path.exists(".env"):
@@ -30,16 +30,18 @@ def respond(user_message, history, selected_method):
     if user_message in valid_methods:
         selected_method = user_message  # Store the method
         method_prompt = get_prompt_for_method(user_message)
-        history.append((user_message, method_prompt))  # Save to history
+        history.append((user_message, method_prompt))  # Ensure tuple format
         return method_prompt, history, selected_method
 
     # ✅ If a method has already been selected, provide feedback
     if selected_method:
         feedback = get_feedback_for_method(selected_method, user_message)
-        history.append((user_message, feedback))  # Save user response and feedback
+        history.append((user_message, feedback))  # Ensure tuple format
         return feedback, history, selected_method
 
-    return "❌ Please select a method first (Bar Model, Double Number Line, or Equation).", history, selected_method
+    error_msg = "❌ Please select a method first (Bar Model, Double Number Line, or Equation)."
+    history.append((user_message, error_msg))  # Ensure tuple format
+    return error_msg, history, selected_method
 
 # ✅ Gradio UI Setup
 with gr.Blocks() as demo: