Update app.py

app.py

@@ -1,7 +1,9 @@
 import streamlit as st
 import instaloader
 import os
+import re
 from dotenv import load_dotenv
+import time
 
 # Load environment variables
 load_dotenv()
@@ -13,9 +15,20 @@ INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 def initialize_instaloader():
     """Initialize and login to Instaloader"""
     try:
-        loader = instaloader.Instaloader()
+        loader = instaloader.Instaloader(
+            download_videos=True,
+            download_video_thumbnails=False,
+            download_geotags=False,
+            download_comments=False,
+            save_metadata=False,
+            compress_json=False,
+            max_connection_attempts=3
+        )
+        
         if INSTAGRAM_USERNAME and INSTAGRAM_PASSWORD:
+            st.info(f"Attempting to login as {INSTAGRAM_USERNAME}...")
             loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+            st.success("Login successful!")
             return loader
         else:
             st.error("Instagram credentials not found in environment variables")
@@ -23,6 +36,9 @@ def initialize_instaloader():
     except instaloader.exceptions.BadCredentialsException:
         st.error("Invalid Instagram credentials")
         return None
+    except instaloader.exceptions.TwoFactorAuthRequiredException:
+        st.error("2FA is enabled on this account. Please use an account without 2FA.")
+        return None
     except Exception as e:
         st.error(f"Error initializing Instaloader: {str(e)}")
         return None
@@ -30,9 +46,18 @@ def initialize_instaloader():
 def extract_shortcode(url):
     """Extract shortcode from Instagram URL"""
     try:
-        # Handle both full URLs and shortened ones
-        parts = url.strip('/').split('/')
-        return parts[-1] if parts[-1] else parts[-2]
+        # Handle different URL formats
+        patterns = [
+            r'instagram.com/reel/([A-Za-z0-9_-]+)',
+            r'instagram.com/p/([A-Za-z0-9_-]+)',
+            r'/([A-Za-z0-9_-]{11})[/]?$'
+        ]
+        
+        for pattern in patterns:
+            match = re.search(pattern, url)
+            if match:
+                return match.group(1)
+        return None
     except Exception:
         return None
 
@@ -41,26 +66,44 @@ def download_reel(url):
     if not url:
         return "Please enter a URL"
     
+    # Initialize loader with debug message
+    st.info("Initializing downloader...")
     loader = initialize_instaloader()
     if not loader:
         return "Failed to initialize downloader"
     
+    # Extract and validate shortcode
     shortcode = extract_shortcode(url)
     if not shortcode:
-        return "Invalid URL format"
+        return "Invalid URL format. Please provide a valid Instagram reel URL."
+    
+    st.info(f"Attempting to download reel with shortcode: {shortcode}")
     
     try:
         # Create download directory
         os.makedirs("reels", exist_ok=True)
         
-        # Download the post
+        # Add delay to prevent rate limiting
+        time.sleep(2)
+        
+        # Download the post with progress updates
+        st.info("Fetching post metadata...")
         post = instaloader.Post.from_shortcode(loader.context, shortcode)
+        
+        st.info("Starting download...")
         loader.download_post(post, target=f"reels/{shortcode}")
         
         return f"Reel downloaded successfully to reels/{shortcode}"
     
     except instaloader.exceptions.InstaloaderException as e:
-        return f"Instagram error: {str(e)}"
+        error_msg = str(e)
+        if "404" in error_msg:
+            return "Post not found. The reel might be private or deleted."
+        elif "429" in error_msg:
+            return "Too many requests. Please try again later."
+        elif "login" in error_msg.lower():
+            return "Session expired. Please try logging in again."
+        return f"Instagram error: {error_msg}"
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
 
@@ -72,7 +115,7 @@ def main():
     
     st.title('📱 Instagram Reel Downloader')
     
-    # Add instructions
+    # Add instructions and debug mode toggle
     st.markdown("""
     ### Instructions:
     1. Paste an Instagram reel URL
@@ -82,10 +125,21 @@ def main():
     Note: This tool requires valid Instagram credentials to work.
     """)
     
+    # Debug mode toggle
+    debug_mode = st.checkbox('Enable debug mode')
+    
     # Input for Reel URL
     url = st.text_input('Enter the Instagram Reel URL', 
                        placeholder='https://www.instagram.com/reel/...')
     
+    if debug_mode and url:
+        st.info(f"URL provided: {url}")
+        shortcode = extract_shortcode(url)
+        if shortcode:
+            st.info(f"Extracted shortcode: {shortcode}")
+        else:
+            st.warning("Could not extract shortcode from URL")
+    
     # Download button
     if st.button('Download Reel', type='primary'):
         if url: