Create app_bak2.py

app_bak2.py

@@ -0,0 +1,110 @@
+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
+        )
+        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()