Update app.py

app.py

@@ -1,28 +1,18 @@
-import gradio as gr
 import openai
+import gradio as gr
 import os
 
 openai.api_key = os.getenv("OPENAI_API_KEY")
-try:
-    response = openai.ChatCompletion.create(...)
-except Exception as e:
-    return f"Error: {str(e)}"
-
-def simple_agent(message, history):
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        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"]
-
-iface = gr.ChatInterface(
-    simple_agent,
-    title="Simple AI Agent",
-    type="messages"  # <--- this avoids the deprecation warning
-)
-
-iface.launch(share=True)  # <--- this creates a public link
 
+def simple_agent(message):
+    try:
+        response = openai.ChatCompletion.create(
+            model="gpt-3.5-turbo",
+            messages=[{"role": "user", "content": message}]
+        )
+        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")
+iface.launch()