Update app.py

app.py

@@ -18,6 +18,18 @@ class FastGPTChat:
             "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():
@@ -155,12 +167,42 @@ def create_chat_interface():
             background-color: #ff6b81 !important;
             transform: translateY(-2px);
         }
+        /* 优化对话区域样式 */
         .chatbot {
             background-color: #f8f9fa;
             border-radius: 10px;
-            padding: 10px;
+            padding: 15px;
             margin-bottom: 20px;
         }
+        /* 用户消息样式 */
+        .chatbot .user-message {
+            background-color: #e3f2fd !important;
+            border-radius: 15px 15px 2px 15px !important;
+            padding: 10px 15px !important;
+            margin: 5px 0 !important;
+            max-width: 85% !important;
+            float: right !important;
+            clear: both !important;
+        }
+        /* AI响应消息样式 */
+        .chatbot .bot-message {
+            background-color: #ffffff !important;
+            border-radius: 15px 15px 15px 2px !important;
+            padding: 10px 15px !important;
+            margin: 5px 0 !important;
+            max-width: 85% !important;
+            float: left !important;
+            clear: both !important;
+            border: 1px solid #e0e0e0 !important;
+        }
+        /* 消息文本样式 */
+        .chatbot .message-text {
+            font-size: 15px !important;
+            line-height: 1.5 !important;
+            color: #2c3e50 !important;
+            margin: 0 !important;
+            word-wrap: break-word !important;
+        }
         """
     ) as interface:
         with gr.Column(elem_classes="container"):
@@ -176,7 +218,9 @@ def create_chat_interface():
                     height=500,
                     container=False,
                     elem_classes="chatbot",
-                    show_label=False
+                    show_label=False,
+                    bubble_full_width=False,
+                    avatar_images=("👤", "🤖")
                 )
                 
                 with gr.Row(elem_id="input-container"):