Spaces:
Sleeping
Sleeping
import gradio as gr | |
import requests | |
# 设置 FastGPT API 的基本信息 | |
API_URL = "https://api.fastgpt.in/api/v1/chat/completions" # 请根据您的部署地址修改 | |
API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF" # 请替换为您的实际 API 密钥 | |
def chat_with_fastgpt(user_input): | |
headers = { | |
"Authorization": f"Bearer {API_KEY}", | |
"Content-Type": "application/json" | |
} | |
data = { | |
"stream": False, | |
"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() | |