jjii commited on
Commit
2dc4133
·
1 Parent(s): 723d86f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -20,7 +20,7 @@ with gr.Blocks() as demo:
20
 
21
  def respond(message, chat_history):
22
  messages.append({"role": "user", "content": message})
23
- response = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=messages)
24
  system_message = response["choices"][0]["message"]
25
  messages.append(system_message)
26
  bot_message = system_message['content']
@@ -29,6 +29,13 @@ with gr.Blocks() as demo:
29
  time.sleep(1)
30
  return "", chat_history
31
 
 
 
 
 
 
 
 
32
  msg.submit(respond, [msg, chatbot], [msg, chatbot])
33
  clear.click(lambda: None, None, chatbot, queue=False)
34
 
 
20
 
21
  def respond(message, chat_history):
22
  messages.append({"role": "user", "content": message})
23
+ response = openai.ChatCompletion.create(model="gpt-4-0613", messages=messages)
24
  system_message = response["choices"][0]["message"]
25
  messages.append(system_message)
26
  bot_message = system_message['content']
 
29
  time.sleep(1)
30
  return "", chat_history
31
 
32
+ chat_transcript = ""
33
+ for message in messages:
34
+ if message['role'] != 'system':
35
+ chat_transcript += message['role'] + ": " + message['content'] + "\n\n"
36
+
37
+ return chat_transcript
38
+
39
  msg.submit(respond, [msg, chatbot], [msg, chatbot])
40
  clear.click(lambda: None, None, chatbot, queue=False)
41