Update app.py

app.py

@@ -3,46 +3,42 @@ import requests
 import time
 
 # 设置 FastGPT API 的基本信息
-API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
-API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
+#API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
+#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
 
 
 # 设置 FastGPT API 的基本信息
-#API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
-#API_KEY = "您的_API_密钥"  # 请替换为您的实际 API 密钥
+API_URL = "https://api.fastgpt.com/generate"  # 请根据您的实际部署地址修改
+API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
 
+# 调用 FastGPT API 的函数
 def chat_with_fastgpt(user_input, chat_history):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
-    data = {
-        "stream": True,
-        "chatId": "unique_chat_id",  # 请替换为唯一的聊天 ID
-        "messages": [{"role": "user", "content": user_input}]
-    }
-    response = requests.post(API_URL, headers=headers, json=data, stream=True)
+    max_tokens = 50  # 生成文本的最大长度
+    temperature = 0.7  # 控制生成文本的随机性程度
+    
+    # 构造API请求URL
+    url = f'{API_URL}?prompt={user_input}&max_tokens={max_tokens}&temperature={temperature}'
+    
+    response = requests.get(url, headers=headers)
+    
     if response.status_code != 200:
         return f"请求失败，状态码：{response.status_code}", chat_history
 
-    bot_response = ""
-    for chunk in response.iter_lines():
-        if chunk:
-            try:
-                message = chunk.decode('utf-8')
-                if message:
-                    bot_response += message
-                    chat_history[-1][1] = bot_response
-                    yield "", chat_history
-            except Exception as e:
-                print(f"解析响应时出错: {e}")
-    chat_history[-1][1] = bot_response
-    yield "", chat_history
+    # 解析并返回生成的文本
+    generated_text = response.json().get('generated_text', '')
+    chat_history.append([user_input, generated_text])
+    return "", chat_history
 
+# 添加用户输入的处理逻辑
 def user(user_input, chat_history):
     chat_history.append([user_input, None])
     return "", chat_history
 
+# 创建 Gradio 界面
 with gr.Blocks() as demo:
     chatbot = gr.Chatbot()
     msg = gr.Textbox(placeholder="请输入您的消息...")
@@ -54,4 +50,5 @@ with gr.Blocks() as demo:
     )
     clear.click(lambda: None, None, chatbot, queue=False)
 
+# 启动 Gradio 界面
 demo.launch()