Update app.py

app.py

@@ -1,8 +1,8 @@
 import os
 import gradio as gr
 from openai import OpenAI
-from prompts.main_prompt import MAIN_PROMPT  # ✅ Fix Import Path
-from prompts.initial_prompt import INITIAL_PROMPT  # ✅ Ensure Initial Prompt is Included
+from prompts.main_prompt import MAIN_PROMPT  # ✅ Import Main Prompt Correctly
+from prompts.initial_prompt import INITIAL_PROMPT  # ✅ Import Initial Prompt
 
 # ✅ Ensure OpenAI API Key is Set (Use Secrets in Hugging Face)
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
@@ -20,7 +20,7 @@ def respond(user_message, history):
         assistant_reply = client.chat.completions.create(
             model="gpt-4o",
             messages=[
-                {"role": "system", "content": MAIN_PROMPT},
+                {"role": "system", "content": MAIN_PROMPT},  # ✅ Ensures MAIN_PROMPT is used first
                 *[
                     {"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": a}
                     for i, (u, a) in enumerate(history)
@@ -40,13 +40,14 @@ def respond(user_message, history):
 # ✅ Gradio UI
 with gr.Blocks() as demo:
     gr.Markdown("# **AI-Guided Math PD Chatbot**")
-    
+
+    # ✅ Start Chatbot with MAIN_PROMPT Instead of Initial Reflection
     chatbot = gr.Chatbot(
-        value=[("", INITIAL_PROMPT)],  # ✅ Shows Initial Prompt on Startup
+        value=[("", MAIN_PROMPT)],  # ✅ Ensures chatbot begins with the math problems
         height=500
     )
 
-    state_history = gr.State([("", INITIAL_PROMPT)])  # ✅ Ensures History Starts with Initial Prompt
+    state_history = gr.State([("", MAIN_PROMPT)])  # ✅ Makes sure history starts correctly
 
     user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")