File size: 5,659 Bytes
86f402d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 | """
Analysis Routes - Image analysis with SSE streaming
"""
from fastapi import APIRouter, Query, HTTPException
from fastapi.responses import StreamingResponse, FileResponse
from pathlib import Path
import json
import tempfile
from backend.services.analysis_service import get_analysis_service
from data.case_store import get_case_store
router = APIRouter()
@router.get("/gradcam")
def get_gradcam_by_path(path: str = Query(...)):
"""Serve a temp visualization image (GradCAM or comparison overlay)"""
if not path:
raise HTTPException(status_code=400, detail="No path provided")
temp_dir = Path(tempfile.gettempdir()).resolve()
resolved_path = Path(path).resolve()
if not str(resolved_path).startswith(str(temp_dir)):
raise HTTPException(status_code=403, detail="Access denied")
allowed_suffixes = ("_gradcam.png", "_comparison.png")
if not any(resolved_path.name.endswith(s) for s in allowed_suffixes):
raise HTTPException(status_code=400, detail="Invalid image path")
if resolved_path.exists():
return FileResponse(str(resolved_path), media_type="image/png")
raise HTTPException(status_code=404, detail="Image not found")
@router.post("/{patient_id}/lesions/{lesion_id}/images/{image_id}/analyze")
async def analyze_image(
patient_id: str,
lesion_id: str,
image_id: str,
question: str = Query(None)
):
"""Analyze an image with SSE streaming"""
store = get_case_store()
# Verify image exists
img = store.get_image(patient_id, lesion_id, image_id)
if not img:
raise HTTPException(status_code=404, detail="Image not found")
if not img.image_path:
raise HTTPException(status_code=400, detail="Image has no file uploaded")
service = get_analysis_service()
async def generate():
try:
for chunk in service.analyze(patient_id, lesion_id, image_id, question):
yield f"data: {json.dumps(chunk)}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
yield f"data: {json.dumps(f'[ERROR]{str(e)}[/ERROR]')}\n\n"
yield "data: [DONE]\n\n"
return StreamingResponse(
generate(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
}
)
@router.post("/{patient_id}/lesions/{lesion_id}/images/{image_id}/confirm")
async def confirm_diagnosis(
patient_id: str,
lesion_id: str,
image_id: str,
confirmed: bool = Query(...),
feedback: str = Query(None)
):
"""Confirm or reject diagnosis and get management guidance"""
service = get_analysis_service()
async def generate():
try:
for chunk in service.confirm(patient_id, lesion_id, image_id, confirmed, feedback):
yield f"data: {json.dumps(chunk)}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
yield f"data: {json.dumps(f'[ERROR]{str(e)}[/ERROR]')}\n\n"
yield "data: [DONE]\n\n"
return StreamingResponse(
generate(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
}
)
@router.post("/{patient_id}/lesions/{lesion_id}/images/{image_id}/compare")
async def compare_to_previous(
patient_id: str,
lesion_id: str,
image_id: str
):
"""Compare this image to the previous one in the timeline"""
store = get_case_store()
# Get current and previous images
current_img = store.get_image(patient_id, lesion_id, image_id)
if not current_img:
raise HTTPException(status_code=404, detail="Image not found")
previous_img = store.get_previous_image(patient_id, lesion_id, image_id)
if not previous_img:
raise HTTPException(status_code=400, detail="No previous image to compare")
service = get_analysis_service()
async def generate():
try:
for chunk in service.compare_images(
patient_id, lesion_id,
previous_img.image_path,
current_img.image_path,
image_id
):
yield f"data: {json.dumps(chunk)}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
yield f"data: {json.dumps(f'[ERROR]{str(e)}[/ERROR]')}\n\n"
yield "data: [DONE]\n\n"
return StreamingResponse(
generate(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
}
)
@router.post("/{patient_id}/lesions/{lesion_id}/chat")
async def chat_message(
patient_id: str,
lesion_id: str,
message: dict
):
"""Send a chat message with SSE streaming response"""
store = get_case_store()
lesion = store.get_lesion(patient_id, lesion_id)
if not lesion:
raise HTTPException(status_code=404, detail="Lesion not found")
service = get_analysis_service()
content = message.get("content", "")
async def generate():
try:
for chunk in service.chat_followup(patient_id, lesion_id, content):
yield f"data: {json.dumps(chunk)}\n\n"
yield "data: [DONE]\n\n"
except Exception as e:
yield f"data: {json.dumps(f'[ERROR]{str(e)}[/ERROR]')}\n\n"
yield "data: [DONE]\n\n"
return StreamingResponse(
generate(),
media_type="text/event-stream",
headers={
"Cache-Control": "no-cache",
"Connection": "keep-alive",
}
)
|