Update app.py

app.py

@@ -12,7 +12,7 @@ app.config['MAX_CONTENT_LENGTH'] = 10 * 1024 * 1024  # 10MB max file size
 
 # Gemini API configuration
 GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
-GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash-image-preview:generateContent"
+GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash-exp:generateContent"
 
 ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg', 'gif'}
 
@@ -20,6 +20,7 @@ def allowed_file(filename):
     return '.' in filename and \
            filename.rsplit('.', 1)[1].lower() in ALLOWED_EXTENSIONS
 
+# 优化图像函数不需要改动
 def optimize_image(image_file, max_size=(1024, 1024), quality=85):
     """优化图像大小和质量"""
     try:
@@ -49,6 +50,7 @@ def optimize_image(image_file, max_size=(1024, 1024), quality=85):
         image_file.seek(0)
         return image_file.read()
 
+# call_gemini_api 函数不需要改动
 def call_gemini_api(text_content=None, image_data=None):
     """调用Gemini API进行分析"""
     
@@ -76,8 +78,8 @@ def call_gemini_api(text_content=None, image_data=None):
 
 ## 🎯 可能诊断
 1. **最可能诊断**: [诊断名称] - 可能性: X%
-   - 依据: [具体理由]
-   
+    - 依据: [具体理由]
+    
 2. **次要考虑**: [其他可能诊断]
 
 ## 💊 治疗建议
@@ -195,12 +197,18 @@ def analyze():
         text_content = request.form.get('text', '').strip()
         image_data = None
         
-        # 处理上传的图像
+        # 修复图像处理逻辑
         if 'image' in request.files:
             file = request.files['image']
             if file.filename != '':
-                if file and allowed_file(file.filename):
+                if allowed_file(file.filename):
                     try:
+                        # 从 FileStorage 对象中直接读取数据
+                        image_data = file.read()
+                        
+                        # 确保文件指针回到开头，以便 optimize_image 函数能正确读取
+                        file.seek(0)
+                        
                         # 优化图像
                         image_data = optimize_image(file)
                     except Exception as e:
@@ -242,14 +250,9 @@ def internal_error(e):
     return jsonify({"error": "服务器内部错误"}), 500
 
 if __name__ == '__main__':
-    # 确保模板目录存在
     os.makedirs('templates', exist_ok=True)
-    
-    # 检查API密钥
     if not GEMINI_API_KEY:
         print("⚠️  警告: 未设置GEMINI_API_KEY环境变量")
-        print("   请设置环境变量: export GEMINI_API_KEY=your_api_key")
-    
-    # 启动应用
-    port = int(os.environ.get('PORT', 7860))  # Hugging Face默认端口
+        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)