Update app.py

app.py

@@ -1,112 +1,79 @@
 import gradio as gr
 import requests
-import time
 import json
-
-
-# 设置 FastGPT API 的基本信息
-#API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
-#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
-
-
-# 设置 FastGPT API 的基本信息
-#API_URL = "https://api.fastgpt.com/generate"  # 请根据您的实际部署地址修改
-#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
-
-#import gradio as gr
-#import requests
-
-#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
-
-#API_URL = "https://api.fastgpt.com/v1/chat"
-
-
-# 替换成你的 FastGPT 的 BaseURL 和 应用特定 API Key
-#BASE_URL = "https://api.fastgpt.in/api/v1"  
-#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 应用特定 key
-
-# 替换成你的 FastGPT 的 BaseURL 和 应用特定 API Key
-BASE_URL = "https://api.fastgpt.in/api/v1"
-API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 应用特定 key
-
-
-import ijson  # Install with: pip install ijson
-
-
-def chat_with_fastgpt(message, history):
-    """
-    与 FastGPT 进行对话的函数
-
-    Args:
-        message: 用户输入的消息
-        history: 聊天历史记录
-
-    Returns:
-        生成器的响应内容
-    """
-    headers = {
-        "Authorization": f"Bearer {API_KEY}",
-        "Content-Type": "application/json",
-    }
-
-    # 构建消息历史
-    messages = []
-    for human, assistant in history:
-        messages.append({"role": "user", "content": human})
-        messages.append({"role": "assistant", "content": assistant})
-    messages.append({"role": "user", "content": message})
-
-    data = {
-        "chatId": "your_chat_id",  # 如果需要，替换成你的 chatId
-        "stream": True,  # 仍然使用流式请求
-        "detail": False,
-        "messages": messages,
-    }
-
-    try:
-        response = requests.post(
-            f"{BASE_URL}/chat/completions", headers=headers, json=data, stream=True
+import uuid
+
+class FastGPTChat:
+    def __init__(self, api_key, base_url="https://api.fastgpt.in/api"):
+        self.api_key = api_key
+        self.base_url = base_url
+        self.headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json"
+        }
+        
+    def chat(self, message, chat_history):
+        # 生成唯一的聊天ID
+        chat_id = str(uuid.uuid4())
+        
+        # 准备请求数据
+        data = {
+            "chatId": chat_id,
+            "stream": False,
+            "detail": False,
+            "messages": [
+                {
+                    "role": "user",
+                    "content": message
+                }
+            ]
+        }
+        
+        try:
+            # 发送请求
+            response = requests.post(
+                f"{self.base_url}/v1/chat/completions",
+                headers=self.headers,
+                json=data
+            )
+            response.raise_for_status()
+            
+            # 解析响应
+            response_data = response.json()
+            ai_message = response_data["choices"][0]["message"]["content"]
+            
+            # 更新对话历史
+            chat_history.append((message, ai_message))
+            return "", chat_history
+            
+        except Exception as e:
+            return f"Error: {str(e)}", chat_history
+
+def create_chat_interface(api_key):
+    # 创建FastGPT聊天实例
+    fastgpt_chat = FastGPTChat(api_key)
+    
+    # 创建Gradio界面
+    with gr.Blocks(title="FastGPT Chat") as interface:
+        gr.Markdown("# FastGPT Chat Interface")
+        
+        chatbot = gr.Chatbot(height=400)
+        message = gr.Textbox(label="Type your message here...", placeholder="Enter your message and press enter")
+        clear = gr.Button("Clear Chat")
+        
+        # 设置事件处理
+        message.submit(
+            fastgpt_chat.chat,
+            inputs=[message, chatbot],
+            outputs=[message, chatbot]
         )
-        response.raise_for_status()
-
-        buffer = ""
-        for chunk in response.iter_content(chunk_size=8192):
-            if chunk:
-                buffer += chunk.decode('utf-8')
-                
-                # 尝试解析 JSON 对象
-                while True:  # Continue processing the buffer until no more complete objects are found
-                    try:
-                        first_brace_index = buffer.find('{')
-                        if first_brace_index == -1:
-                            break
-                        
-                        obj, end_index = json.JSONDecoder().raw_decode(buffer[first_brace_index:])
-                        
-                        # 解析成功，处理 JSON 对象
-                        if obj.get("choices"):
-                            delta = obj["choices"][0].get("delta")
-                            if delta and delta.get("content"):
-                                yield delta["content"]
-                        
-                        buffer = buffer[first_brace_index + end_index:]
-                    
-                    except json.JSONDecodeError as e:
-                        break
-
-    except requests.exceptions.RequestException as e:
-        print(f"请求错误: {e}")
-        yield "请求 FastGPT API 出错，请检查网络或 API Key。"
-    except Exception as e:
-        print(f"未知错误: {e}")
-        print(f"错误信息: {e}")
-        yield "发生未知错误。"
-
-# Gradio 界面
-iface = gr.ChatInterface(
-    fn=chat_with_fastgpt,
-    title="FastGPT Chatbot",
-    description="与 FastGPT 支持的聊天机器人对话",
-)
-
-iface.launch()
+        
+        clear.click(lambda: None, None, chatbot, queue=False)
+        
+    return interface
+
+# 使用示例
+if __name__ == "__main__":
+    API_KEY = "your-fastgpt-api-key-here"  # 替换为你的API密钥
+    demo = create_chat_interface(API_KEY)
+    demo.launch()