commit

app.py

@@ -6,11 +6,21 @@ import mediapipe as mp
 from io import BytesIO
 from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import Response
+from fastapi.middleware.cors import CORSMiddleware  # Add CORS support
 from PIL import Image
 
 # Initialize FastAPI app
 app = FastAPI()
 
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
 # Initialize Mediapipe Pose model
 mp_pose = mp.solutions.pose
 pose = mp_pose.Pose(
@@ -18,7 +28,6 @@ pose = mp_pose.Pose(
     min_detection_confidence=0.5,
     min_tracking_confidence=0.5
 )
-
 # Function to calculate angles between points
 def calculate_angle(a, b, c):
     ab = (b[0] - a[0], b[1] - a[1])
@@ -93,15 +102,13 @@ def process_frame(image):
 # API Route to receive an image and return processed image
 @app.post("/upload")
 async def upload_image(file: UploadFile = File(...)):
-    # Read the image
     contents = await file.read()
     image = Image.open(BytesIO(contents))
     image = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2BGR)
     
-    # Process the image
+    # Process the image in async context
     processed_image = process_frame(image)
     
     # Encode processed image to return
     _, buffer = cv2.imencode(".jpg", processed_image)
-    return Response(content=buffer.tobytes(), media_type="image/jpeg")
-
+    return Response(content=buffer.tobytes(), media_type="image/jpeg")