Update app.py

app.py

@@ -1,8 +1,16 @@
 
-import os
-import urllib.request
+
 import cv2
 import numpy as np
+import os
+import urllib.request
+
+
+# Set writable directories
+os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
+os.environ["MP_MODELS_PATH"] = "/tmp/mediapipe_models"
+
+
 # Now import libraries
 import mediapipe as mp
 import matplotlib.pyplot as plt
@@ -11,22 +19,24 @@ from flask import Flask, request, jsonify
 from flask_cors import CORS
 import torch.nn.functional as F
 
-# ✅ Set writable directories
-os.environ["MPLCONFIGDIR"] = "/tmp/matplotlib"
-os.environ["MP_MODELS_PATH"] = "/tmp/mediapipe_models"
+# MediaPipe Model Download
+MEDIAPIPE_POSE_MODEL_URL = "https://storage.googleapis.com/mediapipe-models/pose_landmark/pose_landmark_heavy/float16/1/pose_landmark_heavy.tflite"
+MODEL_PATH = os.path.join(os.environ["MP_MODELS_PATH"], "pose_landmark_heavy.tflite")
 
-# ✅ Ensure the directories exist
-os.makedirs("/tmp/matplotlib", exist_ok=True)
-os.makedirs("/tmp/mediapipe_models", exist_ok=True)
 
-# ✅ Manually Download the MediaPipe Model (BEFORE loading MediaPipe)
-MODEL_PATH = "/tmp/mediapipe_models/pose_landmark_heavy.tflite"
-MODEL_URL = "https://storage.googleapis.com/mediapipe-assets/pose_landmark_heavy.tflite"
+# Ensure model directory exists and is writable
+os.makedirs(os.environ["MP_MODELS_PATH"], exist_ok=True)
 
+# Download MediaPipe model if not exists
 if not os.path.exists(MODEL_PATH):
-    print("Downloading MediaPipe model...")
-    urllib.request.urlretrieve(MODEL_URL, MODEL_PATH)
-
+    try:
+        print("Downloading MediaPipe model...")
+        urllib.request.urlretrieve(MEDIAPIPE_POSE_MODEL_URL, MODEL_PATH)
+        print(f"Model downloaded successfully to {MODEL_PATH}")
+    except Exception as e:
+        print(f"Error downloading model: {e}")
+
+# Rest of the existing app.py code remains the same
 app = Flask(__name__)
 CORS(app)