Update app.py

app.py

@@ -1,22 +1,71 @@
-def respond(user_message, history, state_history):
+import os
+import gradio as gr
+from dotenv import load_dotenv
+from openai import OpenAI
+from prompts.main_prompt import MAIN_PROMPT
+from prompts.initial_prompt import INITIAL_PROMPT
+
+# ✅ Load API key from .env file
+if os.path.exists(".env"):
+    load_dotenv(".env")
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+
+if not OPENAI_API_KEY:
+    raise ValueError("🚨 OpenAI API key is missing! Set it in your .env file.")
+
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+def respond(user_message, history):
     if not user_message:
         return "", history
 
-    # ✅ Retrieve stored method choice
-    last_method = state_history.get("selected_method", None)
+    # ✅ Ensure proper message handling
+    try:
+        messages = [{"role": "system", "content": MAIN_PROMPT}]
+        
+        for user_text, assistant_text in history:
+            if user_text:
+                messages.append({"role": "user", "content": user_text})
+            if assistant_text:
+                messages.append({"role": "assistant", "content": assistant_text})
+
+        messages.append({"role": "user", "content": user_message})
+
+        # ✅ Get AI response
+        completion = client.chat.completions.create(
+            model="gpt-4o",
+            messages=messages,
+            max_tokens=512,
+            temperature=0.7
+        )
+
+        assistant_reply = completion.choices[0].message.content
+        history.append((user_message, assistant_reply))
+
+        return "", history
+
+    except Exception as e:
+        return f"⚠️ An error occurred: {str(e)}", history
+
+# ✅ Gradio UI Setup
+with gr.Blocks() as demo:
+    gr.Markdown("## 🤖 AI-Guided Math PD Chatbot")
+
+    chatbot = gr.Chatbot(value=[("", INITIAL_PROMPT)], height=500)
+    state_history = gr.State([("", INITIAL_PROMPT)])
 
-    # ✅ If the user selects a method, store it
-    if user_message.lower() in ["bar model", "double number line", "equation"]:
-        state_history["selected_method"] = user_message.lower()  # Store method
-        response = get_prompt_for_method(user_message)
-        history.append((user_message, response))
-        return "", history, state_history
+    user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")
 
-    # ✅ If no method was selected, ask for one
-    if not last_method:
-        return "I didn’t understand that. Please select a method first (Bar Model, Double Number Line, or Equation).", history, state_history
+    user_input.submit(
+        respond,
+        inputs=[user_input, state_history],
+        outputs=[user_input, chatbot]
+    ).then(
+        fn=lambda _, h: h,
+        inputs=[user_input, chatbot],
+        outputs=[state_history]
+    )
 
-    # ✅ Process teacher’s reasoning and give feedback
-    response = get_feedback_for_method(last_method, user_message)
-    history.append((user_message, response))
-    return "", history, state_history
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)