implement yt-dlp

app.py

@@ -1,61 +1,55 @@
 import whisper
-from pytube import YouTube
+import yt_dlp
 import gradio as gr
 import os
 import re
-import logging
 
-logging.basicConfig(level=logging.INFO)
 model = whisper.load_model("base")
 
-def get_text(url):
-    #try:
-    if url != '':
-        output_text_transcribe = ''
-
-    yt = YouTube(url)
-    #video_length = yt.length --- doesn't work anymore - using byte file size of the audio file instead now
-    #if video_length < 5400:
-    video = yt.streams.filter(only_audio=True).first()
-    out_file=video.download(output_path=".")
+def get_audio(url):
+    try:
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'noplaylist': True,
+            'quiet': True,
+            'outtmpl': '%(title)s.%(ext)s'  # Specify output template to get the file path
+        }
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            # Use 'requested_downloads' to get the downloaded file path
+            audio_file = ydl.prepare_filename(info)
+            return audio_file
+    except Exception as e:
+        raise gr.Error(f"Exception: {e}")
 
-    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:
-        base, ext = os.path.splitext(out_file)
-        new_file = base+'.mp3'
-        os.rename(out_file, new_file)
-        a = new_file
-    
-        result = model.transcribe(a)
-        return result['text'].strip()
-    else:
-        logging.error('Videos for transcription on this space are limited to about 1.5 hours. Sorry about this limit but some joker thought they could stop this tool from working by transcribing many extremely long videos. Please visit https://steve.digital to contact me about this space.')
-    #finally:
-    #    raise gr.Error("Exception: There was a problem transcribing the audio.")
+def get_text(url):
+    try:
+        if url != '':
+            audio_file = get_audio(url)
+            result = model.transcribe(audio_file)
+            return result['text'].strip()
+        else:
+            return "Please enter a YouTube video URL."
+    except Exception as e:
+        raise gr.Error(f"Exception: {e}")
 
 def get_summary(article):
-    first_sentences = ' '.join(re.split(r'(?<=[.:;])\s', article)[:5])
-    b = summarizer(first_sentences, min_length = 20, max_length = 120, do_sample = False)
-    b = b[0]['summary_text'].replace(' .', '.').strip()
-    return b
+    try:
+        first_sentences = ' '.join(re.split(r'(?<=[.:;])\s', article)[:5])
+        return first_sentences
+    except Exception as e:
+        raise gr.Error(f"Exception: {e}")
 
 with gr.Blocks() as demo:
     gr.Markdown("<h1><center>Free Fast YouTube URL Video-to-Text using <a href=https://openai.com/blog/whisper/ target=_blank>OpenAI's Whisper</a> Model</center></h1>")
-    #gr.Markdown("<center>Enter the link of any YouTube video to generate a text transcript of the video and then create a summary of the video transcript.</center>")
     gr.Markdown("<center>Enter the link of any YouTube video to generate a text transcript of the video.</center>")
     gr.Markdown("<center><b>'Whisper is a neural net that approaches human level robustness and accuracy on English speech recognition.'</b></center>")
-    gr.Markdown("<center>Transcription takes 5-10 seconds per minute of the video (bad audio/hard accents slow it down a bit). #patience<br />If you have time while waiting, drop a ♥️ and check out my <a href=https://www.artificial-intelligence.blog target=_blank>AI blog</a> (opens in new tab).</center>")
+    gr.Markdown("<center>Transcription takes 5-10 seconds per minute of the video (bad audio/hard accents slow it down a bit). #patience<br />If you have time while waiting, check out my <a href=https://www.artificial-intelligence.blog target=_blank>AI blog</a> (opens in new tab).</center>")
     
-    input_text_url = gr.Textbox(placeholder='Youtube video URL', label='YouTube URL')
-    result_button_transcribe = gr.Button('Transcribe')
+    input_text_url = gr.Textbox(placeholder='Youtube video URL', label='URL')
+    result_button_transcribe = gr.Button('1. Transcribe')
     output_text_transcribe = gr.Textbox(placeholder='Transcript of the YouTube video.', label='Transcript')
-    
-    #result_button_summary = gr.Button('2. Create Summary')
-    #output_text_summary = gr.Textbox(placeholder='Summary of the YouTube video transcript.', label='Summary')
-    
-    result_button_transcribe.click(get_text, inputs = input_text_url, outputs = output_text_transcribe)
-    #result_button_summary.click(get_summary, inputs = output_text_transcribe, outputs = output_text_summary)
 
-demo.queue(default_enabled = True).launch(debug = True)
+    result_button_transcribe.click(get_text, inputs=input_text_url, outputs=output_text_transcribe)
+
+    demo.queue(default_enabled=True).launch(debug=True)