Update app.py

app.py

@@ -16,16 +16,14 @@ GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
 def allowed_file(filename):
-    return '.' in filename and \
-           filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
+    return '.' in filename and filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
-# 优化图像函数，现在接受字节数据作为输入
 def optimize_image(image_bytes, max_size=(1024, 1024), quality=85):
     """优化图像大小和质量"""
     try:
         # 使用 io.BytesIO 将字节数据转换为文件对象，供 Pillow 使用
         img = Image.open(io.BytesIO(image_bytes))
-        
+
         if img.mode in ('RGBA', 'LA'):
             background = Image.new('RGB', img.size, (255, 255, 255))
             if img.mode == 'RGBA':
@@ -43,63 +41,71 @@ def optimize_image(image_bytes, max_size=(1024, 1024), quality=85):
         return img_byte_arr.getvalue()
     except Exception as e:
         print(f"Image optimization error: {e}")
-        # 优化失败，返回原始字节数据
-        return image_bytes
+        return image_bytes  # 优化失败，返回原始字节数据
 
-# call_gemini_api 函数
 def call_gemini_api(text_content=None, image_data=None):
     """调用Gemini API进行分析"""
-    
     if not GEMINI_API_KEY:
         return {"error": "未配置Gemini API密钥，请设置环境变量GEMINI_API_KEY"}
-    
+
     system_prompt = """你是一位经验丰富的皮肤科专家医生，具有多年的临床诊疗经验。请基于提供的信息进行专业的皮肤病分析。
 
-分析要求：
-1. 仔细观察皮肤病变的形态、颜色、分布、边界等特征
-2. 结合患者描述的症状和病史
-3. 提供可能的诊断方向（按可能性大小排序）
-4. 给出详细的诊疗建议
-5. 必要时建议进一步检查
-6. 提醒患者注意事项
+    分析要求：
+
+    仔细观察皮肤病变的形态、颜色、分布、边界等特征
+
+    结合患者描述的症状和病史
+
+    提供可能的诊断方向（按可能性大小排序）
+
+    给出详细的诊疗建议
+
+    必要时建议进一步检查
+
+    提醒患者注意事项
+    请按以下格式回复：
+
+    🔍 图像观察分析
+
+    [详细描述观察到的皮肤病变特征]
 
-请按以下格式回复：
+    📋 症状综合评估
 
-## 🔍 图像观察分析
-[详细描述观察到的皮肤病变特征]
+    [结合文字描述的症状分析]
 
-## 📋 症状综合评估
-[结合文字描述的症状分析]
+    🎯 可能诊断
 
-## 🎯 可能诊断
-1. **最可能诊断**: [诊断名称] - 可能性: X%
-    - 依据: [具体理由]
-    
-2. **次要考虑**: [其他可能诊断]
+    最可能诊断: [诊断名称] - 可能性: X%
 
-## 💊 治疗建议
-### 药物治疗
-- [具体用药建议]
+    依据: [具体理由]
 
-### 生活护理
-- [日常护理要点]
+    次要考虑: [其他可能诊断]
 
-## 🏥 进一步检查建议
-[如需要其他检查，请列出]
+    💊 治疗建议
+    药物治疗
 
-## ⚠️ 注意事项
-[重要提醒和注意事项]
+    [具体用药建议]
 
----
-**免责声明**: 此分析仅供临床参考，最终诊断需结合完整病史、体格检查等综合判断。建议患者及时就医，接受专业医师诊疗。"""
+    生活护理
+
+    [日常护理要点]
+
+    🏥 进一步检查建议
+
+    [如需要其他检查，请列出]
+
+    ⚠️ 注意事项
+    [重要提醒和注意事项]
+
+    免责声明: 此分析仅供临床参考，最终诊断需结合完整病史、体格检查等综合判断。建议患者及时就医，接受专业医师诊疗。"""
 
     parts = []
-    
+
     if text_content:
         parts.append({"text": f"{system_prompt}\n\n患者症状描述：{text_content}"})
     else:
         parts.append({"text": system_prompt + "\n\n请基于图像进行分析。"})
-    
+
     if image_data:
         image_base64 = base64.b64encode(image_data).decode('utf-8')
         parts.append({
@@ -108,7 +114,7 @@ def call_gemini_api(text_content=None, image_data=None):
                 "data": image_base64
             }
         })
-    
+
     request_body = {
         "contents": [
             {
@@ -140,12 +146,12 @@ def call_gemini_api(text_content=None, image_data=None):
             }
         ]
     }
-    
+
     headers = {
         'Content-Type': 'application/json',
         'x-goog-api-key': GEMINI_API_KEY
     }
-    
+
     try:
         response = requests.post(
             GEMINI_API_URL,
@@ -180,7 +186,6 @@ def index():
 @app.route('/analyze', methods=['POST'])
 def analyze():
     """分析皮肤图像和症状的API端点"""
-    
     try:
         text_content = request.form.get('text', '').strip()
         image_data = None
@@ -237,4 +242,4 @@ if __name__ == '__main__':
         print("⚠️  警告: 未设置GEMINI_API_KEY环境变量")
         print("    请设置环境变量: export GEMINI_API_KEY=your_api_key")
     port = int(os.environ.get('PORT', 7860))
-    app.run(host='0.0.0.0', port=port, debug=False)
+    app.run(host='0.0.0.0', port=port, debug=False)