Create app_bak3.py

app_bak3.py

@@ -0,0 +1,284 @@
+import gradio as gr
+import requests
+import json
+import uuid
+import logging
+import os
+
+# 配置日志
+logging.basicConfig(level=logging.DEBUG)
+logger = logging.getLogger(__name__)
+
+class FastGPTChat:
+    """
+    def __init__(self, api_key="", 
+                 system_prompt="我是一名AI用药咨询顾问，请向我提问有关用药的问题"):
+        self.api_key = api_key
+        self.base_url = "https://api.fastgpt.in/api"
+        self.system_prompt = system_prompt
+        self.headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json"
+        }
+    """
+
+    def __init__(self, system_prompt="我是一名AI用药咨询顾问，请向我提问有关用药的问题"):
+        # 从环境变量获取API密钥和基础URL
+        self.api_key = os.environ.get('FASTGPT_API_KEY')
+        self.base_url = os.environ.get('FASTGPT_BASE_URL', 'https://api.fastgpt.in/api')
+        self.system_prompt = system_prompt
+        self.headers = {
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        }
+
+
+    
+    def chat(self, message, chat_history):
+        if not message.strip():
+            return "", chat_history
+            
+        chat_id = str(uuid.uuid4())
+        
+        # 构建消息历史，包含系统提示词
+        messages = []
+        if self.system_prompt:
+            messages.append({
+                "role": "system",
+                "content": self.system_prompt
+            })
+        
+        # 添加聊天历史
+        for user_msg, bot_msg in chat_history:
+            messages.append({"role": "user", "content": user_msg})
+            messages.append({"role": "assistant", "content": bot_msg})
+        
+        # 添加当前用户消息
+        messages.append({"role": "user", "content": message})
+        
+        # 准备请求数据
+        data = {
+            "chatId": chat_id,
+            "stream": True,
+            "detail": True,
+            "model": "gpt-4o",
+            "messages": messages
+        }
+        
+        try:
+            # 发送流式请求
+            response = requests.post(
+                f"{self.base_url}/v1/chat/completions",
+                headers=self.headers,
+                json=data,
+                stream=True,
+                timeout=30
+            )
+            
+            if response.status_code != 200:
+                error_msg = f"API Error: Status {response.status_code}"
+                logger.error(error_msg)
+                chat_history.append((message, f"Error: {error_msg}"))
+                return "", chat_history
+            
+            # 处理流式响应
+            full_response = ""
+            for line in response.iter_lines():
+                if line:
+                    try:
+                        line = line.decode('utf-8')
+                        if line.startswith('data: '):
+                            data = json.loads(line[6:])
+                            if 'choices' in data and len(data['choices']) > 0:
+                                content = data['choices'][0]['delta'].get('content', '')
+                                if content:
+                                    full_response += content
+                                    # 更新聊天历史
+                                    if len(chat_history) > 0 and chat_history[-1][0] == message:
+                                        chat_history[-1] = (message, full_response)
+                                    else:
+                                        chat_history.append((message, full_response))
+                                    yield "", chat_history
+                    except Exception as e:
+                        logger.error(f"Error processing stream: {str(e)}")
+            
+            return "", chat_history
+                
+        except requests.exceptions.RequestException as e:
+            error_msg = f"Request failed: {str(e)}"
+            logger.error(error_msg)
+            chat_history.append((message, f"Error: {error_msg}"))
+            return "", chat_history
+
+def create_chat_interface():
+    fastgpt_chat = FastGPTChat()
+    
+    with gr.Blocks(
+        title="AI用药咨询助手",
+        css="""
+        .container { max-width: 800px; margin: auto; }
+        .chat-header { 
+            text-align: center; 
+            padding: 20px;
+            background: linear-gradient(135deg, #00b4d8, #0077b6);
+            color: white;
+            border-radius: 15px 15px 0 0;
+            margin-bottom: 20px;
+        }
+        .chat-container { 
+            background-color: #ffffff; 
+            border-radius: 15px; 
+            padding: 20px;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+            border: 1px solid #e0e0e0;
+        }
+        .message-box {
+            border: 2px solid #e0e0e0;
+            border-radius: 10px;
+            background-color: white;
+            transition: all 0.3s ease;
+        }
+        .message-box:focus {
+            border-color: #00b4d8;
+            box-shadow: 0 0 0 2px rgba(0, 180, 216, 0.2);
+        }
+        .submit-btn {
+            background-color: #00b4d8 !important;
+            border: none !important;
+            color: white !important;
+            padding: 10px 20px !important;
+            border-radius: 8px !important;
+            transition: all 0.3s ease !important;
+            font-weight: 600 !important;
+        }
+        .submit-btn:hover {
+            background-color: #0077b6 !important;
+            transform: translateY(-2px);
+        }
+        .clear-btn {
+            background-color: #ff4757 !important;
+            border: none !important;
+            color: white !important;
+            padding: 10px 20px !important;
+            border-radius: 8px !important;
+            transition: all 0.3s ease !important;
+            font-weight: 600 !important;
+        }
+        .clear-btn:hover {
+            background-color: #ff6b81 !important;
+            transform: translateY(-2px);
+        }
+        .chatbot {
+            background-color: #f8f9fa;
+            border-radius: 10px;
+            padding: 15px;
+            margin-bottom: 20px;
+        }
+        /* 移除消息背景的白色遮盖，优化对话样式 */
+        .chatbot > div {
+            background: transparent !important;
+        }
+        /* 用户消息样式 */
+        .chatbot .user-message {
+            background-color: #fce4ec !important;
+            border-radius: 15px 15px 2px 15px !important;
+            padding: 12px 18px !important;
+            margin: 8px 0 !important;
+            max-width: 85% !important;
+            float: right !important;
+            clear: both !important;
+            border: 1px solid #f48fb1 !important;
+            color: #1a1a1a !important;
+            font-size: 15px !important;
+        }
+        /* AI响应消息样式 */
+        .chatbot .bot-message {
+            background-color: #e8f5e9 !important;
+            border-radius: 15px 15px 15px 2px !important;
+            padding: 12px 18px !important;
+            margin: 8px 0 !important;
+            max-width: 85% !important;
+            float: left !important;
+            clear: both !important;
+            border: 1px solid #a5d6a7 !important;
+            color: #1a1a1a !important;
+            font-size: 15px !important;
+        }
+        /* 消息容器样式 */
+        .chatbot > div > div {
+            padding: 0 !important;
+            gap: 2rem !important;
+        }
+        /* 头像样式 */
+        .chatbot span.avatar {
+            padding: 8px !important;
+            margin: 8px !important;
+        }
+        /* 确保消息之间有适当的间距 */
+        .chatbot > div > div:not(:last-child) {
+            margin-bottom: 15px !important;
+        }
+        """
+    ) as interface:
+        with gr.Column(elem_classes="container"):
+            gr.Markdown(
+                """
+                # 🏥 AI用药咨询助手
+                ### 您的专业用药咨询顾问，随时为您解答用药相关问题
+                """
+            )
+            
+            with gr.Column(elem_classes="chat-container"):
+                chatbot = gr.Chatbot(
+                    height=500,
+                    container=False,
+                    elem_classes="chatbot",
+                    show_label=False,
+                    bubble_full_width=False,
+                    avatar_images=("👤", "🤖")
+                )
+                
+                with gr.Row(elem_id="input-container"):
+                    with gr.Column(scale=8):
+                        message = gr.Textbox(
+                            show_label=False,
+                            placeholder="请输入您的用药问题...",
+                            container=False,
+                            elem_classes="message-box",
+                            lines=2
+                        )
+                    with gr.Column(scale=1, min_width=100):
+                        submit = gr.Button(
+                            "发送 💊", 
+                            variant="primary",
+                            elem_classes="submit-btn"
+                        )
+                
+                with gr.Row(elem_id="control-container"):
+                    clear = gr.Button(
+                        "清空对话 🗑️", 
+                        variant="secondary",
+                        elem_classes="clear-btn"
+                    )
+        
+        # 设置事件处理
+        submit_click = submit.click(
+            fastgpt_chat.chat,
+            inputs=[message, chatbot],
+            outputs=[message, chatbot],
+            api_name="chat"
+        )
+        
+        message.submit(
+            fastgpt_chat.chat,
+            inputs=[message, chatbot],
+            outputs=[message, chatbot]
+        )
+        
+        clear.click(lambda: None, None, chatbot, queue=False)
+        
+    return interface
+
+if __name__ == "__main__":
+    demo = create_chat_interface()
+    demo.launch(debug=True)