Update app.py

app.py

@@ -1,10 +1,10 @@
 import os
 import gradio as gr
 from openai import OpenAI
-from prompts.main_prompt import MAIN_PROMPT  # ✅ Import Main Prompt Correctly
-from prompts.initial_prompt import INITIAL_PROMPT  # ✅ Import Initial Prompt
+from prompts.main_prompt import MAIN_PROMPT
+from prompts.initial_prompt import INITIAL_PROMPT
 
-# ✅ Ensure OpenAI API Key is Set (Use Secrets in Hugging Face)
+# ✅ Load API Key
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 if not OPENAI_API_KEY:
@@ -12,22 +12,24 @@ if not OPENAI_API_KEY:
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
+# ✅ Ensure Conversation Happens Step by Step
 def respond(user_message, history):
     if not user_message:
         return "", history
 
+    # ✅ AI only sends ONE response at a time (step-by-step)
     try:
         assistant_reply = client.chat.completions.create(
             model="gpt-4o",
             messages=[
-                {"role": "system", "content": MAIN_PROMPT},  # ✅ Ensures MAIN_PROMPT is used first
+                {"role": "system", "content": MAIN_PROMPT},
                 *[
                     {"role": "user", "content": u} if i % 2 == 0 else {"role": "assistant", "content": a}
                     for i, (u, a) in enumerate(history)
                 ],
                 {"role": "user", "content": user_message}
             ],
-            max_tokens=512,
+            max_tokens=256,  # ✅ Limits response size (so AI doesn’t dump everything)
             temperature=0.7,
         ).choices[0].message.content
     except Exception as e:
@@ -37,17 +39,16 @@ def respond(user_message, history):
 
     return "", history
 
-# ✅ Gradio UI
+# ✅ Fix Gradio UI to Start Properly
 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=[("", MAIN_PROMPT)],  # ✅ Ensures chatbot begins with the math problems
+        value=[("", INITIAL_PROMPT)],  # ✅ Starts with an introduction message
         height=500
     )
 
-    state_history = gr.State([("", MAIN_PROMPT)])  # ✅ Makes sure history starts correctly
+    state_history = gr.State([("", INITIAL_PROMPT)])  # ✅ Ensures step-by-step history
 
     user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")