Update app.py

app.py

@@ -2,13 +2,21 @@ import gradio as gr
 import os
 import numpy as np
 from PIL import Image
-from pytube import YouTube
+import re
 import demoji
 from textblob import TextBlob
 from faker import Faker
 import pandas as pd
 from difflib import SequenceMatcher
 
+# Try importing pytube with better error handling
+try:
+    from pytube import YouTube
+    from pytube.exceptions import RegexMatchError, VideoUnavailable
+    PYTUBE_AVAILABLE = True
+except ImportError:
+    PYTUBE_AVAILABLE = False
+
 # Function for Image Mirroring
 def mirror_image(input_img):
     if input_img is None:
@@ -16,28 +24,80 @@ def mirror_image(input_img):
     mirrored_img = Image.fromarray(input_img).transpose(Image.FLIP_LEFT_RIGHT)
     return mirrored_img
 
-# Function for YouTube to MP3 Download
+# Improved YouTube URL validation
+def is_valid_youtube_url(url):
+    if not url:
+        return False
+    
+    # Common YouTube URL patterns
+    youtube_regex = (
+        r'(https?://)?(www\.)?'
+        r'(youtube|youtu|youtube-nocookie)\.(com|be)/'
+        r'(watch\?v=|embed/|v/|.+\?v=)?([^&=%\?]{11})')
+    
+    match = re.match(youtube_regex, url)
+    return match is not None
+
+# Function for YouTube to MP3 Download with improved error handling
 def download_youtube_mp3(url, destination="."):
+    if not PYTUBE_AVAILABLE:
+        return "Error: pytube library is not available. Please install it with 'pip install pytube'."
+    
     if not url:
         return "Please enter a valid URL"
     
+    if not is_valid_youtube_url(url):
+        return "Error: The URL does not appear to be a valid YouTube URL"
+    
     try:
         yt = YouTube(url)
-        video = yt.streams.filter(only_audio=True).first()
+        
+        # Check if video is available
+        title = yt.title  # This will raise an exception if the video is unavailable
+        
+        # Get the highest quality audio stream
+        audio_stream = yt.streams.filter(only_audio=True).order_by('abr').desc().first()
+        
+        if not audio_stream:
+            return "Error: No audio stream found for this video"
         
         if not destination:
             destination = "."
             
-        out_file = video.download(output_path=destination)
+        # Ensure destination directory exists
+        if not os.path.exists(destination):
+            try:
+                os.makedirs(destination)
+            except Exception as e:
+                return f"Error creating destination directory: {str(e)}"
+        
+        # Download the file
+        out_file = audio_stream.download(output_path=destination)
         
         # Save the file with .mp3 extension
         base, ext = os.path.splitext(out_file)
         new_file = base + '.mp3'
+        
+        # Handle file already exists
+        if os.path.exists(new_file):
+            os.remove(new_file)
+            
         os.rename(out_file, new_file)
         
-        return f"{yt.title} has been successfully downloaded in .mp3 format at {new_file}"
+        return f"Success! '{yt.title}' has been downloaded in MP3 format to: {new_file}"
+    
+    except VideoUnavailable:
+        return "Error: This video is unavailable or private"
+    except RegexMatchError:
+        return "Error: The YouTube URL is not valid"
     except Exception as e:
-        return f"Error: {str(e)}"
+        error_message = str(e)
+        if "429" in error_message:
+            return "Error: Too many requests. YouTube is rate-limiting downloads. Please try again later."
+        elif "403" in error_message:
+            return "Error: Access forbidden. YouTube may have restricted this content."
+        else:
+            return f"Error: {error_message}. If this is a recurring issue, YouTube may have updated their site. Try updating pytube with 'pip install --upgrade pytube'."
 
 # Function for Emoji Detection
 def detect_emojis(text):
@@ -149,10 +209,14 @@ with gr.Blocks(title="MultiToolBox") as app:
         # YouTube to MP3 Tab
         with gr.Tab("YouTube to MP3"):
             gr.Markdown("### Download YouTube Videos as MP3")
-            yt_url = gr.Textbox(label="YouTube URL")
+            
+            if not PYTUBE_AVAILABLE:
+                gr.Markdown("⚠️ **Warning:** pytube library is not installed. This feature will not work properly.")
+            
+            yt_url = gr.Textbox(label="YouTube URL", placeholder="https://www.youtube.com/watch?v=dQw4w9WgXcQ")
             yt_destination = gr.Textbox(label="Destination Folder (leave empty for current directory)", placeholder=".")
             yt_download_btn = gr.Button("Download")
-            yt_output = gr.Textbox(label="Result")
+            yt_output = gr.Textbox(label="Result", lines=3)
             
             yt_download_btn.click(fn=download_youtube_mp3, inputs=[yt_url, yt_destination], outputs=yt_output)
             
@@ -163,6 +227,8 @@ with gr.Blocks(title="MultiToolBox") as app:
             3. Click "Download" to convert and save as MP3
             
             **Example:** https://www.youtube.com/watch?v=dQw4w9WgXcQ
+            
+            **Note:** If you encounter errors, try running `pip install --upgrade pytube` to update the YouTube library
             """)
         
         # Emoji Detection Tab
@@ -221,7 +287,7 @@ with gr.Blocks(title="MultiToolBox") as app:
             **Examples:**
             - 135 (Disarium number)
             - 89 (Disarium number: 8^1 + 9^2 = 8 + 81 = 89)
-            - 175 (Not a Disarium number)
+            - 165 (Not a Disarium number)
             """)
         
         # Fake Data Generator Tab