Update app.py

app.py

@@ -43,7 +43,35 @@ except Exception:
 # 设置 API Key
 dashscope.api_key = API_KEY
 
-# ==================== 事件处理函数 ====================
+# ==================== 工具函数 ====================
+def _messages_to_tuples(history):
+    """
+    将 Chatbot 的 messages 格式（[{role, content}, ...]）转换为 [(user, bot), ...]。
+    若已是 tuples，则原样返回。
+    """
+    if not history:
+        return []
+
+    # 已经是列表且第一个元素为 dict，则认为是 messages 模式
+    if isinstance(history, list) and history and isinstance(history[0], dict):
+        pairs = []
+        last_user = None
+        for msg in history:
+            role = msg.get("role")
+            content = msg.get("content", "")
+            if role == "user":
+                last_user = content
+            elif role == "assistant":
+                pairs.append((last_user or "", content))
+                last_user = None
+            else:
+                # 其他角色（如 system）直接忽略到导出/兼容对里
+                pass
+        return pairs
+
+    # 否则当作老的 [(user, bot)] 结构
+    return history
+
 def clear_cache():
     """清空缓存"""
     from cache_manager import file_cache, request_cache
@@ -59,10 +87,18 @@ def update_file_status(files):
     """更新文件状态显示"""
     if not files:
         return "📁 文件状态：未上传"
-    
     file_list = files if isinstance(files, (list, tuple)) else [files]
-    file_names = [os.path.basename(getattr(f, "name", str(f))) for f in file_list]
-    return f"📁 文件状态：已上传 {len(file_names)} 个文件\n" + "\n".join(f"  • {name}" for name in file_names[:3]) + ("\n  ..." if len(file_names) > 3 else "")
+    names = []
+    for f in file_list:
+        # gr.File(type="filepath") 传回的是路径字符串
+        if isinstance(f, str):
+            names.append(os.path.basename(f))
+        else:
+            # 兼容其他情况
+            names.append(os.path.basename(getattr(f, "name", str(f))))
+    head = "\n".join(f"  • {n}" for n in names[:3])
+    tail = "\n  ..." if len(names) > 3 else ""
+    return f"📁 文件状态：已上传 {len(names)} 个文件\n{head}{tail}"
 
 def export_history_to_markdown(history):
     """将聊天历史导出为 Markdown 文件，并返回文件路径供下载"""
@@ -70,21 +106,20 @@ def export_history_to_markdown(history):
     import pathlib
     from datetime import datetime
 
-    # history 为 [(user, bot), ...]
+    # 统一转为 [(user, bot)] 再导出
+    pairs = _messages_to_tuples(history)
+
     lines = ["# 对话记录", ""]
     lines.append(f"- 导出时间：{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
     lines.append("")
-    for idx, pair in enumerate(history or [], start=1):
-        user_msg, bot_msg = pair if isinstance(pair, (list, tuple)) and len(pair) == 2 else ("", "")
+    for idx, (user_msg, bot_msg) in enumerate(pairs, start=1):
         lines.append(f"## 轮次 {idx}")
         if user_msg:
-            lines.append("**用户**:")
-            lines.append("")
+            lines.append("**用户**:\n")
             lines.append(user_msg)
             lines.append("")
         if bot_msg:
-            lines.append("**助手**:")
-            lines.append("")
+            lines.append("**助手**:\n")
             lines.append(bot_msg)
             lines.append("")
 
@@ -97,6 +132,16 @@ def export_history_to_markdown(history):
         f.write(content)
     return str(filename)
 
+# 为 ChatInterface 提供一个向后兼容的适配器
+def chat_fn_adapter(user_message, history, files, custom_prompt, prompt_mode):
+    """
+    ChatInterface(type='messages') 会传入 history 为 messages 列表。
+    这里做一次转换，调用你现有的 design_poker_game（假设其使用 [(user, bot)] 历史）。
+    返回值保持为纯文本（由 ChatInterface 追加到对话）。
+    """
+    history_tuples = _messages_to_tuples(history)
+    return design_poker_game(user_message, history_tuples, files, custom_prompt, prompt_mode)
+
 # ==================== Gradio 界面（Poker Skin） ====================
 with gr.Blocks(
     theme=gr.themes.Soft(),
@@ -114,7 +159,7 @@ with gr.Blocks(
     """)
 
     with gr.Row(equal_height=True):
-        # 左侧：设置区（半透明卡片）
+        # 左侧：设置区
         with gr.Column(scale=1, min_width=MIN_WIDTH_LEFT):
             with gr.Group(elem_classes="side-card"):
                 gr.Markdown("### 输入与约束")
@@ -153,64 +198,70 @@ with gr.Blocks(
             with gr.Group(elem_classes="side-card"):
                 gr.Markdown("### 快捷操作")
                 with gr.Row():
-                    clear_cache_btn = gr.Button("清空缓存", variant="secondary", size="sm")
-                    clear_files_btn = gr.Button("清空文件", variant="secondary", size="sm")
-                
+                    clear_cache_btn = gr.Button("清空缓存", variant="secondary")
+                    clear_files_btn = gr.Button("清空文件", variant="secondary")
+                    # “清空对话”使用 ClearButton 以兼容新版
+                    # 先在右侧定义完 chatbot / user_input 后再绑定（见下文）
+
                 # 缓存状态显示
                 cache_status = gr.Markdown("💾 缓存状态：正常", elem_classes="hint")
-                
                 # 文件上传状态
                 file_status = gr.Markdown("📁 文件状态：未上传", elem_classes="hint")
 
-        # 右侧：聊天区（放在"扑克桌"里）
+        # 右侧：聊天区
         with gr.Column(scale=2, min_width=MIN_WIDTH_RIGHT):
             with gr.Group(elem_classes="table"):
-                # 单独定义 Chatbot，便于导出历史消息
+                # 单独定义 Chatbot（新版使用 messages 格式）
                 chatbot = gr.Chatbot(
                     height=CHATBOT_HEIGHT,
-                    bubble_full_width=True,
+                    type="messages",                 # ✅ 新格式，避免弃用警告
                     elem_classes="custom-chatbot",
                     avatar_images=("landlord.png", "bot.png")
                 )
 
+                user_input = gr.Textbox(
+                    placeholder="例如：设计一个适合3-5人的派对风格扑克游戏（请写清目标人群/时长/创新点）…",
+                    show_label=False,
+                    max_lines=5,
+                    lines=3,
+                    show_copy_button=True
+                )
+
+                # ChatInterface：去掉已移除的 retry_btn/undo_btn/clear_btn
                 chat = gr.ChatInterface(
-                    fn=design_poker_game,
-                    additional_inputs=[file_uploader, custom_prompt_box, prompt_mode],
+                    fn=chat_fn_adapter,
                     chatbot=chatbot,
-                    textbox=gr.Textbox(
-                        placeholder="例如：设计一个适合3-5人的派对风格扑克游戏（请写清目标人群/时长/创新点）…",
-                        show_label=False,
-                        max_lines=5,
-                        lines=3,
-                        show_copy_button=True
-                    ),
+                    textbox=user_input,
+                    additional_inputs=[file_uploader, custom_prompt_box, prompt_mode],
+                    type="messages",                # ✅ 与 Chatbot 保持一致
                     submit_btn="发送",
-                    retry_btn="重新生成",
-                    undo_btn="撤销",
-                    clear_btn="清空对话",
-                    description="""
-                    **使用方式**  
-                    左侧上传 GDL 与（可选）自定义 Prompt，然后在这里输入需求。我会遵循上传的规范，并结合你的目标人群/复杂度/时长做设计。
-                    """
+                    stop_btn=True,
+                    description=(
+                        "**使用方式**  \n"
+                        "左侧上传 GDL 与（可选）自定义 Prompt，然后在这里输入需求。"
+                        "我会遵循上传的规范，并结合你的目标人群/复杂度/时长做设计。"
+                    ),
                 )
 
                 with gr.Row():
                     export_btn = gr.Button("导出对话（Markdown）", variant="secondary")
                     export_file = gr.File(label="点击下载导出文件", interactive=False)
 
+                # 在这里补上“清空对话”按钮（等同于你以前的 clear_btn）
+                with gr.Row():
+                    clear_dialog_btn = gr.ClearButton([chatbot, user_input], value="清空对话", variant="secondary")
+
     # ==================== 事件绑定 ====================
     # 清空缓存按钮
     clear_cache_btn.click(
         fn=clear_cache,
         outputs=[cache_status]
     )
-    
     # 清空文件按钮
     clear_files_btn.click(
         fn=clear_files,
         outputs=[file_uploader]
     )
-    
     # 文件上传状态更新
     file_uploader.change(
         fn=update_file_status,
@@ -218,14 +269,13 @@ with gr.Blocks(
         outputs=[file_status]
     )
 
-    # 导出历史对话（从 Chatbot 读取 state）
+    # 导出历史对话
     def _export_wrapper(chat_history):
         try:
             path = export_history_to_markdown(chat_history)
             return path
         except Exception as e:
-            import tempfile, os
-            # 导出失败则写入一个错误说明文件
+            import tempfile
             tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".md")
             with open(tmp.name, 'w', encoding='utf-8') as f:
                 f.write(f"导出失败：{type(e).__name__}: {str(e)}\n")
@@ -239,4 +289,5 @@ with gr.Blocks(
 
 # ==================== 启动应用 ====================
 if __name__ == "__main__":
-    demo.launch(share=True, show_api=False)
+    # 在 Hugging Face Spaces 上 share 参数会被忽略，但本地调试可用
+    demo.queue().launch(share=True, show_api=False)