Update app.py

app.py

@@ -13,7 +13,6 @@ 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)
@@ -24,13 +23,12 @@ 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."
 
-        # Copy uploaded video to a temporary safe path
+        # Copy video to temp file
         temp_input = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
         temp_input.close()
         shutil.copy(video_path, temp_input.name)
@@ -44,7 +42,6 @@ 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)
 
-        # Output temporary file
         temp_output = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
         temp_output.close()
         out_path = temp_output.name
@@ -57,7 +54,7 @@ def detect_pose_video(video_path, max_duration=20):
             if not ret:
                 break
 
-            # Resize if frame is too large
+            # Resize if too large
             max_dim = 640
             h, w, _ = frame.shape
             if max(h, w) > max_dim:
@@ -76,7 +73,7 @@ def detect_pose_video(video_path, max_duration=20):
                     mp_drawing.DrawingSpec(color=(0,0,255), thickness=2)
                 )
 
-                # Example: left elbow angle
+                # 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]]
@@ -94,14 +91,14 @@ def detect_pose_video(video_path, max_duration=20):
 
         cap.release()
         out.release()
-        os.remove(temp_input.name)  # Clean up input temp file
+        os.remove(temp_input.name)  # Clean up input
 
         return out_path, "Video processed successfully."
 
     except Exception as e:
         return None, f"Runtime Error: {str(e)}"
 
-# Gradio interface with submit button enabled only after video upload
+# Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("## Human Pose Estimation on Video\nUpload a video (max 20 seconds). Submit is enabled only after upload.")
 
@@ -110,9 +107,9 @@ with gr.Blocks() as demo:
     video_output = gr.Video(label="Annotated Video")
     submit_btn = gr.Button("Submit", interactive=False)
 
-    # Enable submit button only if a video is uploaded
+    # Correct: enable button using .update()
     def enable_submit(video):
-        return True if video else False
+        return submit_btn.update(interactive=True) if video else submit_btn.update(interactive=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])