Update app.py

app.py

@@ -47,43 +47,56 @@ def safe_model_call(prompt):
         print(f"尝试调用模型，使用token长度: {len(hf_token)}")
         print(f"使用的HF_ENDPOINT: {os.environ.get('HF_ENDPOINT', '默认')}")
         
-        # 使用更适合中文医疗问答的模型
+        # 尝试使用专门针对医疗领域的模型
         try:
-            # 首先尝试BLOOMZ-7B1模型（支持中文且参数量更大）
+            # 首先尝试使用医疗领域的模型
             return client.text_generation(
                 prompt=prompt,
-                model="bigscience/bloomz-7b1",  # 更大的多语言模型
-                max_new_tokens=200,
-                temperature=0.5,  # 降低温度以获得更确定性的回答
+                model="epfl-llm/meditron-7b",  # 医疗领域专用模型
+                max_new_tokens=300,  # 增加输出长度
+                temperature=0.3,  # 降低温度以获得更确定性的回答
                 repetition_penalty=1.2,
                 do_sample=True,
                 top_p=0.9
             )
-        except Exception as model_error:
-            print(f"首选模型调用失败: {str(model_error)}")
+        except Exception as meditron_error:
+            print(f"Meditron模型调用失败: {str(meditron_error)}")
             print("尝试使用备用模型...")
             
-            # 尝试使用备用模型
+            # 尝试使用中文医疗模型
             try:
-                # 尝试使用Qwen模型（阿里开源的中文大模型）
                 return client.text_generation(
                     prompt=prompt,
-                    model="Qwen/Qwen-7B",  # 备用中文模型
-                    max_new_tokens=200,
-                    temperature=0.5,
+                    model="IDEA-CCNL/Ziya-LLaMA-13B-v1",  # 中文大模型
+                    max_new_tokens=300,
+                    temperature=0.3,
                     repetition_penalty=1.2
                 )
-            except Exception as qwen_error:
-                print(f"Qwen模型调用失败: {str(qwen_error)}")
+            except Exception as ziya_error:
+                print(f"Ziya模型调用失败: {str(ziya_error)}")
                 
-                # 最后尝试使用BLOOMZ-1B7
-                return client.text_generation(
-                    prompt=prompt,
-                    model="bigscience/bloomz-1b7",  # 最后备选
-                    max_new_tokens=200,
-                    temperature=0.5,
-                    repetition_penalty=1.2
-                )
+                # 尝试使用通用大模型
+                try:
+                    return client.text_generation(
+                        prompt=prompt,
+                        model="bigscience/bloomz-7b1",  # 多语言模型
+                        max_new_tokens=300,
+                        temperature=0.3,
+                        repetition_penalty=1.2,
+                        do_sample=True,
+                        top_p=0.9
+                    )
+                except Exception as bloomz_error:
+                    print(f"BLOOMZ模型调用失败: {str(bloomz_error)}")
+                    
+                    # 最后尝试使用小模型
+                    return client.text_generation(
+                        prompt=prompt,
+                        model="bigscience/bloomz-1b7",  # 最后备选
+                        max_new_tokens=200,
+                        temperature=0.5,
+                        repetition_penalty=1.2
+                    )
     except Exception as e:
         error_type = type(e).__name__
         error_msg = str(e)
@@ -106,17 +119,92 @@ def local_fallback_response(query):
     """当API调用失败时提供基本的本地响应"""
     query = query.lower()
     
-    # 简单的关键词匹配
+    # 详细的头痛响应
+    if "头痛" in query:
+        days_match = None
+        if "三天" in query or "3天" in query:
+            days_match = "三天"
+        elif "两天" in query or "2天" in query:
+            days_match = "两天"
+        elif "一天" in query or "1天" in query:
+            days_match = "一天"
+        
+        response = f"""关于{'持续'+days_match+'的' if days_match else ''}头痛问题，我可以提供以下建议：
+
+**可能的原因**：
+- 紧张性头痛：由压力、焦虑或肌肉紧张引起
+- 偏头痛：常伴有对光、声音敏感，有时伴随恶心
+- 脱水：体内水分不足可导致头痛
+- 睡眠不足或睡眠质量差
+- 眼睛疲劳：长时间用眼或需要更新眼镜处方
+- 感染：如感冒、流感或鼻窦感染
+
+**建议措施**：
+1. 休息：在安静、黑暗的房间休息
+2. 水分补充：确保充分饮水
+3. 热/冷敷：在疼痛部位使用热敷或冷敷
+4. 按摩：轻轻按摩太阳穴或颈部
+5. 非处方止痛药：如对乙酰氨基酚或布洛芬（请按说明使用）
+6. 减轻压力：尝试放松技巧如深呼吸或冥想
+
+**注意事项**：
+- {'由于头痛已持续'+days_match+'，建议咨询医生' if days_match else '如果头痛持续超过3天，应咨询医生'}
+- 如果头痛非常剧烈、突然发作或伴随其他症状（如发热、颈部僵硬、视力问题、意识混乱），请立即就医
+- 定期头痛需要专业医生评估
+
+希望这些信息对您有所帮助。请记住，这只是一般性建议，不能替代专业医疗咨询。"""
+        return response
+    
+    # 其他关键词匹配
     responses = {
-        "头痛": "头痛可能由多种原因引起，如疲劳、压力、脱水或感染。建议休息、保持水分摄入、可以尝试按摩太阳穴或服用非处方止痛药。如果头痛剧烈或持续时间长，请咨询医生。",
-        "发烧": "发热是身体对抗感染的自然反应。轻度发热可多喝水、休息。如果体温超过38.5°C，可以使用退烧药。高烧或持续发热超过3天应就医。",
-        "咳嗽": "咳嗽可能是由感冒、过敏或其他呼吸道问题引起。保持水分摄入，可使用非处方止咳药。如果咳嗽持续超过2周或伴有血痰，请咨询医生。",
+        "发烧": """关于发热问题：
+
+**可能的原因**：
+- 感染（病毒、细菌）
+- 免疫反应
+- 疫苗接种后反应
+- 其他炎症反应
+
+**建议措施**：
+1. 休息：保持充分休息
+2. 水分补充：多喝水防止脱水
+3. 降温：使用退热药物（如对乙酰氨基酚）
+4. 轻便衣物：穿着轻便，避免过热
+
+**注意事项**：
+- 如果体温超过38.5°C，可以使用退烧药
+- 高烧（39°C以上）或持续发热超过3天应就医
+- 如果发热伴随严重症状（如呼吸困难、剧烈头痛、意识模糊），请立即就医
+
+请记住，这只是一般性建议，不能替代专业医疗咨询。""",
+
+        "咳嗽": """关于咳嗽问题：
+
+**可能的原因**：
+- 上呼吸道感染（感冒、流感）
+- 过敏
+- 哮喘
+- 胃酸反流
+- 环境刺激物
+
+**建议措施**：
+1. 保持水分摄入：多喝温水
+2. 使用蜂蜜柠檬水：可缓解喉咙刺激
+3. 湿度：使用加湿器增加空气湿度
+4. 非处方止咳药：根据咳嗽类型选择
+
+**注意事项**：
+- 如果咳嗽持续超过2周，应咨询医生
+- 如果咳嗽伴有血痰、呼吸困难、高热或胸痛，请立即就医
+- 夜间咳嗽加重或影响睡眠也需要医疗关注
+
+请记住，这只是一般性建议，不能替代专业医疗咨询。""",
+
         "疫苗": "疫苗接种后出现接种部位疼痛、轻微发热是正常反应，通常会在几天内消失。如出现严重过敏反应或高烧，应立即就医。",
         "感冒": "感冒通常会自行痊愈，建议多休息、多喝水、可使用非处方药缓解症状。如症状加重或持续一周以上，请咨询医生。",
         "腹痛": "腹痛可能由多种原因引起，如消化不良、胃肠炎等。建议注意饮食、避免辛辣刺激食物。如果疼痛剧烈或伴随其他症状，请咨询医生。",
         "失眠": "改善睡眠可以尝试规律作息、睡前避免使用电子设备、创造舒适的睡眠环境。长期失眠建议咨询医生。",
-        "过敏": "避免接触过敏原，可使用抗组胺药缓解症状。如果出现呼吸困难等严重症状，请立即就医。",
-        "三天": "症状持续三天以上，建议及时就医进行专业诊断和治疗。"
+        "过敏": "避免接触过敏原，可使用抗组胺药缓解症状。如果出现呼吸困难等严重症状，请立即就医。"
     }
     
     # 检查查询中是否包含关键词
@@ -125,12 +213,21 @@ def local_fallback_response(query):
             return response
     
     # 默认回复
-    return "您好，我是AI医疗助手。我可以提供一些基本的健康建议，但不能替代专业医生的诊断。建议您咨询专业医生获取准确诊断和治疗方案。"
+    return """您好，我是AI医疗助手。我可以提供一些基本的健康建议，但不能替代专业医生的诊断。
+
+为了给您提供更有针对性的建议，请尽量详细描述您的症状，包括：
+- 症状的具体表现
+- 持续时间
+- 是否有加重或缓解因素
+- 是否尝试过任何缓解方法
+
+对于任何严重或持续的健康问题，建议您咨询专业医生获取准确诊断和治疗方案。"""
 
 def medical_chat(user_input, history):
-    # 构建更详细的医学对话提示词
+    # 构建更详细的医学对话提示词，针对常见问题提供具体指导
     system_prompt = """你是一位经验丰富的医生，专注于提供准确、客观、专业的医疗建议。
 请根据患者描述的症状，提供科学的建议和可能的解决方案。
+
 你的回答应该：
 1. 基于医学事实和科学证据
 2. 清晰、简洁但全面
@@ -138,8 +235,16 @@ def medical_chat(user_input, history):
 4. 在必要时建议患者就医
 5. 不做确定性的诊断，因为这需要面诊和检查
 6. 不推荐具体的药品品牌
+7. 不要说"给你开药"，因为你不能开处方
+8. 提供分析和建议，而不是简单的答复
 
-请记住，你的建议仅供参考，不能替代专业医生的面诊。"""
+对于头痛问题，你应该：
+- 解释可能的原因（如压力、疲劳、脱水等）
+- 建议一些自我缓解方法（如休息、按摩、保持水分摄入）
+- 提醒患者如果症状持续或加重应该就医
+- 不要简单地建议服药，而是提供全面的建议
+
+请记住，你的建议仅供参考，不能替代专业医生的面诊。每个回答都应该包含症状分析、可能的原因、建议措施和注意事项。"""
     
     # 构建完整的提示词
     prompt = f"{system_prompt}\n\n患者：{user_input}\n医生："
@@ -184,9 +289,30 @@ def medical_chat(user_input, history):
         
         print(f"处理后的响应: {doctor_response}")
         
+        # 增强响应质量检测
+        low_quality_indicators = [
+            "没药", "不知道", "开点", "开药", "处方药", "非处方药", 
+            "既然是", "那就", "我就给你", "吧", "呢", "哦", "呀",
+            "简单", "随便", "试试看", "看看", "应该没事"
+        ]
+        
         # 检查响应质量
-        if len(doctor_response.strip()) < 10 or "没药" in doctor_response or "不知道" in doctor_response:
-            print(f"响应质量不佳: '{doctor_response}'，使用本地备用模式")
+        if len(doctor_response.strip()) < 50:  # 增加最小长度要求
+            print(f"响应太短: '{doctor_response}'，使用本地备用模式")
+            return local_fallback_response(user_input)
+            
+        # 检查是否包含低质量指标
+        for indicator in low_quality_indicators:
+            if indicator in doctor_response:
+                print(f"响应包含低质量指标 '{indicator}': '{doctor_response}'，使用本地备用模式")
+                return local_fallback_response(user_input)
+                
+        # 检查是否包含足够的医疗建议内容
+        medical_advice_indicators = ["建议", "可以", "应该", "如果", "症状", "原因"]
+        has_medical_advice = any(indicator in doctor_response for indicator in medical_advice_indicators)
+        
+        if not has_medical_advice:
+            print(f"响应缺乏医疗建议: '{doctor_response}'，使用本地备用模式")
             return local_fallback_response(user_input)
         
         # 如果响应中没有"医生："，则直接返回整个响应