Added StreamingResponse

app.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import JSONResponse
+from fastapi.responses import StreamingResponse, JSONResponse
 from fastapi.middleware.cors import CORSMiddleware
 from transformers import DetrImageProcessor, DetrForObjectDetection
 from PIL import Image, ImageDraw
@@ -12,7 +12,7 @@ app = FastAPI()
 # Add CORS middleware to allow communication with external clients
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Change this to the specific domain in production
+    allow_origins=["*"],  # Change this to specific domains in production
     allow_methods=["*"],
     allow_headers=["*"],
 )
@@ -35,7 +35,8 @@ def detect_accident(image):
     for box, label, score in zip(results["boxes"], results["labels"], results["scores"]):
         x_min, y_min, x_max, y_max = box
         draw.rectangle((x_min, y_min, x_max, y_max), outline="red", width=3)
-        draw.text((x_min, y_min), f"{label}: {score:.2f}", fill="red")
+        label_name = model.config.id2label[label.item()]
+        draw.text((x_min, y_min), f"{label_name}: {score:.2f}", fill="red")
     
     return image
 
@@ -45,7 +46,7 @@ async def process_frame(file: UploadFile = File(...)):
     try:
         # Read and preprocess image
         image = Image.open(io.BytesIO(await file.read()))
-        image = image.resize((256, int(image.height * 256 / image.width)))  # Resize while maintaining aspect ratio
+        image = image.convert("RGB")  # Ensure compatibility with the model
 
         # Detect accidents
         processed_image = detect_accident(image)
@@ -55,10 +56,8 @@ async def process_frame(file: UploadFile = File(...)):
         processed_image.save(img_byte_arr, format="JPEG")
         img_byte_arr.seek(0)
 
-        return JSONResponse(
-            content={"status": "success", "message": "Frame processed successfully"},
-            media_type="image/jpeg"
-        )
+        # Return the image as a streaming response
+        return StreamingResponse(img_byte_arr, media_type="image/jpeg")
     except Exception as e:
         return JSONResponse(content={"status": "error", "message": str(e)}, status_code=500)
 
@@ -75,6 +74,7 @@ if __name__ == "__main__":
 
 
 
+
     
 
 # import gradio as gr