Update app.py

app.py

@@ -203,7 +203,7 @@ def battle_arena(prompt):
     response_a, response_b, model_a, model_b = generate_responses(prompt)
     
     # Check for API errors in responses
-    if any("Error: Unable to get response from the model" in msg["content"] 
+    if any("Error: Unable to get response from the model" in msg["content"]
            for msg in response_a + response_b 
            if msg["role"] == "assistant"):
         return (
@@ -394,14 +394,13 @@ def continue_conversation(prompt, left_chat, right_chat, left_model, right_model
     if not prompt or prompt == previous_prompt:
         prompt = random.choice(config.example_prompts)
     
+    # Get responses (which are lists of messages)
     left_response = call_ollama_api(left_model, prompt)
     right_response = call_ollama_api(right_model, prompt)
     
-    left_chat.append({"role": "user", "content": prompt})
-    left_chat.append({"role": "assistant", "content": left_response})
-    
-    right_chat.append({"role": "user", "content": prompt})
-    right_chat.append({"role": "assistant", "content": right_response})
+    # Append messages from the response lists
+    left_chat.extend(left_response)
+    right_chat.extend(right_response)
     
     tie_count += 1
     tie_button_state = gr.update(interactive=True) if tie_count < 3 else gr.update(interactive=False, value="Max ties reached. Please vote!")