Update app.py

app.py

@@ -1,27 +1,57 @@
 import gradio as gr
 import requests
+import time
 
 # 设置 FastGPT API 的基本信息
 API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
 API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
 
-def chat_with_fastgpt(user_input):
+
+# 设置 FastGPT API 的基本信息
+#API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
+#API_KEY = "您的_API_密钥"  # 请替换为您的实际 API 密钥
+
+def chat_with_fastgpt(user_input, chat_history):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
     data = {
-        "stream": False,
+        "stream": True,
         "chatId": "unique_chat_id",  # 请替换为唯一的聊天 ID
         "messages": [{"role": "user", "content": user_input}]
     }
-    response = requests.post(API_URL, headers=headers, json=data)
-    if response.status_code == 200:
-        response_data = response.json()
-        return response_data['choices'][0]['message']['content']
-    else:
-        return f"请求失败，状态码：{response.status_code}"
-
-# 创建 Gradio 接口
-interface = gr.Interface(fn=chat_with_fastgpt, inputs="text", outputs="text", live=True)
-interface.launch()
+    response = requests.post(API_URL, headers=headers, json=data, stream=True)
+    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
+
+def user(user_input, chat_history):
+    chat_history.append([user_input, None])
+    return "", chat_history
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="请输入您的消息...")
+    clear = gr.Button("清除")
+    submit = gr.Button("发送")
+
+    submit.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+        chat_with_fastgpt, [msg, chatbot], [msg, chatbot]
+    )
+    clear.click(lambda: None, None, chatbot, queue=False)
+
+demo.launch()