Update app.py

app.py

@@ -1,46 +1,77 @@
 import gradio as gr
-import librosa
-import librosa.display
-import numpy as np
-import soundfile as sf
-import io
+from pytube import YouTube
+import os
+import re
 
-def trim_audio(audio_url, start_time, end_time):
-    """Trims an audio file based on the provided start and end times."""
-    try:
-        # Load audio from URL
-        y, sr = librosa.load(audio_url)
-
-        # Check for valid start and end times
-        if start_time < 0 or end_time > len(y) / sr or start_time >= end_time:
-            return "Invalid start or end time. Please check your input."
-
-
-        start_sample = int(start_time * sr)
-        end_sample = int(end_time * sr)
-        trimmed_audio = y[start_sample:end_sample]
-
-        # Save the trimmed audio to a BytesIO object
-        output_audio = io.BytesIO()
-        sf.write(output_audio, trimmed_audio, sr, format='wav')
-        output_audio.seek(0)  # Reset the file pointer to the beginning
-
-        return gr.Audio(output_audio, label="Trimmed Audio")
+def sanitize_filename(title):
+    """Remove invalid characters from filename"""
+    return re.sub(r'[<>:"/\\|?*]', '', title)
 
+def download_audio(url, output_dir="downloads"):
+    try:
+        # Create output directory if it doesn't exist
+        if not os.path.exists(output_dir):
+            os.makedirs(output_dir)
+        
+        # Create YouTube object
+        yt = YouTube(url)
+        
+        # Get audio stream
+        audio_stream = yt.streams.filter(only_audio=True).first()
+        
+        # Sanitize the title for filename
+        safe_title = sanitize_filename(yt.title)
+        
+        # Download the audio
+        output_file = audio_stream.download(
+            output_path=output_dir,
+            filename=f"{safe_title}.mp3"
+        )
+        
+        return {
+            "status": "success",
+            "message": f"Successfully downloaded: {safe_title}",
+            "title": yt.title,
+            "author": yt.author,
+            "duration": f"{yt.length // 60}:{yt.length % 60:02d}",
+            "file_path": output_file
+        }
+    
     except Exception as e:
-        return f"An error occurred: {e}"
+        return {
+            "status": "error",
+            "message": f"Error: {str(e)}",
+            "title": "",
+            "author": "",
+            "duration": "",
+            "file_path": ""
+        }
 
+def process_url(url):
+    if not url.strip():
+        return "Please enter a YouTube URL"
+    
+    result = download_audio(url)
+    
+    if result["status"] == "error":
+        return result["message"]
+    
+    return f"""Download successful!
+Title: {result['title']}
+Author: {result['author']}
+Duration: {result['duration']}
+Saved to: {result['file_path']}"""
 
+# Create Gradio interface
 iface = gr.Interface(
-    fn=trim_audio,
-    inputs=[
-        gr.Textbox(label="Audio URL", lines=1),
-        gr.Number(label="Start Time (seconds)", value=0),
-        gr.Number(label="End Time (seconds)", value=5),
-    ],
-    outputs=gr.Audio(label="Output Audio"),
-    title="Audio Trimmer",
-    description="Enter the URL of an audio file and specify the start and end times to trim it.",
+    fn=process_url,
+    inputs=gr.Textbox(label="YouTube URL", placeholder="Enter YouTube video URL here..."),
+    outputs=gr.Textbox(label="Status"),
+    title="YouTube to MP3 Downloader",
+    description="Enter a YouTube URL to download the audio as MP3",
+    examples=[["https://www.youtube.com/watch?v=dQw4w9WgXcQ"]],
+    theme=gr.themes.Base()
 )
 
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()