Update app.py

app.py

@@ -1,29 +1,23 @@
 import os
+import subprocess
 import instaloader
-from moviepy.editor import VideoFileClip
 import gradio as gr
 from pydantic import BaseModel
 
-# Custom Pydantic Config to avoid schema generation issues
 class Config(BaseModel):
     class Config:
         arbitrary_types_allowed = 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
-
+    loader.download_video_thumbnails = False
     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"):
@@ -32,21 +26,39 @@ def download_instagram_reel(url, output_folder="downloads"):
     except Exception as e:
         return f"Error downloading reel: {e}"
 
-# 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)
-
-        # 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)
+        
+        # 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}"
 
-# Gradio interface function
 def process_reel(url):
     # Download reel
     video_path = download_instagram_reel(url)
@@ -54,16 +66,22 @@ def process_reel(url):
         # 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 "Failed to extract audio.", None
+            return f"Failed to extract audio: {audio_path}", None
     else:
-        return "Failed to download reel.", None
+        return f"Failed to download reel: {video_path}", None
 
 # Gradio Interface
 interface = gr.Interface(
     fn=process_reel,
-    inputs=gr.Textbox(label="Instagram Reel URL", placeholder="Enter the Instagram reel URL here"),
+    inputs=gr.Textbox(label="Instagram Reel URL", 
+                     placeholder="Enter the Instagram reel URL here"),
     outputs=[
         gr.Textbox(label="Status"),
         gr.Audio(label="Downloaded Audio")
@@ -73,6 +91,5 @@ interface = gr.Interface(
     theme="compact"
 )
 
-# Launch the interface
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()