Update app/routes/upload.py

app/routes/upload.py

@@ -254,6 +254,27 @@ async def update_report(
     }
 
 
+@router.get("/upload/file/{file_id}")
+async def get_raw_file(file_id: str, db: Session = Depends(get_db)):
+    """Stream raw file from PostgreSQL vault or disk storage."""
+    from fastapi.responses import Response
+    doc = db.query(Document).filter(Document.file_id == file_id).first()
+    if doc and doc.file_data:
+        ftype = (doc.file_type or "").lower()
+        content_type = "image/jpeg" if ftype in ["jpg", "jpeg", "image/jpeg", "image/jpg"] else "application/pdf" if ftype in ["pdf", "application/pdf"] else "image/png" if ftype in ["png", "image/png"] else "application/octet-stream"
+        return Response(content=doc.file_data, media_type=content_type)
+
+    if os.path.exists(UPLOAD_FOLDER):
+        for fname in os.listdir(UPLOAD_FOLDER):
+            if fname.startswith(file_id):
+                file_path = os.path.join(UPLOAD_FOLDER, fname)
+                with open(file_path, "rb") as f:
+                    data = f.read()
+                return Response(content=data, media_type="image/jpeg")
+
+    raise HTTPException(status_code=404, detail="Document file not found in EMR vault")
+
+
 @router.delete("/upload/document/{file_id}")
 async def delete_document(
     file_id: str,