Pushing to space Dicom

main.py

@@ -1,7 +1,11 @@
 from fastapi import FastAPI, File, UploadFile, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
+from fastapi.responses import JSONResponse
 from generate_report import generate_report
 from utils import convert_to_png
+from utils import dicom_to_png
+import pydicom
+import tempfile
 import io
 
 
@@ -41,6 +45,76 @@ async def upload_image(file: UploadFile = File(...)):
         raise HTTPException(status_code=500, detail=str(e))
     
 
+@app.post("/generate_report_dicom")
+async def generate_report_dicom(file: UploadFile = File(...)):
+    """
+    Generate radiology report from DICOM file.
+    Checks if the DICOM is an X-ray, converts it to PNG, and generates a report.
+    """
+    try:
+        # Validate file type
+        if not file.filename.lower().endswith(('.dcm', '.dicom')):
+            raise HTTPException(status_code=400, detail="File must be a DICOM file (.dcm or .dicom)")
+        
+        # Read DICOM file
+        file_content = await file.read()
+        
+        # Create a temporary file to store DICOM data
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.dcm') as temp_file:
+            temp_file.write(file_content)
+            temp_file_path = temp_file.name
+        
+        try:
+            # Read DICOM data
+            dicom_data = pydicom.dcmread(temp_file_path)
+            
+            # # Check if it's an X-ray
+            # if not is_xray_dicom(dicom_data):
+            #     return JSONResponse(
+            #         {"error": "DICOM file does not appear to be an X-ray image"},
+            #         status_code=400
+            #     )
+            
+            # Convert DICOM to PNG
+            image = dicom_to_png(dicom_data)
+
+            report= generate_report(image)  # Generate report using the image
+            
+            # Get additional DICOM metadata for context
+            metadata = {}
+            if hasattr(dicom_data, 'PatientID'):
+                metadata['patient_id'] = str(dicom_data.PatientID)
+            if hasattr(dicom_data, 'StudyDate'):
+                metadata['study_date'] = str(dicom_data.StudyDate)
+            if hasattr(dicom_data, 'Modality'):
+                metadata['modality'] = str(dicom_data.Modality)
+            if hasattr(dicom_data, 'BodyPartExamined'):
+                metadata['body_part'] = str(dicom_data.BodyPartExamined)
+            
+            return JSONResponse({
+                "generated_report": report,
+                "dicom_metadata": metadata,
+                "image_info": {
+                    "width": image.width,
+                    "height": image.height,
+                    "mode": image.mode
+                }
+            })
+            
+        finally:
+            # Clean up temporary file
+            import os
+            try:
+                os.unlink(temp_file_path)
+            except:
+                pass
+                
+    except HTTPException:
+        raise
+    except Exception as e:
+        return JSONResponse({"error": str(e)}, status_code=500)
+    
+
 
 
     

utils.py

@@ -36,4 +36,20 @@ async def convert_to_png(file: UploadFile) -> Image.Image:
         image = Image.fromarray(pixel_array).convert("RGB")
         return image
     
-    raise HTTPException(status_code=400, detail="Unsupported media type")
+    raise HTTPException(status_code=400, detail="Unsupported media type")
+
+def dicom_to_png(ds):
+    pixel_array = ds.pixel_array
+
+    # Normalize if needed (handle 16-bit images)
+    if pixel_array.dtype != np.uint8:
+        # Scale to 0-255
+        pixel_array = pixel_array.astype(np.float32)
+        pixel_array -= pixel_array.min()
+        pixel_array /= pixel_array.max()
+        pixel_array *= 255.0
+        pixel_array = pixel_array.astype(np.uint8)
+
+    # Convert to PIL Image
+    img = Image.fromarray(pixel_array).convert("RGB")
+    return img