Update app.py

app.py

@@ -1,9 +1,9 @@
 import streamlit as st
 import instaloader
 import os
-import re
 from dotenv import load_dotenv
 import time
+import glob
 
 # Load environment variables
 load_dotenv()
@@ -12,54 +12,27 @@ load_dotenv()
 INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
 INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
-def initialize_instaloader():
-    """Initialize and login to Instaloader"""
-    try:
-        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")
-            return None
-    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
-
-def extract_shortcode(url):
-    """Extract shortcode from Instagram URL"""
-    try:
-        # 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
+def verify_download(shortcode):
+    """Verify that files were actually downloaded"""
+    download_path = f"reels/{shortcode}"
+    
+    # Check if directory exists
+    if not os.path.exists(download_path):
+        return False, "Download directory not found"
+    
+    # Look for video files (mp4)
+    video_files = glob.glob(f"{download_path}/*.mp4")
+    
+    if video_files:
+        # Get the full path of the first video file
+        video_path = video_files[0]
+        file_size = os.path.getsize(video_path)
+        return True, {
+            'path': video_path,
+            'size': f"{file_size / (1024*1024):.2f} MB",
+            'exists': os.path.exists(video_path)
+        }
+    return False, "No video files found in download directory"
 
 def download_reel(url):
     """Download Instagram reel"""
@@ -68,87 +41,87 @@ def download_reel(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. Please provide a valid Instagram reel URL."
-    
-    st.info(f"Attempting to download reel with shortcode: {shortcode}")
+    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,
+        dirname_pattern="reels/{target}"  # Explicitly set download directory
+    )
     
     try:
+        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!")
+        else:
+            st.error("Instagram credentials not found")
+            return "Missing credentials"
+            
+        # Extract shortcode
+        shortcode = url.strip('/').split('/')[-1]
+        if not shortcode:
+            shortcode = url.strip('/').split('/')[-2]
+            
+        st.info(f"Attempting to download reel with shortcode: {shortcode}")
+        
         # Create download directory
-        os.makedirs("reels", exist_ok=True)
+        os.makedirs(f"reels/{shortcode}", exist_ok=True)
         
-        # Add delay to prevent rate limiting
-        time.sleep(2)
+        # Get absolute path
+        abs_download_path = os.path.abspath(f"reels/{shortcode}")
+        st.info(f"Download directory: {abs_download_path}")
         
-        # Download the post with progress updates
+        # Download the post
         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}")
+        st.info(f"Starting download of post type: {post.typename}")
+        loader.download_post(post, target=shortcode)
+        
+        # Verify the download
+        success, result = verify_download(shortcode)
+        
+        if success:
+            return f"""Download completed:
+            - File: {result['path']}
+            - Size: {result['size']}
+            - File exists: {result['exists']}"""
+        else:
+            return f"Download failed: {result}"
         
-        return f"Reel downloaded successfully to reels/{shortcode}"
-    
     except instaloader.exceptions.InstaloaderException as 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}"
+        return f"Instagram error: {str(e)}\nCurrent working directory: {os.getcwd()}"
     except Exception as e:
         return f"An unexpected error occurred: {str(e)}"
 
 def main():
-    st.set_page_config(
-        page_title="Instagram Reel Downloader",
-        page_icon="📱"
-    )
+    st.set_page_config(page_title="Instagram Reel Downloader", page_icon="📱")
     
     st.title('📱 Instagram Reel Downloader')
     
-    # Add instructions and debug mode toggle
-    st.markdown("""
-    ### Instructions:
-    1. Paste an Instagram reel URL
-    2. Click 'Download Reel'
-    3. Wait for the download to complete
-    
-    Note: This tool requires valid Instagram credentials to work.
-    """)
-    
-    # Debug mode toggle
-    debug_mode = st.checkbox('Enable debug mode')
+    # Show current working directory
+    st.text(f"Working directory: {os.getcwd()}")
     
     # 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:
             with st.spinner('Downloading reel...'):
                 result = download_reel(url)
-                if "successfully" in result:
-                    st.success(result)
-                else:
-                    st.error(result)
+                st.text(result)  # Show detailed result
+                
+                # List contents of download directory
+                if "reels" in os.listdir():
+                    st.text("Contents of reels directory:")
+                    for item in os.listdir("reels"):
+                        st.text(f"- {item}")
         else:
             st.warning('Please enter a URL.')