refactor: Enhance chatbot UI by updating comments, fixing clear logic, and ensuring message handling is consistent across models.

app.py

@@ -14,17 +14,15 @@ def bot_response(user_message, history, model_name, system_message, max_tokens,
     token = oauth_token.token if oauth_token else None
     client = InferenceClient(token=token, model=model_name)
     
-    # 1. 构造发送给 API 的消息序列
-    # 注意：history 在 Gradio 6 中已经是 [{"role": "...", "content": "..."}] 格式
+    # 1. Construct messages for API
     api_messages = [{"role": "system", "content": system_message}] + history + [{"role": "user", "content": user_message}]
     
-    # 2. 更新 UI 的历史记录
-    # 先添加用户的消息，再添加一个空的助手消息用于流式占位
+    # 2. Update UI history (Gradio 5/6 format)
     new_history = history + [
         {"role": "user", "content": user_message},
         {"role": "assistant", "content": ""}
     ]
-    yield new_history, "" # 立即更新 UI 显示用户消息
+    yield new_history, "" 
 
     try:
         response_text = ""
@@ -39,7 +37,6 @@ def bot_response(user_message, history, model_name, system_message, max_tokens,
         for chunk in stream:
             token_content = chunk.choices[0].delta.content or ""
             response_text += token_content
-            # 更新最后一条消息的内容
             new_history[-1]["content"] = response_text
             yield new_history, ""
             
@@ -59,33 +56,41 @@ with gr.Blocks() as demo:
     gr.Markdown("# ⚔️ 独立输入 Chatbot Arena")
 
     with gr.Row():
-        # --- 模型 A 列 ---
+        # --- Model A ---
         with gr.Column():
             model_a_name = gr.Dropdown(MODELS, label="选择模型 A", value=MODELS[0])
-            chatbot_a = gr.Chatbot(label="Model A 聊天窗口")
+            # Explicitly set type="messages"
+            chatbot_a = gr.Chatbot(label="Model A 聊天窗口", type="messages")
             msg_a = gr.Textbox(placeholder="给模型 A 发送消息...", label="模型 A 输入")
             btn_a = gr.Button("发送给 A")
 
-        # --- 模型 B 列 ---
+        # --- Model B ---
         with gr.Column():
             model_b_name = gr.Dropdown(MODELS, label="选择模型 B", value=MODELS[1])
-            chatbot_b = gr.Chatbot(label="Model B 聊天窗口")
+            # Explicitly set type="messages"
+            chatbot_b = gr.Chatbot(label="Model B 聊天窗口", type="messages")
             msg_b = gr.Textbox(placeholder="给模型 B 发送消息...", label="模型 B 输入")
             btn_b = gr.Button("发送给 B")
 
-    # --- 绑定事件 ---
+    # --- Bind Events ---
 
-    # 模型 A 的发送逻辑
     a_inputs = [msg_a, chatbot_a, model_a_name, system_msg, max_t, temp, top_p_val]
     msg_a.submit(bot_response, a_inputs, [chatbot_a, msg_a])
     btn_a.click(bot_response, a_inputs, [chatbot_a, msg_a])
 
-    # 模型 B 的发送逻辑
     b_inputs = [msg_b, chatbot_b, model_b_name, system_msg, max_t, temp, top_p_val]
     msg_b.submit(bot_response, b_inputs, [chatbot_b, msg_b])
     btn_b.click(bot_response, b_inputs, [chatbot_b, msg_b])
 
-    gr.Button("🗑️ 清空所有对话").click(lambda: ([], []), None, [chatbot_a, chatbot_b])
+    # Fix: Clear logic returning empty lists for both chatbots
+    def clear_chats():
+        return [], []
+
+    gr.Button("🗑️ 清空所有对话").click(
+        fn=clear_chats, 
+        inputs=None, 
+        outputs=[chatbot_a, chatbot_b]
+    )
 
 if __name__ == "__main__":
     demo.launch()