Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,134 +1,45 @@
|
|
| 1 |
import os
|
| 2 |
-
import
|
| 3 |
-
import json
|
| 4 |
import logging
|
| 5 |
-
from
|
| 6 |
|
| 7 |
-
#
|
| 8 |
-
logging.basicConfig(level=logging.INFO)
|
| 9 |
logger = logging.getLogger(__name__)
|
| 10 |
|
| 11 |
-
|
| 12 |
-
app = Flask(__name__, static_folder='static', static_url_path='')
|
| 13 |
|
| 14 |
-
|
| 15 |
-
GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
|
| 16 |
|
| 17 |
-
|
| 18 |
-
GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"
|
| 19 |
|
| 20 |
-
|
| 21 |
-
SYSTEM_PROMPT = """你是一个世界顶级的皮肤科医生AI助手。你的任务是基于用户上传的皮肤图像和病历文本,进行专业的临床诊疗辅助分析。你的输出必须是一个单一、有效的JSON对象,不包含任何JSON格式之外的文字或标记(例如不要使用 "```json")。
|
| 22 |
-
|
| 23 |
-
请严格遵循以下JSON结构:
|
| 24 |
-
{
|
| 25 |
-
"image_features": ["从图像中识别出的关键皮损形态学特征列表", "例如:色素网络不规则", "例如:蓝白结构"],
|
| 26 |
-
"differential_diagnosis": [
|
| 27 |
-
{
|
| 28 |
-
"diagnosis": "可能性最高的诊断名称",
|
| 29 |
-
"likelihood": "高",
|
| 30 |
-
"evidence": "支持该诊断的关键图像和文本依据"
|
| 31 |
-
},
|
| 32 |
-
{
|
| 33 |
-
"diagnosis": "可能性次之的诊断名称",
|
| 34 |
-
"likelihood": "中",
|
| 35 |
-
"evidence": "支持该诊断的关键依据"
|
| 36 |
-
}
|
| 37 |
-
],
|
| 38 |
-
"comprehensive_analysis": "一段对整个病情的全面、综合性分析和总结。",
|
| 39 |
-
"recommendations": {
|
| 40 |
-
"further_examinations": ["建议的下一步检查列表", "例如:皮肤镜检查", "例如:皮肤活检"],
|
| 41 |
-
"treatment_suggestions": ["初步的、符合临床指南的治疗方向建议列表", "例如:外用糖皮质激素", "例如:建议手术切除"]
|
| 42 |
-
}
|
| 43 |
-
}"""
|
| 44 |
|
| 45 |
@app.route('/healthz')
|
| 46 |
def health_check():
|
| 47 |
-
"
|
| 48 |
-
logger.info("Health check endpoint accessed")
|
| 49 |
return "OK", 200
|
| 50 |
|
| 51 |
@app.route('/')
|
| 52 |
def index():
|
| 53 |
-
""
|
| 54 |
-
|
| 55 |
-
|
| 56 |
-
|
| 57 |
-
|
| 58 |
-
|
| 59 |
-
|
| 60 |
-
|
| 61 |
-
|
| 62 |
-
|
| 63 |
-
|
| 64 |
-
|
| 65 |
-
|
| 66 |
-
|
| 67 |
-
|
| 68 |
-
|
| 69 |
-
|
| 70 |
-
|
| 71 |
-
image_base64 = data.get('image')
|
| 72 |
-
text_prompt = data.get('text')
|
| 73 |
-
|
| 74 |
-
headers = {
|
| 75 |
-
'Content-Type': 'application/json',
|
| 76 |
-
}
|
| 77 |
-
|
| 78 |
-
payload = {
|
| 79 |
-
"contents": [
|
| 80 |
-
{
|
| 81 |
-
"parts": [
|
| 82 |
-
{"text": SYSTEM_PROMPT},
|
| 83 |
-
{"text": f"临床病历信息如下:\n{text_prompt}"},
|
| 84 |
-
{
|
| 85 |
-
"inline_data": {
|
| 86 |
-
"mime_type": "image/jpeg",
|
| 87 |
-
"data": image_base64
|
| 88 |
-
}
|
| 89 |
-
}
|
| 90 |
-
]
|
| 91 |
-
}
|
| 92 |
-
]
|
| 93 |
-
}
|
| 94 |
-
|
| 95 |
-
try:
|
| 96 |
-
logger.info("Calling Gemini API")
|
| 97 |
-
response = requests.post(
|
| 98 |
-
f"{GEMINI_API_URL}?key={GEMINI_API_KEY}",
|
| 99 |
-
headers=headers,
|
| 100 |
-
json=payload,
|
| 101 |
-
timeout=60
|
| 102 |
-
)
|
| 103 |
-
response.raise_for_status()
|
| 104 |
-
|
| 105 |
-
gemini_result = response.json()
|
| 106 |
-
logger.info("Received response from Gemini API")
|
| 107 |
-
|
| 108 |
-
# 修复解析错误
|
| 109 |
-
try:
|
| 110 |
-
analysis_content_string = gemini_result['candidates'][0]['content']['parts'][0]['text']
|
| 111 |
-
except (KeyError, IndexError) as e:
|
| 112 |
-
logger.error(f"Error parsing Gemini response structure: {e}")
|
| 113 |
-
logger.error(f"Gemini response: {gemini_result}")
|
| 114 |
-
return jsonify({"error": "解析Gemini API响应失败,格式不正确", "details": str(gemini_result)}), 500
|
| 115 |
-
|
| 116 |
-
try:
|
| 117 |
-
parsed_json = json.loads(analysis_content_string)
|
| 118 |
-
logger.info("Successfully parsed JSON response")
|
| 119 |
-
return jsonify(parsed_json)
|
| 120 |
-
except json.JSONDecodeError as e:
|
| 121 |
-
logger.error(f"JSON decode error: {e}")
|
| 122 |
-
logger.error(f"Content: {analysis_content_string}")
|
| 123 |
-
return jsonify({"error": "Gemini API未返回有效的JSON格式", "details": analysis_content_string}), 500
|
| 124 |
-
|
| 125 |
-
except requests.exceptions.RequestException as e:
|
| 126 |
-
logger.error(f"Network error calling Gemini API: {e}")
|
| 127 |
-
return jsonify({"error": f"调用Gemini API时发生网络错误: {e}"}), 500
|
| 128 |
-
except Exception as e:
|
| 129 |
-
logger.error(f"Unexpected error: {e}")
|
| 130 |
-
return jsonify({"error": f"发生未知错误: {e}"}), 500
|
| 131 |
|
| 132 |
if __name__ == '__main__':
|
| 133 |
-
logger.info("Starting Flask
|
| 134 |
-
app.run(host='0.0.0.0', port=8080
|
|
|
|
| 1 |
import os
|
| 2 |
+
import sys
|
|
|
|
| 3 |
import logging
|
| 4 |
+
from datetime import datetime
|
| 5 |
|
| 6 |
+
# 设置日志
|
| 7 |
+
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
| 8 |
logger = logging.getLogger(__name__)
|
| 9 |
|
| 10 |
+
logger.info(f"Starting application at {datetime.now()}")
|
|
|
|
| 11 |
|
| 12 |
+
from flask import Flask, jsonify
|
|
|
|
| 13 |
|
| 14 |
+
app = Flask(__name__)
|
|
|
|
| 15 |
|
| 16 |
+
logger.info("Flask app created")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
@app.route('/healthz')
|
| 19 |
def health_check():
|
| 20 |
+
logger.info("Health check called")
|
|
|
|
| 21 |
return "OK", 200
|
| 22 |
|
| 23 |
@app.route('/')
|
| 24 |
def index():
|
| 25 |
+
logger.info("Root endpoint called")
|
| 26 |
+
return jsonify({
|
| 27 |
+
"message": "皮肤科AI诊断助手",
|
| 28 |
+
"status": "running",
|
| 29 |
+
"timestamp": datetime.now().isoformat()
|
| 30 |
+
})
|
| 31 |
+
|
| 32 |
+
@app.route('/test')
|
| 33 |
+
def test():
|
| 34 |
+
return jsonify({
|
| 35 |
+
"test": "success",
|
| 36 |
+
"env_vars": list(os.environ.keys())[:10], # 只显示前10个环境变量
|
| 37 |
+
"working_dir": os.getcwd(),
|
| 38 |
+
"python_version": sys.version
|
| 39 |
+
})
|
| 40 |
+
|
| 41 |
+
logger.info("All routes registered")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
|
| 43 |
if __name__ == '__main__':
|
| 44 |
+
logger.info("Starting Flask app...")
|
| 45 |
+
app.run(host='0.0.0.0', port=8080)
|