Update app.py

app.py

@@ -8,47 +8,35 @@ import time
 
 
 # 设置 FastGPT API 的基本信息
-API_URL = "https://api.fastgpt.com/generate"  # 请根据您的实际部署地址修改
+#API_URL = "https://api.fastgpt.com/generate"  # 请根据您的实际部署地址修改
+#API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
+
+#import gradio as gr
+#import requests
+
 API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
 
-# 调用 FastGPT API 的函数
-def chat_with_fastgpt(user_input, chat_history):
+API_URL = "https://api.fastgpt.com/v1/chat"
+
+def chat_with_fastgpt(prompt):
     headers = {
         "Authorization": f"Bearer {API_KEY}",
         "Content-Type": "application/json"
     }
-    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
-
-    # 解析并返回生成的文本
-    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="请输入您的消息...")
-    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)
-
-# 启动 Gradio 界面
-demo.launch()
+    payload = {
+        "model": "fastgpt",
+        "prompt": prompt
+    }
+    response = requests.post(API_URL, headers=headers, json=payload)
+    if response.status_code == 200:
+        return response.json().get("response", "I didn't get that.")
+    else:
+        return "Failed to get a response from the API."
+
+iface = gr.Interface(fn=chat_with_fastgpt, 
+                     inputs="text", 
+                     outputs="text", 
+                     title="Chat with FastGPT")
+
+if __name__ == "__main__":
+    iface.launch()