app.py

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)