Tobias Geisler commited on
Commit
43f3a3c
·
1 Parent(s): ab638cf

fix chat history display

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -30,12 +30,17 @@ def chat_with_gpt(user_input, system_message, temperature, history):
30
  history.append({"role": "assistant", "content": assistant_message})
31
 
32
  # Convert history to the format expected by gr.Chatbot
33
- # Note: Gradio Chatbot automatically aligns user messages to the right and assistant messages to the left
34
- formatted_history = [(msg["content"], "user" if msg["role"] == "user" else "assistant") for msg in history]
 
 
 
 
35
 
36
  return formatted_history, history
37
 
38
 
 
39
  def reset_history(system_message):
40
  # Return an empty conversation history
41
  return [], []
 
30
  history.append({"role": "assistant", "content": assistant_message})
31
 
32
  # Convert history to the format expected by gr.Chatbot
33
+ formatted_history = []
34
+ for msg in history:
35
+ if msg["role"] == "user":
36
+ formatted_history.append([msg["content"], None])
37
+ else: # msg["role"] == "assistant"
38
+ formatted_history.append([None, msg["content"]])
39
 
40
  return formatted_history, history
41
 
42
 
43
+
44
  def reset_history(system_message):
45
  # Return an empty conversation history
46
  return [], []