Anmol3263 commited on
Commit
d9bbc78
·
verified ·
1 Parent(s): 225a5b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -13
app.py CHANGED
@@ -4,29 +4,26 @@ import os
4
 
5
  client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
6
 
7
- def chat(user_input, history=[]):
8
  try:
9
  messages = [{"role": "system", "content": "You are Jarvis, a smart AI assistant."}]
10
 
11
- for h in history:
12
- messages.append({"role": "user", "content": h[0]})
13
- messages.append({"role": "assistant", "content": h[1]})
14
 
15
- messages.append({"role": "user", "content": user_input})
16
 
17
- response = client.chat.completions.create(
18
  model="gpt-4o-mini",
19
- messages=messages
20
  )
21
 
22
- reply = response.choices[0].message.content
23
- history.append((user_input, reply))
24
-
25
- return history, history
26
 
27
  except Exception as e:
28
- return str(e), history
29
-
30
 
31
  iface = gr.ChatInterface(fn=chat)
32
  iface.launch()
 
4
 
5
  client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
6
 
7
+ def chat(message, history):
8
  try:
9
  messages = [{"role": "system", "content": "You are Jarvis, a smart AI assistant."}]
10
 
11
+ for user, bot in history:
12
+ messages.append({"role": "user", "content": user})
13
+ messages.append({"role": "assistant", "content": bot})
14
 
15
+ messages.append({"role": "user", "content": message})
16
 
17
+ response = client.responses.create(
18
  model="gpt-4o-mini",
19
+ input=messages
20
  )
21
 
22
+ reply = response.output_text
23
+ return reply
 
 
24
 
25
  except Exception as e:
26
+ return f"Error: {str(e)}"
 
27
 
28
  iface = gr.ChatInterface(fn=chat)
29
  iface.launch()