first commit

.gitignore

@@ -0,0 +1 @@
+.ipynb_checkpoints

README.md

@@ -10,4 +10,6 @@ pinned: false
 license: unlicense
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+## YouTube Sprite Generator
+
+Generate sprites from a YouTube URL

app.py

@@ -0,0 +1,74 @@
+import yt_dlp
+import ffmpeg
+import os
+import shutil
+import gradio as gr
+
+def download_video(url):
+    ydl_opts = {
+        'format': 'bestvideo[height<=480]+bestaudio/best[height<=480]',
+        'merge_output_format': 'mp4',
+        'quiet': True,
+    }
+    with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+        info_dict = ydl.extract_info(url, download=False)
+        video_id = info_dict.get("id", None)
+        filename = f"{video_id}.mp4"
+        ydl_opts['outtmpl'] = filename
+        ydl = yt_dlp.YoutubeDL(ydl_opts)
+        ydl.download([url])
+    return filename
+
+def create_sprite(video_path, sprite_output="sprite.jpg", frame_interval=10, num_frames=20):
+    frames_folder = "frames"
+    if not os.path.exists(frames_folder):
+        os.makedirs(frames_folder)
+    
+    frames_pattern = os.path.join(frames_folder, "frame_%03d.jpg")
+    
+    # Extract frames from the video at the specified interval
+    (
+        ffmpeg
+        .input(video_path)
+        .filter('fps', fps=1/frame_interval)
+        .output(frames_pattern, vframes=num_frames)
+        .run(overwrite_output=True)
+    )
+
+    # Calculate the tile layout
+    rows = (num_frames + 3) // 4  # Ensure at least one row
+
+    # Create the sprite from the extracted frames
+    (
+        ffmpeg
+        .input(os.path.join(frames_folder, "frame_*.jpg"), pattern_type='glob', framerate=1)
+        .filter('scale', 320, -1)
+        .filter('tile', f"4x{rows}")
+        .output(sprite_output)
+        .run(overwrite_output=True)
+    )
+
+    # Clean up frame files
+    shutil.rmtree(frames_folder)
+    
+    return sprite_output
+
+def process_youtube_url(url):
+    video_path = download_video(url)
+    sprite_output = "sprite.jpg"
+    sprite_path = create_sprite(video_path, sprite_output)
+    if os.path.exists(video_path):
+        os.remove(video_path)
+    
+    return sprite_path
+
+iface = gr.Interface(
+    fn=process_youtube_url,
+    inputs=gr.Textbox(label="YouTube URL"),
+    outputs=gr.Image(type="filepath", label="Sprite Image"),
+    title="YouTube Video Sprite Generator",
+    description="Enter a YouTube URL to download the video and create a sprite image of frames."
+)
+
+iface.queue()
+iface.launch()

packages.txt

@@ -0,0 +1 @@
+ffmpeg

requirements.txt

@@ -0,0 +1,2 @@
+ffmpeg-python
+yt_dlp