Fix

app.py

@@ -218,21 +218,31 @@ async def process_image(file: UploadFile = File(...)):
         shutil.copyfileobj(file.file, buffer)
 
     # Load the YOLO model
-    model = YOLO("yolo11n-pose_float16.tflite")
-
-    print("Model loaded successfully")
+    try:
+        model = YOLO("yolo11n-pose_float16.tflite")
+        print("Model loaded successfully")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Model loading failed: {str(e)}")
 
     # Run YOLO detection on the image
     try:
         results = model(
             source=image_path,
-            show=False,     # Set to False for API use
+            show=False,
             save=False
         )
-        resultsKeypointsXY = results[0].keypoints.xy
-        confidences = results[0].boxes.conf
-    
+
+        if not results or len(results) == 0 or results[0].keypoints is None or results[0].keypoints.xy is None:
+            return JSONResponse(content={"message": "No keypoints detected"}, status_code=200)
+
+        keypoints = results[0].keypoints.xy
+        confidences = results[0].boxes.conf if results[0].boxes is not None else []
+
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
 
-    return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY.tolist(), "confidences": confidences.tolist()})
+    return JSONResponse(content={
+        "message": "Image processed successfully",
+        "KeypointsXY": keypoints.tolist(),
+        "confidences": confidences.tolist()
+    })

main.py

@@ -218,21 +218,31 @@ async def process_image(file: UploadFile = File(...)):
         shutil.copyfileobj(file.file, buffer)
 
     # Load the YOLO model
-    model = YOLO("yolo11n-pose_float16.tflite")
-
-    print("Model loaded successfully")
+    try:
+        model = YOLO("yolo11n-pose_float16.tflite")
+        print("Model loaded successfully")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Model loading failed: {str(e)}")
 
     # Run YOLO detection on the image
     try:
         results = model(
             source=image_path,
-            show=False,     # Set to False for API use
+            show=False,
             save=False
         )
-        resultsKeypointsXY = results[0].keypoints.xy
-        confidences = results[0].boxes.conf
-    
+
+        if not results or len(results) == 0 or results[0].keypoints is None or results[0].keypoints.xy is None:
+            return JSONResponse(content={"message": "No keypoints detected"}, status_code=200)
+
+        keypoints = results[0].keypoints.xy
+        confidences = results[0].boxes.conf if results[0].boxes is not None else []
+
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"YOLO processing error: {str(e)}")
 
-    return JSONResponse(content={"message": "Image processed successfully", "KeypointsXY": resultsKeypointsXY.tolist(), "confidences": confidences.tolist()})
+    return JSONResponse(content={
+        "message": "Image processed successfully",
+        "KeypointsXY": keypoints.tolist(),
+        "confidences": confidences.tolist()
+    })