update

.gradio/flagged/dataset1.csv

@@ -0,0 +1,3 @@
+Query,Generated Words,timestamp
+give me some english words,"成功生成 141 个英文单词: ['🚀 开始生成200个英文单词...\n', 'the', 'be', 'to', 'of', 'and', 'a', 'in', 'that', 'have', 'I', 'it', 'for', 'not', 'on', 'with', 'he', 'as', 'you', 'do', 'at', 'this', 'but', 'his', 'by', 'from', 'they', 'we', 'say', 'her', 'she', 'or', 'an', 'will', 'my', 'one', 'all', 'would', 'there', 'their', 'what', 'so', 'up', 'out', 'if', 'about', 'who', 'get', 'which', 'go', 'me', 'when', 'make', 'can', 'like', 'time', 'no', 'just', 'him', 'know', 'take', 'people', 'into', 'year', 'your', 'good', 'some', 'could', 'them', 'see', 'other', 'than', 'then', 'now', 'look', 'only', 'come', 'its', 'over', 'think', 'also', 'back', 'after', 'use', 'two', 'how', 'our', 'work', 'first', 'well', 'way', 'even', 'new', 'want', 'because', 'any', 'these', 'give', 'day', 'most', 'us', 'very', 'life', 'after', 'call', 'world', 'over', 'still', 'take', 'every', 'through', 'before', 'long', 'where', 'much', 'should', 'well', 'people', 'down', 'own', 'work', 'first', 'good', 'new', 'write', 'our', 'used', 'me', 'man', 'too', 'any', 'day', 'same', 'right', 'look', 'think', 'also', 'around', 'another', 'came', '\n✅ 200个英文单词生成完成！']",2025-08-26 11:09:12.950476
+give me some english words,"成功生成 141 个英文单词: ['🚀 开始生成200个英文单词...\n', 'the', 'be', 'to', 'of', 'and', 'a', 'in', 'that', 'have', 'I', 'it', 'for', 'not', 'on', 'with', 'he', 'as', 'you', 'do', 'at', 'this', 'but', 'his', 'by', 'from', 'they', 'we', 'say', 'her', 'she', 'or', 'an', 'will', 'my', 'one', 'all', 'would', 'there', 'their', 'what', 'so', 'up', 'out', 'if', 'about', 'who', 'get', 'which', 'go', 'me', 'when', 'make', 'can', 'like', 'time', 'no', 'just', 'him', 'know', 'take', 'people', 'into', 'year', 'your', 'good', 'some', 'could', 'them', 'see', 'other', 'than', 'then', 'now', 'look', 'only', 'come', 'its', 'over', 'think', 'also', 'back', 'after', 'use', 'two', 'how', 'our', 'work', 'first', 'well', 'way', 'even', 'new', 'want', 'because', 'any', 'these', 'give', 'day', 'most', 'us', 'very', 'life', 'after', 'call', 'world', 'over', 'still', 'take', 'every', 'through', 'before', 'long', 'where', 'much', 'should', 'well', 'people', 'down', 'own', 'work', 'first', 'good', 'new', 'write', 'our', 'used', 'me', 'man', 'too', 'any', 'day', 'same', 'right', 'look', 'think', 'also', 'around', 'another', 'came', '\n✅ 200个英文单词生成完成！']",2025-08-26 11:09:20.766123

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 import asyncio
 import json
@@ -40,37 +41,35 @@ async def generate_words_stream(query: str) -> AsyncGenerator[str, None]:
     yield f"\n✅ 200个英文单词生成完成！"
 
 def generate_words_mcp(query: str = "give me some english words") -> str:
-    """MCP工具包装函数：流式获取英文单词，返回结果给MCP客户端"""
+    """
+    Generate 200 English words based on a query.
+    
+    Args:
+        query (str): The input query for generating words. Default is "give me some english words".
+    
+    Returns:
+        str: A message indicating the number of words generated and the complete list of words.
+    """
     try:
-        # 调用原始函数并收集结果
-        result = []
-        # 注意：这里需要处理异步函数
-        async def collect_tokens():
-            async for token in generate_words_stream(query):
-                if token.strip():
-                    result.append(token)
-            return result
-        
-        # 运行异步函数
-        loop = asyncio.new_event_loop()
-        asyncio.set_event_loop(loop)
-        tokens = loop.run_until_complete(collect_tokens())
-        loop.close()
-        
-        # 返回结果给MCP客户端 - 返回全部单词
-        return f"成功生成 {len(tokens)} 个英文单词: {tokens}"
+        # 直接返回200个英文单词，简化MCP函数
+        words_text = " ".join(ENGLISH_WORDS)
+        return f"成功生成 {len(ENGLISH_WORDS)} 个英文单词: {words_text}"
     except Exception as e:
         return f"生成单词时出错: {str(e)}"
 
-async def generate_words_mcp_stream(query: str = "give me some english words") -> AsyncGenerator[str, None]:
-    """MCP工具：流式返回英文单词"""
-    try:
-        async for token in generate_words_stream(query):
-            if token.strip():
-                yield token
-    except Exception as e:
-        yield f"生成单词时出错: {str(e)}"
+def get_word_count(query: str = "count words") -> str:
+    """
+    Get the count of English words available.
+    
+    Args:
+        query (str): The input query. Default is "count words".
+    
+    Returns:
+        str: The total number of English words available.
+    """
+    return f"总共有 {len(ENGLISH_WORDS)} 个英文单词可用"
 
+# 创建对话界面
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🧠 English Words Generator")
     gr.Markdown("输入你的查询，AI 将为你生成 200 个英文单词（流式输出）")
@@ -94,11 +93,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             send_btn = gr.Button("发送", variant="primary", size="lg")
         with gr.Column(scale=1):
             clear_btn = gr.Button("清空", variant="secondary")
+        with gr.Column(scale=1):
+            download_btn = gr.Button("下载", variant="secondary")
 
     def handle_user_input(query, messages):
         """添加用户消息"""
         if query.strip():
-            new_messages = messages + [{"role": "user", "content": query}]
+            # 使用正确的格式：[user_message, bot_message]
+            new_messages = messages + [[query, None]]
             return "", new_messages
         return query, messages
 
@@ -106,24 +108,58 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         """清空对话"""
         return []
 
+    def download_chat(messages):
+        """下载对话记录为文本文件"""
+        if not messages:
+            return None
+        
+        # 生成对话内容
+        from datetime import datetime
+        current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        
+        chat_content = "# 英文单词生成对话记录\n\n"
+        chat_content += f"导出时间: {current_time}\n"
+        chat_content += f"对话条数: {len(messages)}\n\n"
+        chat_content += "=" * 50 + "\n\n"
+        
+        for i, message in enumerate(messages, 1):
+            if len(message) == 2:
+                user_msg, bot_msg = message
+                chat_content += f"## 对话 {i}\n\n"
+                chat_content += f"**用户**: {user_msg}\n\n"
+                if bot_msg:
+                    chat_content += f"**AI**: {bot_msg}\n\n"
+                else:
+                    chat_content += f"**AI**: 正在生成中...\n\n"
+                chat_content += "-" * 30 + "\n\n"
+        
+        # 创建临时文件
+        import tempfile
+        
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        
+        # 创建临时文件
+        temp_file = tempfile.NamedTemporaryFile(mode='w', delete=False, suffix='.txt', prefix='chat_')
+        temp_file.write(chat_content)
+        temp_file.close()
+        
+        return temp_file.name
+
     async def generate_response(messages):
         """根据用户最后一条消息流式生成 AI 回复"""
-        if not messages or messages[-1]["role"] != "user":
+        if not messages or not messages[-1] or len(messages[-1]) != 2:
             yield messages
             return
 
-        last_user_msg = messages[-1]["content"]
-
-        # 添加 AI 回复占位符
-        messages.append({"role": "assistant", "content": "正在生成单词..."})
-        yield messages
+        last_user_msg = messages[-1][0]  # 获取用户消息
 
         # 流式拼接回复
         full_response = ""
         async for token in generate_words_stream(last_user_msg):
             if token.strip():
                 full_response += " " + token
-                messages[-1] = {"role": "assistant", "content": full_response.strip()}
+                # 更新最后一条消息的bot回复部分
+                messages[-1][1] = full_response.strip()
                 yield messages
 
     # 用户提交时，先加消息 → 再生成回复
@@ -134,12 +170,10 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             .then(generate_response, inputs=chatbot, outputs=chatbot)
 
     clear_btn.click(clear_chat, outputs=chatbot)
+    
+    # 下载按钮事件
+    download_btn.click(download_chat, inputs=chatbot, outputs=gr.File(label="下载对话记录"))
 
-    # 注册MCP工具 - 使用包装函数并正确配置输出
-    demo.load(generate_words_mcp, inputs=None, outputs=gr.Textbox(label="MCP结果", visible=False))
-    # 注册流式MCP工具
-    demo.load(generate_words_mcp_stream, inputs=None, outputs=gr.Textbox(label="流式MCP结果", visible=False))
-
-# 开启 MCP server
+# 启动应用
 if __name__ == "__main__":
     demo.launch(mcp_server=True)

requirements.txt

@@ -1 +1 @@
-gradio[mcp]>=4.44.1
+gradio[mcp]>=4.43.1