Update app.py

app.py

@@ -2,12 +2,18 @@ import gradio as gr
 import dashscope
 import os
 
-# 导入自定义模块
+# ===== 你的自定义模块 =====
 from config import API_KEY, APP_TITLE, CHATBOT_HEIGHT, MIN_WIDTH_LEFT, MIN_WIDTH_RIGHT
 from styles import POKER_THEME_CSS
 from ai_service import design_poker_game
+# 优先尝试原生流式；没有则自动回退为伪流式
+try:
+    from ai_service import design_poker_game_stream
+except Exception:
+    design_poker_game_stream = None
+
 
-# ====== 🔧 Hotfix: 兼容 gradio_client 对 additionalProperties 为布尔的解析 ======
+# ====== 🔧 Hotfix: 兼容 gradio_client 对 additionalProperties(bool) 的解析 ======
 try:
     import gradio_client.utils as _gc_utils
 
@@ -40,8 +46,9 @@ except Exception:
     pass
 # ====== 🔧 Hotfix 结束 ======
 
-# 设置 API Key
-dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") 
+# ====== 设置 API Key（环境变量优先，兜底 config.API_KEY）======
+dashscope.api_key = os.getenv("DASHSCOPE_API_KEY") or os.getenv("API_KEY") or API_KEY
+
 
 # ==================== 工具函数 ====================
 def _messages_to_tuples(history):
@@ -52,7 +59,6 @@ def _messages_to_tuples(history):
     if not history:
         return []
 
-    # 已经是列表且第一个元素为 dict，则认为是 messages 模式
     if isinstance(history, list) and history and isinstance(history[0], dict):
         pairs = []
         last_user = None
@@ -64,14 +70,18 @@ def _messages_to_tuples(history):
             elif role == "assistant":
                 pairs.append((last_user or "", content))
                 last_user = None
-            else:
-                # 其他角色（如 system）直接忽略到导出/兼容对里
-                pass
         return pairs
 
-    # 否则当作老的 [(user, bot)] 结构
     return history
 
+
+def _chunk_fake_stream(text, step=40):
+    """把整段文本切成小块，伪流式输出。"""
+    s = str(text or "")
+    for i in range(0, len(s), step):
+        yield s[i:i + step]
+
+
 def clear_cache():
     """清空缓存"""
     from cache_manager import file_cache, request_cache
@@ -79,10 +89,12 @@ def clear_cache():
     request_cache.clear()
     return "✅ 缓存已清空"
 
+
 def clear_files():
     """清空文件上传"""
     return None
 
+
 def update_file_status(files):
     """更新文件状态显示"""
     if not files:
@@ -90,23 +102,21 @@ def update_file_status(files):
     file_list = files if isinstance(files, (list, tuple)) else [files]
     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 文件，并返回文件路径供下载"""
     import time
     import pathlib
     from datetime import datetime
 
-    # 统一转为 [(user, bot)] 再导出
     pairs = _messages_to_tuples(history)
 
     lines = ["# 对话记录", ""]
@@ -132,15 +142,6 @@ 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(
@@ -194,36 +195,27 @@ with gr.Blocks(
                     "- 长输出建议分步生成（先 GDL，再自然语言规则说明）。\n"
                     "- 需要更强制约，可在 Prompt 中明确输出格式与边界。"
                 )
-                
+
             with gr.Group(elem_classes="side-card"):
                 gr.Markdown("### 快捷操作")
                 with gr.Row():
                     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")
 
-        # 右侧：聊天区
-        # 右侧：聊天区（改为手动事件绑定，彻底规避 ChatInterface 的样式/版本差异）
+        # 右侧：聊天区（手动事件绑定 + 流式输出）
         with gr.Column(scale=2, min_width=MIN_WIDTH_RIGHT):
             with gr.Group(elem_classes="table"):
-                # 聊天消息采用新格式 messages
                 chatbot = gr.Chatbot(
                     height=CHATBOT_HEIGHT,
                     type="messages",
                     elem_classes="custom-chatbot",
-                    #elem_id="poker_chat",
                     avatar_images=("landlord.png", "bot.png"),
                 )
-                # 用一个 State 保存 messages 历史，便于多处复用
-                chat_state = gr.State([])  # list[dict]: [{"role":"user","content":...}, {"role":"assistant","content":...}, ...]
+                # 用 State 保存 messages 历史
+                chat_state = gr.State([])  # list[dict]: [{"role":"user","content":...}, {"role":"assistant","content":...}]
 
-                # 输入框（支持 Shift+Enter 换行；回车提交我们手动绑定）
                 user_input = gr.Textbox(
                     placeholder="例如：设计一个适合3-5人的派对风格扑克游戏（请写清目标人群/时长/创新点）…",
                     show_label=False,
@@ -233,48 +225,57 @@ with gr.Blocks(
                     autofocus=True,
                 )
 
-                # 显式“发送”按钮（大按钮）
                 with gr.Row():
                     send_btn = gr.Button("发送", variant="primary")
-                    stop_info = gr.Markdown("", visible=False)  # 预留：如需做流式/停止提示
-
-                # —— 提交处理：把 messages 历史交给你的 design_poker_game，返回助手回复并更新 UI ——
-                def _messages_to_tuples(history_msgs):
-                    """[{role,content}] -> [(user, bot)] 供现有 design_poker_game 使用"""
-                    pairs, last_user = [], None
-                    for m in history_msgs or []:
-                        r, c = m.get("role"), m.get("content", "")
-                        if r == "user":
-                            last_user = c
-                        elif r == "assistant":
-                            pairs.append((last_user or "", c))
-                            last_user = None
-                    return pairs
+                    stop_info = gr.Markdown("", visible=False)
 
+                # ====== 核心：流式提交回调（生成器） ======
                 def on_submit(user_text, history_msgs, files, custom_prompt, mode):
                     user_text = (user_text or "").strip()
-                    # 空文本不提交（与 Gradio 5 的默认交互一致）
                     if not user_text:
-                        return gr.update(), gr.update(), history_msgs
-
-                    # 1) 先把用户消息写入 messages（前端立即看到气泡）
-                    history_msgs = list(history_msgs or [])
-                    history_msgs.append({"role": "user", "content": user_text})
-
-                    # 2) 调用你的核心函数（传入 tuple 历史以兼容）
-                    tuples_hist = _messages_to_tuples(history_msgs)
-                    try:
-                        bot_reply = design_poker_game(user_text, tuples_hist, files, custom_prompt, mode)
-                    except Exception as e:
-                        bot_reply = f"（出错）{type(e).__name__}: {e}"
-
-                    # 3) 追加助手回复
-                    history_msgs.append({"role": "assistant", "content": str(bot_reply)})
-
-                    # 4) 更新 Chatbot 与清空输入框，同时回写 state
-                    return history_msgs, "", history_msgs
-
-                # 绑定：回车提交（Enter=提交；Shift+Enter=换行由浏览器原生处理）
+                        # 不提交空消息：输出不变
+                        yield gr.update(), gr.update(), history_msgs
+                        return
+
+                    # 1) 立即显示“用户消息”
+                    history = list(history_msgs or [])
+                    history.append({"role": "user", "content": user_text})
+                    yield history, "", history
+
+                    # 2) 追加一个“空的助手气泡”，用于逐步填充
+                    history.append({"role": "assistant", "content": ""})
+                    yield history, "", history
+
+                    # 3) 原生流式优先，否则退化为伪流式
+                    tuples_hist = _messages_to_tuples(history)
+
+                    if design_poker_game_stream is not None:
+                        try:
+                            for piece in design_poker_game_stream(
+                                user_text, tuples_hist, files, custom_prompt, mode
+                            ):
+                                if not piece:
+                                    continue
+                                history[-1]["content"] += str(piece)
+                                # 每片刷新一次 UI（需要可做节流：攒到 N 字再 yield）
+                                yield history, "", history
+                        except Exception as e:
+                            history[-1]["content"] += f"\n（流式出错）{type(e).__name__}: {e}"
+                            yield history, "", history
+                    else:
+                        # 伪流式：先算整段，再切片逐步输出
+                        try:
+                            full = design_poker_game(user_text, tuples_hist, files, custom_prompt, mode)
+                        except Exception as e:
+                            full = f"（出错）{type(e).__name__}: {e}"
+                        for piece in _chunk_fake_stream(str(full), step=40):
+                            history[-1]["content"] += piece
+                            yield history, "", history
+
+                    # 4) 收尾：再 yield 一次，确保最终内容同步
+                    yield history, "", history
+
+                # 绑定：回车提交（Enter=提交；Shift+Enter=换行由浏览器处理）
                 user_input.submit(
                     fn=on_submit,
                     inputs=[user_input, chat_state, file_uploader, custom_prompt_box, prompt_mode],
@@ -282,7 +283,7 @@ with gr.Blocks(
                     preprocess=True,
                 )
 
-                # 绑定：点击“发送”按钮提交（与回车完全等价）
+                # 绑定：点击“发送”
                 send_btn.click(
                     fn=on_submit,
                     inputs=[user_input, chat_state, file_uploader, custom_prompt_box, prompt_mode],
@@ -290,12 +291,12 @@ with gr.Blocks(
                     preprocess=True,
                 )
 
-                # 导出对话（直接读取 Chatbot 的 messages 值）
+                # 导出对话
                 with gr.Row():
                     export_btn = gr.Button("导出对话（Markdown）", variant="secondary")
                     export_file = gr.File(label="点击下载导出文件", interactive=False)
 
-                # 清空对话（同时清空 Chatbot + 输入 + state）
+                # 清空对话
                 with gr.Row():
                     clear_dialog_btn = gr.Button("清空对话", variant="secondary")
 
@@ -308,26 +309,12 @@ with gr.Blocks(
                     outputs=[chatbot, user_input, chat_state],
                 )
 
+    # ==================== 事件绑定（左侧） ====================
+    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, inputs=[file_uploader], outputs=[file_status])
 
-    # ==================== 事件绑定 ====================
-    # 清空缓存按钮
-    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,
-        inputs=[file_uploader],
-        outputs=[file_status]
-    )
-
-    # 导出历史对话
+    # 导出按钮
     def _export_wrapper(chat_history):
         try:
             path = export_history_to_markdown(chat_history)
@@ -336,16 +323,13 @@ with gr.Blocks(
             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")
+                f.write(f"导出失败：{type(e).__name__}: {e}\n")
             return tmp.name
 
-    export_btn.click(
-        fn=_export_wrapper,
-        inputs=[chatbot],
-        outputs=[export_file]
-    )
+    export_btn.click(fn=_export_wrapper, inputs=[chatbot], outputs=[export_file])
+
 
 # ==================== 启动应用 ====================
 if __name__ == "__main__":
-    # 在 Hugging Face Spaces 上 share 参数会被忽略，但本地调试可用
-    demo.queue().launch(share=True, show_api=False)
+    # 更顺滑的流式 & 更高并发
+    demo.queue(concurrency_count=4, status_update_rate=0.2).launch(share=True, show_api=False)