Spaces:
Running
Running
File size: 5,199 Bytes
89bbef2 4a3f789 89bbef2 ab45a2c 89bbef2 246c64e 89bbef2 ab45a2c 89bbef2 246c64e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
"""
YouTube tab for Video Model Studio UI.
Handles downloading videos from YouTube URLs.
"""
import gradio as gr
import logging
from pathlib import Path
from typing import Dict, Any, Optional
from vms.utils import BaseTab
logger = logging.getLogger(__name__)
class YouTubeTab(BaseTab):
"""YouTube tab for downloading videos from YouTube"""
def __init__(self, app_state):
super().__init__(app_state)
self.id = "youtube_tab"
self.title = "Download from YouTube"
# Initialize components that will be shared from parent
if "components" not in self.__dict__:
self.components = {}
self.components["import_status"] = None
self.components["enable_automatic_video_split"] = None
self.components["enable_automatic_content_captioning"] = None
def create(self, parent=None) -> gr.Tab:
"""Create the YouTube tab UI components"""
with gr.Tab(self.title, id=self.id) as tab:
with gr.Column():
with gr.Row():
gr.Markdown("## Import a YouTube video")
with gr.Row():
with gr.Column():
with gr.Row():
gr.Markdown("You can use a YouTube video as reference, by pasting its URL here:")
with gr.Row():
gr.Markdown("Please be aware of the [known limitations](https://stackoverflow.com/questions/78160027/how-to-solve-http-error-400-bad-request-in-pytube) and [issues](https://stackoverflow.com/questions/79226520/pytube-throws-http-error-403-forbidden-since-a-few-days)")
with gr.Column():
self.components["youtube_url"] = gr.Textbox(
label="Import YouTube Video",
placeholder="https://www.youtube.com/watch?v=..."
)
with gr.Row():
self.components["youtube_download_btn"] = gr.Button("Download YouTube Video", variant="primary")
return tab
def connect_events(self) -> None:
"""Connect event handlers to UI components"""
# Check if required shared components exist before connecting events
if not self.components.get("import_status"):
logger.warning("import_status component is not set in YouTubeTab")
return
if not self.components.get("enable_automatic_video_split"):
logger.warning("enable_automatic_video_split component is not set in YouTubeTab")
return
if not self.components.get("enable_automatic_content_captioning"):
logger.warning("enable_automatic_content_captioning component is not set in YouTubeTab")
return
# Only try to access custom_prompt_prefix if the caption_tab exists
custom_prompt_prefix = None
try:
if hasattr(self.app.tabs, "caption_tab") and "custom_prompt_prefix" in self.app.tabs["caption_tab"].components:
custom_prompt_prefix = self.app.tabs["caption_tab"].components["custom_prompt_prefix"]
except (AttributeError, KeyError):
logger.warning("Could not access custom_prompt_prefix component")
# Check if we have access to project_tabs_component
if hasattr(self.app, "project_tabs_component"):
tabs_component = self.app.project_tabs_component
else:
logger.warning("project_tabs_component not found in app, using None for tab switching")
tabs_component = None
# YouTube download event
download_event = self.components["youtube_download_btn"].click(
fn=self.download_youtube_with_splitting,
inputs=[self.components["youtube_url"], self.components["enable_automatic_video_split"]],
outputs=[self.components["import_status"]]
)
# Add success handler if all components exist
if hasattr(self.app.tabs, "import_tab") and custom_prompt_prefix is not None:
try:
# Add the success handler
download_event.success(
fn=self.app.tabs["import_tab"].on_import_success,
inputs=[
self.components["enable_automatic_video_split"],
self.components["enable_automatic_content_captioning"],
custom_prompt_prefix
],
outputs=[
tabs_component,
self.components["import_status"]
]
)
except (AttributeError, KeyError) as e:
logger.error(f"Error connecting success handler in YouTubeTab: {str(e)}")
# Continue without the success handler
def download_youtube_with_splitting(self, url, enable_splitting):
"""Download YouTube video with splitting option"""
return self.app.importing.download_youtube_video(url, enable_splitting, gr.Progress()) |