Update app.py

app.py

@@ -90,9 +90,10 @@ def compare_models(
     max_new_tokens: int,
     temperature: float,
     top_p: float,
-) -> Tuple[str, str, List[Tuple[str, str]], List[Tuple[str, str]]]:
+) -> Tuple[List[Tuple[str, str]], List[Tuple[str, str]], List[Tuple[str, str]], List[Tuple[str, str]]]:
     if model1_name == model2_name:
-        return "Error: Please select two different models.", "Error: Please select two different models.", chat_history1, chat_history2
+        error_message = [("System", "Error: Please select two different models.")]
+        return error_message, error_message, chat_history1, chat_history2
 
     output1 = "".join(list(generate(model1_name, message, chat_history1, max_new_tokens, temperature, top_p)))
     output2 = "".join(list(generate(model2_name, message, chat_history2, max_new_tokens, temperature, top_p)))
@@ -102,7 +103,7 @@ def compare_models(
 
     log_results(model1_name, model2_name, message, output1, output2)
 
-    return output1, output2, chat_history1, chat_history2
+    return chat_history1, chat_history2, chat_history1, chat_history2
 
 def log_results(model1_name: str, model2_name: str, question: str, answer1: str, answer2: str, winner: str = None):
     log_data = {
@@ -165,4 +166,4 @@ with gr.Blocks(css="style.css") as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=10).launch()
+    demo.queue(max_size=10).launch()