Update app.py

app.py

@@ -1,3 +1,13 @@
+# This script is designed for Hugging Face Spaces
+# Dependencies are specified in requirements.txt in the Space's repository root
+# Expected dependencies:
+# gradio>=4.0.0
+# opencv-python>=4.8.0
+# requests>=2.28.0
+# ultralytics>=8.2.0
+# pytubefix>=6.0.0
+# numpy>=1.23.0
+
 import sys
 import gradio as gr
 import os
@@ -6,6 +16,7 @@ import cv2
 import requests
 from ultralytics import YOLO
 
+# Remove extra CLI arguments that Spaces might pass
 sys.argv = [arg for arg in sys.argv if arg != "--import"]
 
 model = YOLO("yolo11n-pose.pt")
@@ -65,8 +76,7 @@ def process_input(uploaded_file, youtube_link, image_url, sensitivity):
                 return None, None, None, "Invalid video properties detected."
             
             output_path = os.path.join(tempfile.gettempdir(), f"out_{os.urandom(8).hex()}.mp4")
-            # Use H.264 codec which is more widely supported
-            fourcc = cv2.VideoWriter_fourcc(*'avc1')  # Changed from 'mp4v' to 'avc1'
+            fourcc = cv2.VideoWriter_fourcc(*'avc1')
             out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
             
             if not out.isOpened():
@@ -94,7 +104,7 @@ def process_input(uploaded_file, youtube_link, image_url, sensitivity):
             if processed_frames == 0:
                 return None, None, None, "No frames processed from video."
                 
-            if not os.path.exists(output_path) or os.path.getsize(output_path) < 1024:  # Check if file is too small
+            if not os.path.exists(output_path) or os.path.getsize(output_path) < 1024:
                 return None, None, None, f"Output video created but too small ({os.path.getsize(output_path)} bytes) - processing failed."
                 
             return output_path, None, output_path, f"Video processed successfully! ({processed_frames}/{frame_count} frames)"
@@ -119,7 +129,6 @@ def process_input(uploaded_file, youtube_link, image_url, sensitivity):
                 except:
                     pass
 
-# Gradio interface (unchanged)
 with gr.Blocks(css="""
 .result_img > img {
   width: 100%;