Update app.py

app.py

@@ -1,94 +1,63 @@
 import os
-import subprocess
-import instaloader
 import gradio as gr
+from yt_dlp import YoutubeDL
 from pydantic import BaseModel
 
 class Config(BaseModel):
     class Config:
         arbitrary_types_allowed = True
 
-def download_instagram_reel(url, output_folder="downloads"):
-    loader = instaloader.Instaloader()
-    loader.download_video_thumbnails = False
+def download_reel_audio(url, output_folder="downloads"):
+    """
+    Download audio from Instagram reel using yt-dlp
+    """
     try:
-        # Extract shortcode from URL
-        shortcode = url.split("/")[-2]
         # Ensure output folder exists
         os.makedirs(output_folder, exist_ok=True)
-        # Download reel
-        loader.download_post(instaloader.Post.from_shortcode(loader.context, shortcode), target=output_folder)
-        # Find downloaded video file
-        for file in os.listdir(output_folder):
-            if file.endswith(".mp4"):
-                return os.path.join(output_folder, file)
-        return None
-    except Exception as e:
-        return f"Error downloading reel: {e}"
-
-def extract_audio(video_path, output_folder="downloads"):
-    try:
-        # Ensure output folder exists
-        os.makedirs(output_folder, exist_ok=True)
-        
-        # Generate output audio path
-        audio_path = os.path.join(output_folder, 
-                                os.path.splitext(os.path.basename(video_path))[0] + ".mp3")
         
-        # FFmpeg command to extract audio
-        command = [
-            'ffmpeg',
-            '-i', video_path,  # Input file
-            '-vn',  # Disable video
-            '-acodec', 'libmp3lame',  # Use MP3 codec
-            '-ab', '192k',  # Audio bitrate
-            '-ar', '44100',  # Audio sample rate
-            '-y',  # Overwrite output file if it exists
-            audio_path
-        ]
-        
-        # Run FFmpeg command
-        result = subprocess.run(command, 
-                              stdout=subprocess.PIPE, 
-                              stderr=subprocess.PIPE)
-        
-        if result.returncode != 0:
-            return f"FFmpeg error: {result.stderr.decode()}"
-        
-        return audio_path
-    except Exception as e:
-        return f"Error extracting audio: {e}"
+        # Configure yt-dlp options
+        ydl_opts = {
+            'format': 'bestaudio/best',
+            'outtmpl': os.path.join(output_folder, '%(title)s.%(ext)s'),
+            'postprocessors': [{
+                'key': 'FFmpegExtractAudio',
+                'preferredcodec': 'mp3',
+                'preferredquality': '192',
+            }],
+            'quiet': True,
+            'no_warnings': True
+        }
+
+        # Download the audio
+        with YoutubeDL(ydl_opts) as ydl:
+            info = ydl.extract_info(url, download=True)
+            audio_path = os.path.join(output_folder, f"{info['title']}.mp3")
+            return "Audio downloaded successfully!", audio_path
 
-def process_reel(url):
-    # Download reel
-    video_path = download_instagram_reel(url)
-    if isinstance(video_path, str) and video_path.endswith(".mp4"):
-        # Extract audio
-        audio_path = extract_audio(video_path)
-        if isinstance(audio_path, str) and audio_path.endswith(".mp3"):
-            # Clean up video file after extraction
-            try:
-                os.remove(video_path)
-            except:
-                pass
-            return "Audio extracted successfully!", audio_path
-        else:
-            return f"Failed to extract audio: {audio_path}", None
-    else:
-        return f"Failed to download reel: {video_path}", None
+    except Exception as e:
+        return f"Error downloading audio: {str(e)}", None
 
 # Gradio Interface
 interface = gr.Interface(
-    fn=process_reel,
-    inputs=gr.Textbox(label="Instagram Reel URL", 
-                     placeholder="Enter the Instagram reel URL here"),
+    fn=download_reel_audio,
+    inputs=gr.Textbox(
+        label="Instagram Reel URL",
+        placeholder="Enter the Instagram reel URL here (e.g., https://www.instagram.com/reel/...)"
+    ),
     outputs=[
         gr.Textbox(label="Status"),
         gr.Audio(label="Downloaded Audio")
     ],
     title="Instagram Reel to Audio Downloader",
-    description="Enter the URL of an Instagram reel to download the audio as an MP3 file.",
-    theme="compact"
+    description="""
+    Enter the URL of an Instagram reel to download its audio as an MP3 file.
+    This tool uses yt-dlp which is more reliable and doesn't require authentication.
+    """,
+    theme="default",
+    examples=[
+        ["https://www.instagram.com/reel/example1"],
+        ["https://www.instagram.com/reel/example2"]
+    ]
 )
 
 if __name__ == "__main__":