Update app.py

app.py

@@ -1,150 +1,27 @@
-import os
-import json
 import gradio as gr
 import requests
-from dotenv import load_dotenv
-import logging
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+# 设置 FastGPT API 的基本信息
+API_URL = "https://api.fastgpt.in/api/v1/chat/completions"  # 请根据您的部署地址修改
+API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 请替换为您的实际 API 密钥
 
-# Load environment variables
-load_dotenv()
-
-# API configuration
-BASE_URL = "https://api.fastgpt.in/api"  # FastGPT base URL
-API_KEY = os.getenv("API_KEY", "")  # Ensure you have your API key set
-
-# Verify the API key is available
-if not API_KEY:
-    raise ValueError("Please ensure that the API_KEY environment variable is set.")
-
-class ChatBot:
-    def __init__(self):
-        self.headers = {
-            "Authorization": f"Bearer {API_KEY}",
-            "Content-Type": "application/json"
-        }
-
-def format_message(role, content):
-    return {"role": role, "content": content}
-
-def chat_stream(message, history):
-    chatbot = ChatBot()
-    
-    logger.info(f"Sending message: {message}")
-    
-    messages = [format_message("user", message)]
-    for human, assistant in history:
-        messages.append(format_message("user", human))
-        messages.append(format_message("assistant", assistant))
-    
-    payload = {
-        "messages": messages,
-        "stream": True  # Set to True if you want streaming responses
+def chat_with_fastgpt(user_input):
+    headers = {
+        "Authorization": f"Bearer {API_KEY}",
+        "Content-Type": "application/json"
     }
-    
-    try:
-        logger.info(f"Sending request data: {json.dumps(payload, ensure_ascii=False)}")
-        
-        response = requests.post(
-            f"{BASE_URL}/v1/chat/completions",
-            headers=chatbot.headers,
-            json=payload,
-            stream=True
-        )
-        
-        if response.status_code != 200:
-            error_msg = f"API returned error status code: {response.status_code}\nError message: {response.text}"
-            logger.error(error_msg)
-            history.append((message, error_msg))
-            return history
-        
-        partial_message = ""
-        
-        for line in response.iter_lines():
-            if line:
-                try:
-                    line = line.decode('utf-8')
-                    logger.info(f"Received data: {line}")
-                    
-                    if line.startswith('data: '):
-                        line = line[6:]
-                        
-                        if line == '[DONE]':
-                            break
-                            
-                        chunk = json.loads(line)
-                        if chunk and "choices" in chunk and len(chunk["choices"]) > 0:
-                            delta = chunk["choices"][0]["delta"]
-                            if "content" in delta:
-                                content = delta["content"]
-                                partial_message += content
-                                history.append((message, partial_message))
-                                yield history
-                                history.pop()
-                        else:
-                            logger.warning("Received empty or malformed choice from API.")
-                except json.JSONDecodeError as e:
-                    logger.error(f"JSON parsing error: {e}")
-                    continue
-        
-        if not partial_message:
-            error_msg = "No valid response content received"
-            logger.error(error_msg)
-            history.append((message, error_msg))
-        else:
-            history.append((message, partial_message))
-        
-    except Exception as e:
-        error_msg = f"Request error: {str(e)}"
-        logger.error(error_msg)
-        history.append((message, error_msg))
-    
-    return history
-
-# Gradio interface configuration
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    chatbot = gr.Chatbot(
-        height=600,
-        show_copy_button=True,
-        avatar_images=["assets/user.png", "assets/assistant.png"],
-    )
-    msg = gr.Textbox(
-        placeholder="Type your question here...",
-        container=False,
-        scale=7,
-    )
-    with gr.Row():
-        submit = gr.Button("Send", scale=2, variant="primary")
-        clear = gr.Button("Clear Conversation", scale=1)
-    
-    # Event handling
-    msg.submit(
-        chat_stream,
-        [msg, chatbot],
-        [chatbot],
-        api_name="chat"
-    ).then(
-        lambda: "",
-        None,
-        [msg],
-        api_name="clear_input"
-    )
-    
-    submit.click(
-        chat_stream,
-        [msg, chatbot],
-        [chatbot],
-    ).then(
-        lambda: "",
-        None,
-        [msg],
-    )
-    
-    clear.click(lambda: [], None, chatbot)
-
-# Start the application
-if __name__ == "__main__":
-    demo.queue().launch(debug=True)
+    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()