Create app.py

app.py

@@ -0,0 +1,86 @@
+import cv2
+import mediapipe as mp
+import numpy as np
+import gradio as gr
+from moviepy import VideoFileClip
+import os
+
+# Initialize MediaPipe Pose
+mp_pose = mp.solutions.pose
+mp_drawing = mp.solutions.drawing_utils
+
+# Create "Meaty" drawing specs for EbSynth
+# High thickness (10) and larger circles (15) for better pixel tracking
+MEATY_CONNECTIONS = mp_drawing.DrawingSpec(color=(0, 255, 0), thickness=10, circle_radius=5)
+MEATY_JOINTS = mp_drawing.DrawingSpec(color=(255, 255, 255), thickness=12, circle_radius=15)
+
+def extract_pose(video_path):
+    if video_path is None:
+        return None
+    
+    # 1. Load Video
+    cap = cv2.VideoCapture(video_path)
+    width  = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
+    height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
+    fps    = cap.get(cv2.CAP_PROP_FPS)
+    
+    output_path = "stickman_output.mp4"
+    temp_output = "temp_no_audio.mp4"
+    
+    # Define Video Writer
+    fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+    out = cv2.VideoWriter(temp_output, fourcc, fps, (width, height))
+    
+    with mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5, model_complexity=1) as pose:
+        while cap.isOpened():
+            ret, frame = cap.read()
+            if not ret:
+                break
+            
+            # Convert to RGB for MediaPipe
+            rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            results = pose.process(rgb_frame)
+            
+            # Create Black Canvas (The "Clear" Background)
+            canvas = np.zeros((height, width, 3), dtype=np.uint8)
+            
+            # Draw Meaty Stickman
+            if results.pose_landmarks:
+                mp_drawing.draw_landmarks(
+                    canvas, 
+                    results.pose_landmarks, 
+                    mp_pose.POSE_CONNECTIONS,
+                    landmark_drawing_spec=MEATY_JOINTS,
+                    connection_drawing_spec=MEATY_CONNECTIONS
+                )
+            
+            out.write(canvas)
+            
+    cap.release()
+    out.release()
+    
+    # 2. Merge Original Audio back using MoviePy v2.x syntax
+    try:
+        original_clip = VideoFileClip(video_path)
+        stickman_clip = VideoFileClip(temp_output)
+        
+        # New MoviePy v2.x method: with_audio instead of set_audio
+        final_video = stickman_clip.with_audio(original_clip.audio)
+        final_video.write_videofile(output_path, codec="libx264", audio_codec="aac")
+        
+        return output_path
+    except Exception as e:
+        print(f"Audio merge failed: {e}")
+        return temp_output # Fallback to silent video
+
+# Gradio UI
+interface = gr.Interface(
+    fn=extract_pose,
+    inputs=gr.Video(label="Upload Dancing Clip (15-30s)"),
+    outputs=gr.Video(label="Meaty Stickman Output"),
+    title="Pose-to-Stickman Extractor",
+    description="Extracts high-thickness pose data for EbSynth & Motion Transfer."
+)
+
+if __name__ == "__main__":
+    interface.launch()