Update app.py

app.py

@@ -1,18 +1,29 @@
-import openai
 import gradio as gr
 import os
+from openai import OpenAI
 
-openai.api_key = os.getenv("OPENAI_API_KEY")
+# Load your OpenAI API key from Hugging Face Secrets or environment
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-def simple_agent(message):
+# Chat function using new v1 OpenAI SDK
+def simple_agent(message, history):
     try:
-        response = openai.ChatCompletion.create(
+        response = client.chat.completions.create(
             model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": message}]
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant."}
+            ] + [{"role": "user", "content": m} for m, _ in history] + [{"role": "user", "content": message}]
         )
-        return response.choices[0].message["content"]
+        return response.choices[0].message.content
     except Exception as e:
         return f"Error: {str(e)}"
 
-iface = gr.Interface(fn=simple_agent, inputs="text", outputs="text", title="Simple AI Agent")
+# Gradio chat interface
+iface = gr.ChatInterface(
+    fn=simple_agent,
+    title="Simple AI Agent",
+    theme="default",
+    type="messages"
+)
+
 iface.launch()