# Import required libraries
import os
import re
import logging
import whisper
from pytube import YouTube
import gradio as gr

# Setup logging
logging.basicConfig(level=logging.INFO)

# Load the Whisper model
model = whisper.load_model("base")

def download_audio_from_youtube(url):
    """
    Download the audio from a YouTube video and return the path to the audio file.
    """
    yt = YouTube(url)
    video = yt.streams.filter(only_audio=True).first()
    out_file = video.download(output_path=".")
    return out_file

def get_text(url):
    """
    Transcribe the audio from a YouTube video and return the transcript.
    """
    if not url:
        return ''
    
    out_file = download_audio_from_youtube(url)
    file_stats = os.stat(out_file)
    
    logging.info(f'Size of audio file in Bytes: {file_stats.st_size}')
    
    if file_stats.st_size > 30000000:
        logging.error('Videos for transcription on this space are limited to about 1.5 hours...')
        return ''
    
    base, ext = os.path.splitext(out_file)
    new_file = base + '.mp3'
    os.rename(out_file, new_file)
    
    result = model.transcribe(new_file)
    return result['text'].strip()

def create_gradio_interface():
    """
    Create and launch a Gradio interface for transcribing YouTube videos.
    """
    with gr.Blocks() as demo:
        gr.Markdown("<h1><center>Trascribe Videos using <a href=https://openai.com/blog/whisper/ target=_blank>Whisper</a></center></h1>")
        gr.Markdown("<center>Enter the link of any YouTube video.</center>")
        input_text_url = gr.Textbox(placeholder='Youtube video URL', label='YouTube URL')
        result_button_transcribe = gr.Button('Transcribe')
        output_text_transcribe = gr.Textbox(placeholder='Transcript of the YouTube video.', label='Transcript')
        
        result_button_transcribe.click(get_text, inputs=input_text_url, outputs=output_text_transcribe)
    
    demo.queue().launch(debug=True)

# Launch the Gradio interface
if __name__ == "__main__":
    create_gradio_interface()