Update app.py

app.py

@@ -3,12 +3,17 @@ import mediapipe as mp
 import numpy as np
 import gradio as gr
 import tempfile
+import shutil
 import os
 
+# Initialize MediaPipe Pose
 mp_pose = mp.solutions.pose
 mp_drawing = mp.solutions.drawing_utils
-pose = mp_pose.Pose(static_image_mode=False, min_detection_confidence=0.5, min_tracking_confidence=0.5)
+pose = mp_pose.Pose(static_image_mode=False,
+                    min_detection_confidence=0.5,
+                    min_tracking_confidence=0.5)
 
+# Function to calculate joint angles
 def calculate_angle(a, b, c):
     a = np.array(a)
     b = np.array(b)
@@ -19,12 +24,18 @@ def calculate_angle(a, b, c):
     angle = np.arccos(np.clip(cosine_angle, -1.0, 1.0))
     return np.degrees(angle)
 
+# Main video processing function
 def detect_pose_video(video_path, max_duration=20):
     try:
         if not os.path.exists(video_path):
             return None, "Error: Video file does not exist."
 
-        cap = cv2.VideoCapture(video_path)
+        # Copy uploaded video to a temporary safe path
+        temp_input = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        temp_input.close()
+        shutil.copy(video_path, temp_input.name)
+
+        cap = cv2.VideoCapture(temp_input.name)
         if not cap.isOpened():
             return None, "Error: Cannot open video file."
 
@@ -33,11 +44,10 @@ def detect_pose_video(video_path, max_duration=20):
         height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT) or 480)
         max_frames = int(fps * max_duration)
 
-        # 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()
-
+        # Output temporary file
+        temp_output = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
+        temp_output.close()
+        out_path = temp_output.name
         fourcc = cv2.VideoWriter_fourcc(*'mp4v')
         out = cv2.VideoWriter(out_path, fourcc, fps, (width, height))
 
@@ -47,7 +57,7 @@ def detect_pose_video(video_path, max_duration=20):
             if not ret:
                 break
 
-            # Resize if too big
+            # Resize if frame is too large
             max_dim = 640
             h, w, _ = frame.shape
             if max(h, w) > max_dim:
@@ -66,6 +76,7 @@ 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]]
@@ -83,7 +94,27 @@ def detect_pose_video(video_path, max_duration=20):
 
         cap.release()
         out.release()
+        os.remove(temp_input.name)  # Clean up input temp file
+
         return out_path, "Video processed successfully."
 
     except Exception as e:
-        return None, f"Error: {str(e)}"
+        return None, f"Runtime Error: {str(e)}"
+
+# Gradio interface with submit button enabled only after video upload
+with gr.Blocks() as demo:
+    gr.Markdown("## Human Pose Estimation on Video\nUpload a video (max 20 seconds). Submit is enabled only after upload.")
+
+    video_input = gr.Video(label="Upload Video")
+    status_output = gr.Textbox(label="Status")
+    video_output = gr.Video(label="Annotated Video")
+    submit_btn = gr.Button("Submit", interactive=False)
+
+    # Enable submit button only if a video is uploaded
+    def enable_submit(video):
+        return True if video else False
+
+    video_input.change(enable_submit, inputs=[video_input], outputs=[submit_btn])
+    submit_btn.click(detect_pose_video, inputs=[video_input], outputs=[video_output, status_output])
+
+demo.launch()