fix: validate invalid dicom uploads before processing

Made-with: Cursor

Author: AbbyShenCs

apps/backend/app/routers/reports.py

@@ -7,6 +7,7 @@
 from typing import List, Optional
 from .data import data
 from pyaslreport import generate_report, get_bids_metadata
+from pyaslreport.main import get_dicom_header
 from pyaslreport.enums import ModalityTypeValues
 from fastapi.responses import FileResponse
 from weasyprint import HTML
@@ -89,6 +90,16 @@ async def get_report_dicom(
         except Exception as e:
             print(f"Error reading DICOM file {file.filename}: {e}")
 
+    try:
+        get_dicom_header(data["dicom_dir"])
+    except (TypeError, ValueError, OSError, FileNotFoundError) as e:
+        print(f"Invalid DICOM upload: {e}")
+        await remove_dir(base_dir)
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail="No valid DICOM files provided"
+        )
+
     try:
         metadata, asl_context = get_bids_metadata(data)
         print("Generated BIDS metadata:", metadata)

apps/backend/tests/test_report.py

@@ -23,12 +23,12 @@ def test_get_report_dicom_with_invalid_file(tmp_path):
     file_path.write_text("not a dicom")
     with open(file_path, "rb") as f:
         response = client.post(
-            "/report/process/dicom",
+            "/api/report/process/dicom",
             files={"dcm_files": ("not_a_dicom.txt", f, "text/plain")},
             data={"modality": "ASL"}
         )
-    # Should still return 500 due to invalid dicom
-    assert response.status_code in [500, 200]
+    assert response.status_code == 400
+    assert response.json()["detail"] == "No valid DICOM files provided"
 
 def test_report_pdf_endpoint():
     # Minimal valid report_data for rendering