Update app.py

app.py

@@ -9,9 +9,10 @@ logging.basicConfig(level=logging.DEBUG)
 logger = logging.getLogger(__name__)
 
 class FastGPTChat:
-    def __init__(self, api_key, system_prompt="", base_url="https://api.fastgpt.in/api"):
+    def __init__(self, api_key="fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF", 
+                 system_prompt="我是一名AI用药咨询顾问，请向我提问有关用药的问题"):
         self.api_key = api_key
-        self.base_url = base_url
+        self.base_url = "https://api.fastgpt.in/api"
         self.system_prompt = system_prompt
         self.headers = {
             "Authorization": f"Bearer {api_key}",
@@ -43,7 +44,7 @@ class FastGPTChat:
         # 准备请求数据
         data = {
             "chatId": chat_id,
-            "stream": True,  # 启用流式响应
+            "stream": True,
             "detail": True,
             "model": "gpt-4o",
             "messages": messages
@@ -94,105 +95,134 @@ class FastGPTChat:
             chat_history.append((message, f"Error: {error_msg}"))
             return "", chat_history
 
-def create_chat_interface(api_key):
+def create_chat_interface():
     # 创建FastGPT聊天实例
-    fastgpt_chat = FastGPTChat(api_key)
+    fastgpt_chat = FastGPTChat()
     
     # 创建美化后的Gradio界面
     with gr.Blocks(
-        title="FastGPT Chat",
+        title="AI用药咨询助手",
         css="""
         .container { max-width: 800px; margin: auto; }
-        .chat-header { text-align: center; padding: 20px; }
-        .chat-container { background-color: #f5f5f5; border-radius: 10px; padding: 20px; }
+        .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: 10px;
+            margin-bottom: 20px;
+        }
         """
     ) as interface:
         with gr.Column(elem_classes="container"):
             gr.Markdown(
                 """
-                # 🤖 FastGPT Chat Interface
-                ### Powered by GPT-4o
+                # 🏥 AI用药咨询助手
+                ### 您的专业用药咨询顾问，随时为您解答用药相关问题
                 """
             )
             
-            with gr.Row():
-                api_key_input = gr.Textbox(
-                    label="API Key",
-                    value=api_key,
-                    type="password",
-                    container=False,
-                    scale=4
-                )
-            
-            with gr.Row():
-                system_prompt = gr.Textbox(
-                    label="System Prompt",
-                    placeholder="Set the behavior of AI assistant...",
-                    lines=2,
-                    container=False
-                )
-            
             with gr.Column(elem_classes="chat-container"):
                 chatbot = gr.Chatbot(
                     height=500,
                     container=False,
-                    elem_classes="chatbot"
+                    elem_classes="chatbot",
+                    show_label=False
                 )
                 
-                with gr.Row():
-                    message = gr.Textbox(
-                        label="Message",
-                        placeholder="Type your message here...",
-                        lines=2,
-                        container=False,
-                        scale=4
-                    )
+                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():
-                    submit = gr.Button("Send 📤", variant="primary")
-                    clear = gr.Button("Clear 🗑️", variant="secondary")
-            
-            # 状态显示
-            status = gr.Textbox(label="Status", interactive=False)
-        
-        def update_chat_settings(new_key, new_prompt):
-            nonlocal fastgpt_chat
-            fastgpt_chat = FastGPTChat(new_key, new_prompt)
-            return "Settings updated"
+                with gr.Row(elem_id="control-container"):
+                    clear = gr.Button(
+                        "清空对话 🗑️", 
+                        variant="secondary",
+                        elem_classes="clear-btn"
+                    )
         
         # 设置事件处理
-        submit_event = submit.click(
+        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)
         
-        # 更新设置
-        api_key_input.change(
-            update_chat_settings,
-            inputs=[api_key_input, system_prompt],
-            outputs=[status]
-        )
-        
-        system_prompt.change(
-            update_chat_settings,
-            inputs=[api_key_input, system_prompt],
-            outputs=[status]
-        )
-        
     return interface
 
 # 使用示例
 if __name__ == "__main__":
-    API_KEY = "fastgpt-aleL83PzPiul9lZDJQFD4NHYfVIxP7mDIWFCyin2FuMhYgwlRC3NMkV2K5lxc8gF"  # 替换为你的API密钥
-    demo = create_chat_interface(API_KEY)
+    demo = create_chat_interface()
     demo.launch(debug=True)