Update app.py

app.py

@@ -1,34 +1,27 @@
 import gradio as gr
 from openai import OpenAI
 
-client = OpenAI()
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
 def chat(user_input, history=[]):
-    messages = [{"role": "system", "content": "You are Jarvis, a smart AI assistant."}]
-    
-    for h in history:
-        messages.append({"role": "user", "content": h[0]})
-        messages.append({"role": "assistant", "content": h[1]})
-    
-    messages.append({"role": "user", "content": user_input})
-
-    response = client.chat.completions.create(
-        model="gpt-4o-mini",
-        messages=messages
-    )
-
-    reply = response.choices[0].message.content
-    history.append((user_input, reply))
-    
-    return history, history
-
-with gr.Blocks() as demo:
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Ask Jarvis anything...")
-    
-    def respond(message, chat_history):
-        return chat(message, chat_history)
-    
-    msg.submit(respond, [msg, chatbot], [chatbot, chatbot])
-
-demo.launch()
+try:
+messages = [{"role": "system", "content": "You are Jarvis, a smart AI assistant."}]
+
+for h in history:
+messages.append({"role": "user", "content": h[0]})
+messages.append({"role": "assistant", "content": h[1]})
+
+messages.append({"role": "user", "content": user_input})
+
+response = client.chat.completions.create(
+model="gpt-4o-mini",
+messages=messages
+)
+
+reply = response.choices[0].message.content
+history.append((user_input, reply))
+
+return history, history
+
+except Exception as e:
+return [(user_input, "Error: " + str(e))], history