akhatr-phyniks commited on
Commit
27ad2f6
·
verified ·
1 Parent(s): 2ed4440

fixes chat.

Browse files
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -136,9 +136,8 @@ Write like the stakes are real. Because they are.
136
  ]
137
 
138
  # Add chat history
139
- for user_msg, assistant_msg in chat_history:
140
- messages.append({"role": "user", "content": user_msg})
141
- messages.append({"role": "assistant", "content": assistant_msg})
142
 
143
  # Add current message
144
  messages.append({"role": "user", "content": message})
@@ -146,12 +145,17 @@ Write like the stakes are real. Because they are.
146
  # Get response from OpenAI with Langfuse tracking
147
  assistant_message = get_completion(messages)
148
 
149
- chat_history.append((message, assistant_message))
 
 
 
150
  return chat_history, ""
151
 
152
  except Exception as e:
153
  error_message = f"Error: {str(e)}"
154
- return chat_history + [(message, error_message)], ""
 
 
155
 
156
  # Create Gradio interface
157
  with gr.Blocks() as demo:
 
136
  ]
137
 
138
  # Add chat history
139
+ for msg in chat_history:
140
+ messages.append(msg)
 
141
 
142
  # Add current message
143
  messages.append({"role": "user", "content": message})
 
145
  # Get response from OpenAI with Langfuse tracking
146
  assistant_message = get_completion(messages)
147
 
148
+ # Update chat history with new messages
149
+ chat_history.append({"role": "user", "content": message})
150
+ chat_history.append({"role": "assistant", "content": assistant_message})
151
+
152
  return chat_history, ""
153
 
154
  except Exception as e:
155
  error_message = f"Error: {str(e)}"
156
+ chat_history.append({"role": "user", "content": message})
157
+ chat_history.append({"role": "assistant", "content": error_message})
158
+ return chat_history, ""
159
 
160
  # Create Gradio interface
161
  with gr.Blocks() as demo: