Update app.py

app.py

@@ -2,6 +2,7 @@ import cv2
 import mediapipe as mp
 import numpy as np
 import gradio as gr
+import tempfile
 import os
 
 mp_pose = mp.solutions.pose
@@ -27,13 +28,16 @@ def detect_pose_video(video_path, max_duration=20):
         if not cap.isOpened():
             return None, "Error: Cannot open video file."
 
-        # Video properties
         fps = cap.get(cv2.CAP_PROP_FPS) or 20.0
         width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH) or 640)
         height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT) or 480)
         max_frames = int(fps * max_duration)
 
-        out_path = "annotated_video.mp4"
+        # Use tempfile for a safe path Hugging Face can access
+        tmp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        out_path = tmp_file.name
+        tmp_file.close()
+
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         out = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
 
@@ -43,7 +47,7 @@ def detect_pose_video(video_path, max_duration=20):
             if not ret:
                 break
 
-            # Resize frame if too big
+            # Resize if too big
             max_dim = 640
             h, w, _ = frame.shape
             if max(h, w) > max_dim:
@@ -62,7 +66,6 @@ def detect_pose_video(video_path, max_duration=20):
                     mp_drawing.DrawingSpec(color=(0,0,255), thickness=2)
                 )
 
-                # Example: left elbow angle
                 landmarks = results.pose_landmarks.landmark
                 shoulder = [landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].x * frame.shape[1],
                             landmarks[mp_pose.PoseLandmark.LEFT_SHOULDER.value].y * frame.shape[0]]
@@ -84,15 +87,3 @@ def detect_pose_video(video_path, max_duration=20):
 
     except Exception as e:
         return None, f"Error: {str(e)}"
-
-# Gradio interface
-demo = gr.Interface(
-    fn=detect_pose_video,
-    inputs=gr.Video(label="Upload Video"),
-    outputs=[gr.Video(label="Annotated Video"), gr.Textbox(label="Status")],
-    title="Human Pose Estimation on Video",
-    description="Upload a video (max 20 seconds will be processed) and see pose landmarks & joint angles."
-)
-
-if __name__ == "__main__":
-    demo.launch()