commit

app.py

@@ -6,21 +6,11 @@ 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(
@@ -28,6 +18,7 @@ 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])
@@ -102,13 +93,15 @@ 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 in async context
+    # Process the image
     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")
+