Update app.py

app.py

@@ -1,3 +1,22 @@
+import os
+import gradio as gr
+from dotenv import load_dotenv
+from openai import OpenAI
+from prompts.main_prompt import MAIN_PROMPT, get_prompt_for_method
+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.")
+
+# ✅ Correct OpenAI Client Initialization
+client = OpenAI()
+
 def respond(user_message, history):
     if not user_message:
         return "", history
@@ -23,7 +42,7 @@ def respond(user_message, history):
 
         messages.append({"role": "user", "content": user_message})
 
-        # ✅ Get AI response
+        # ✅ Get AI response (Updated call to OpenAI API)
         completion = client.chat.completions.create(
             model="gpt-4o",
             messages=messages,
@@ -38,3 +57,25 @@ def respond(user_message, 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)])
+
+    user_input = gr.Textbox(placeholder="Type your message here...", label="Your Input")
+
+    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]
+    )
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=7860)