Update app.py

app.py

@@ -139,4 +139,41 @@ async def detect(request: Request):
         )
 
     result = process_image_np(image_np)
-    return result
+    return result
+
+
+from fastapi import Response
+
+# ----------------------------
+# Convert Image To JPEG (For Browser Display)
+# ----------------------------
+
+@app.post("/convert")
+async def convert(request: Request):
+    body = await request.body()
+
+    if not body:
+        return JSONResponse(
+            {"error": "Empty request body"},
+            status_code=400
+        )
+
+    image_np = decode_image(body)
+
+    if image_np is None:
+        return JSONResponse(
+            {"error": "Unsupported or invalid image format"},
+            status_code=400
+        )
+
+    # Convert BGR -> RGB
+    image_rgb = cv2.cvtColor(image_np, cv2.COLOR_BGR2RGB)
+    pil_image = Image.fromarray(image_rgb)
+
+    buffer = io.BytesIO()
+    pil_image.save(buffer, format="JPEG", quality=90)
+
+    return Response(
+        content=buffer.getvalue(),
+        media_type="image/jpeg"
+    )