Update app.py

app.py

@@ -1,43 +1,48 @@
+import os
+import openai
 import gradio as gr
-from openai import OpenAI
-
-def generate_response(user_input):
-    client = OpenAI()
-    response = client.responses.create(
-        model="gpt-4o",
-        input=[
-            {
-                "role": "system",
-                "content": [
-                    {
-                        "type": "input_text",
-                        "text": user_input
-                    }
-                ]
-            }
-        ],
-        text={
-            "format": {
-                "type": "text"
-            }
-        },
-        reasoning={},
-        tools=[],
-        temperature=1,
-        max_output_tokens=2048,
-        top_p=1,
-        store=True
-    )
+
+# Retrieve OpenAI API key from Hugging Face Secrets
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+# Initialize OpenAI client
+client = openai.OpenAI(api_key=openai_api_key)
+
+# Chatbot function
+def chatbot(user_input, history=[]):
+    if not openai_api_key:
+        return "⚠️ API key is missing. Please configure it in Hugging Face Secrets.", history
+
+    history.append({"role": "user", "content": user_input})
+
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o",
+            messages=history,
+            temperature=0.7,
+            max_tokens=200,
+            top_p=1
+        )
+
+        bot_reply = response.choices[0].message.content
+        history.append({"role": "assistant", "content": bot_reply})
+
+    except Exception as e:
+        bot_reply = f"❌ Error: {str(e)}"
     
-    return response.text  # Adjust based on OpenAI response format
-
-# Gradio UI
-demo = gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=gr.Textbox(label="Generated Response"),
-    title="OpenAI GPT-4o Bot Guide Generator",
-    description="Enter a request, and the bot will generate a response using OpenAI's API."
+    return bot_reply, history
+
+# Gradio Interface
+chatbot_ui = gr.ChatInterface(
+    fn=chatbot,
+    title="AI Chatbot",
+    description="A simple chatbot powered by GPT-4o.",
+    theme="soft",
+    retry_btn="🔄 Retry",
+    undo_btn="↩️ Undo",
+    clear_btn="🗑️ Clear"
 )
 
-demo.launch()
+# Launch the app
+if __name__ == "__main__":
+    chatbot_ui.launch()