Spaces:
Runtime error
Runtime error
| import whisper | |
| import gradio as gr | |
| import os | |
| import asyncio | |
| import subprocess | |
| # Ensure Playwright is installed and browsers are downloaded | |
| def install_playwright(): | |
| try: | |
| # Install Playwright via pip if not already installed | |
| subprocess.run(["pip", "install", "playwright"], check=True) | |
| # Install the Playwright browsers | |
| subprocess.run(["playwright", "install"], check=True) | |
| print("Playwright and browsers installed.") | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error during Playwright setup: {e}") | |
| exit(1) | |
| # Call the function to install Playwright | |
| install_playwright() | |
| from download_video import download_mp3_playwright | |
| # Function to convert MP4 to MP3 using FFmpeg | |
| def convert_to_mp3(input_file, output_file): | |
| command = ["ffmpeg", "-i", input_file, "-q:a", "0", "-map", "a", output_file] | |
| try: | |
| subprocess.run(command, check=True) | |
| except subprocess.CalledProcessError as e: | |
| print(f"Error converting {input_file} to {output_file}: {e}") | |
| # Function to download the audio, title, and thumbnail from YouTube | |
| async def download_video_info(url): | |
| try: | |
| # Call the async function to download video and get title and thumbnail | |
| title, thumbnail_url = await download_mp3_playwright(url) | |
| audio_file = "downloaded_video.mp4" # Path to the downloaded audio | |
| # Convert MP4 to MP3 before passing to Whisper | |
| convert_to_mp3(audio_file, "downloaded_audio.mp3") | |
| return "downloaded_audio.mp3", title, thumbnail_url | |
| except Exception as e: | |
| return None, None, str(e) | |
| # Function to transcribe the downloaded audio using Whisper | |
| def transcribe_audio(audio_path, model_size="base"): | |
| model = whisper.load_model(model_size) | |
| result = model.transcribe(audio_path) | |
| return result['text'] | |
| # Split logic: First fetch title and thumbnail, then transcribe | |
| async def get_video_info_and_transcribe(youtube_url, model_size="base"): | |
| # Fetch title and thumbnail first | |
| audio_path, title, thumbnail_url = await download_video_info(youtube_url) | |
| # If fetching video info fails | |
| if not audio_path or not os.path.exists(audio_path): | |
| return gr.update(value=f"Error fetching video: {thumbnail_url}"), None, None, None | |
| # Show title and thumbnail to the user while the transcription is happening | |
| title_output = gr.update(value=title) | |
| # Show the thumbnail if available | |
| if thumbnail_url: | |
| thumbnail_output = gr.update(value=thumbnail_url) | |
| else: | |
| thumbnail_output = gr.update(visible=False) # Hide if no thumbnail | |
| # Start transcription | |
| transcription = transcribe_audio(audio_path, model_size) | |
| return title_output, thumbnail_output, gr.update(value=transcription) | |
| # Gradio interface setup using gradio.components | |
| with gr.Blocks() as interface: | |
| with gr.Row(): | |
| youtube_url = gr.Textbox(label="YouTube Link", elem_id="yt_link", scale=5) | |
| model_size = gr.Dropdown(choices=["tiny", "base", "small", "medium", "large"], label="Model Size", value="base", scale=1) | |
| title_output = gr.Textbox(label="Video Title", interactive=False) | |
| with gr.Row(): | |
| thumbnail_output = gr.Image(label="Thumbnail", interactive=False, scale=1) | |
| transcription_output = gr.Textbox(label="Transcription", interactive=False, scale=1) | |
| transcribe_button = gr.Button("Transcribe") | |
| transcribe_button.click( | |
| get_video_info_and_transcribe, | |
| inputs=[youtube_url, model_size], | |
| outputs=[title_output, thumbnail_output, transcription_output] | |
| ) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| interface.launch(share=True) |