Update app.py

app.py

@@ -1,74 +1,72 @@
-import gradio as gr
-import yt_dlp
 import os
+import instaloader
+from moviepy.editor import VideoFileClip
+import gradio as gr
 
-def download_video_as_mp3(video_url, output_dir="downloads"):
-    """
-    Downloads a YouTube video as an MP3 file using yt-dlp.
-
-    :param video_url: The URL of the YouTube video.
-    :param output_dir: Directory where the MP3 file will be saved.
-    :return: A tuple containing the filepath of the downloaded MP3 and a status message.
-             Returns (None, error_message) if an error occurred.
-    """
-    if not os.path.exists(output_dir):
-        os.makedirs(output_dir)
-
-    ydl_opts = {
-        'format': 'bestaudio/best',
-        'postprocessors': [{
-            'key': 'FFmpegExtractAudio',
-            'preferredcodec': 'mp3',
-            'preferredquality': '192',
-        }],
-        'outtmpl': os.path.join(output_dir, '%(title)s.%(ext)s'),
-        'quiet': True, # Set to True for cleaner Gradio output, remove if you want yt-dlp's progress in terminal
-        'noplaylist': True,
-    }
+# Function to download Instagram reel
+def download_instagram_reel(url, output_folder="downloads"):
+    loader = instaloader.Instaloader()
+    loader.download_video_thumbnails = False  # Disable thumbnail downloads
 
     try:
-        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info_dict = ydl.extract_info(video_url, download=True)
-            video_title = info_dict.get('title', 'video')
-            output_filepath = os.path.join(output_dir, f"{video_title}.mp3")
-        status_message = "Download complete!"
-        return output_filepath, status_message
+        # Extract shortcode from URL
+        shortcode = url.split("/")[-2]
 
-    except Exception as e:
-        error_message = f"An error occurred: {e}"
-        return None, error_message
+        # 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)
 
-def gradio_interface(video_url):
-    """
-    Gradio interface function to download YouTube video as MP3.
-
-    :param video_url: YouTube video URL from Gradio Textbox.
-    :return: A tuple containing the filepath of the downloaded MP3 (for Gradio File output) and status message (for Gradio Textbox).
-    """
-    if not video_url:
-        return None, "Please enter a YouTube video URL."
+        # 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}"
 
-    output_directory = "downloads" # Default output directory
+# Function to extract audio from video
+def extract_audio(video_path, output_folder="downloads"):
+    try:
+        # Ensure output folder exists
+        os.makedirs(output_folder, exist_ok=True)
 
-    filepath, status_message = download_video_as_mp3(video_url, output_directory)
+        # Extract audio
+        video = VideoFileClip(video_path)
+        audio_path = os.path.join(output_folder, os.path.splitext(os.path.basename(video_path))[0] + ".mp3")
+        video.audio.write_audiofile(audio_path)
+        return audio_path
+    except Exception as e:
+        return f"Error extracting audio: {e}"
 
-    return filepath, status_message
+# Gradio interface function
+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"):
+            return "Audio extracted successfully!", audio_path
+        else:
+            return "Failed to extract audio.", None
+    else:
+        return "Failed to download reel.", None
 
+# Gradio Interface
+interface = gr.Interface(
+    fn=process_reel,
+    inputs=gr.Textbox(label="Instagram Reel URL", placeholder="Enter the Instagram reel URL here"),
+    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"
+)
 
+# Launch the interface
 if __name__ == "__main__":
-    iface = gr.Interface(
-        fn=gradio_interface,
-        inputs=gr.Textbox(label="YouTube Video URL", placeholder="Enter YouTube URL here"),
-        outputs=[
-            gr.File(label="Downloaded MP3 File"),
-            gr.Textbox(label="Status"),
-        ],
-        title="YouTube to MP3 Downloader",
-        description="Enter a YouTube video URL and click 'Submit' to download it as MP3. The file will be saved in the 'downloads' directory.",
-        examples=[
-            ["https://www.youtube.com/watch?v=dQw4w9WgXcQ"],
-            ["https://www.youtube.com/watch?v=your_video_id"],
-        ]
-    )
-    iface.launch()
+    interface.launch()