arubaDev commited on
Commit
e70c62f
·
verified ·
1 Parent(s): d53da60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -177,7 +177,7 @@ def delete_chat_cb(selected_label):
177
  selected = labels[0] if labels else None
178
  return gr.update(choices=labels, value=selected), []
179
 
180
- def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice):
181
  sid = label_to_id(selected_label)
182
  if sid is None:
183
  sid = create_session("New chat")
@@ -188,9 +188,9 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
188
  update_session_title_if_needed(sid, user_text)
189
 
190
  api_messages = build_api_messages(sid, system_message)
191
- display_msgs = chatbot_msgs[:] # Use existing chat content
192
 
193
- # Add initial assistant message for typing
194
  display_msgs.append({"role": "assistant", "content": "…"})
195
  yield (display_msgs, "", selected_label)
196
 
@@ -217,6 +217,7 @@ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens,
217
 
218
 
219
 
 
220
  def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice):
221
  sid = label_to_id(selected_label)
222
  if sid is None:
 
177
  selected = labels[0] if labels else None
178
  return gr.update(choices=labels, value=selected), []
179
 
180
+ def send_cb(user_text, selected_label, chatbot_msgs, system_message, max_tokens, temperature, top_p, model_choice, *args):
181
  sid = label_to_id(selected_label)
182
  if sid is None:
183
  sid = create_session("New chat")
 
188
  update_session_title_if_needed(sid, user_text)
189
 
190
  api_messages = build_api_messages(sid, system_message)
191
+ display_msgs = chatbot_msgs[:] # preserve current chat
192
 
193
+ # Initial typing message
194
  display_msgs.append({"role": "assistant", "content": "…"})
195
  yield (display_msgs, "", selected_label)
196
 
 
217
 
218
 
219
 
220
+
221
  def regenerate_cb(selected_label, system_message, max_tokens, temperature, top_p, model_choice, dataset_choice):
222
  sid = label_to_id(selected_label)
223
  if sid is None: