Update app.py

app.py

@@ -4,7 +4,6 @@ import os
 from dotenv import load_dotenv
 import re
 import requests
-import shutil
 
 # Load environment variables
 load_dotenv()
@@ -16,10 +15,7 @@ INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 def clean_url(url):
     """Clean Instagram URL and extract shortcode"""
     try:
-        # Remove query parameters and fragments
         cleaned_url = url.split('?')[0].split('#')[0].strip('/')
-        
-        # Extract shortcode using regex
         patterns = [
             r'instagram.com/reel/([A-Za-z0-9_-]+)',
             r'instagram.com/p/([A-Za-z0-9_-]+)',
@@ -31,7 +27,6 @@ def clean_url(url):
             if match:
                 return match.group(1)
         
-        # If patterns don't match, try getting the last segment
         segments = cleaned_url.split('/')
         if segments:
             return segments[-1]
@@ -47,13 +42,9 @@ def download_file(url, path):
         response = requests.get(url, stream=True)
         response.raise_for_status()
         
-        # Get file size for progress bar
         file_size = int(response.headers.get('content-length', 0))
-        
-        # Create progress bar
         progress_bar = st.progress(0)
         
-        # Download with progress updates
         with open(path, 'wb') as f:
             if file_size == 0:
                 f.write(response.content)
@@ -74,16 +65,14 @@ def download_file(url, path):
 def download_reel(url):
     """Download Instagram reel"""
     if not url:
-        return "Please enter a URL"
+        return None, "Please enter a URL"
     
-    # Clean URL and extract shortcode
     shortcode = clean_url(url)
     if not shortcode:
-        return "Could not extract valid shortcode from URL"
+        return None, "Could not extract valid shortcode from URL"
     
     st.info(f"Extracted shortcode: {shortcode}")
     
-    # Initialize loader with debugging
     loader = instaloader.Instaloader(
         download_videos=True,
         download_video_thumbnails=False,
@@ -94,70 +83,58 @@ def download_reel(url):
     )
     
     try:
-        # Login
         if INSTAGRAM_USERNAME and INSTAGRAM_PASSWORD:
             st.info(f"Logging in as {INSTAGRAM_USERNAME}...")
             loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
             st.success("Login successful!")
         else:
-            return "Missing Instagram credentials"
+            return None, "Missing Instagram credentials"
         
-        # Create download directory
         download_path = os.path.join('reels', shortcode)
         os.makedirs(download_path, exist_ok=True)
-        st.info(f"Download directory created: {os.path.abspath(download_path)}")
         
-        # Get post metadata
         st.info("Fetching post metadata...")
         post = instaloader.Post.from_shortcode(loader.context, shortcode)
         
-        # Debug post information
-        st.info(f"Post type: {post.typename}")
-        st.info(f"Is video: {post.is_video}")
-        
         if post.is_video:
-            # Get video URL
             video_url = post.video_url
-            st.info("Video URL obtained")
-            
-            # Download video file
             video_path = os.path.join(download_path, f"{shortcode}.mp4")
-            st.info("Starting direct video download...")
+            st.info("Starting download...")
             
             if download_file(video_url, video_path):
-                # Verify file exists and has size
                 if os.path.exists(video_path):
                     size = os.path.getsize(video_path)
                     if size > 0:
-                        return f"""Download successful!
-                        Location: {video_path}
-                        Size: {size/1024/1024:.2f} MB"""
+                        return video_path, f"Download successful! Size: {size/1024/1024:.2f} MB"
                     else:
-                        return "Download failed: File is empty"
+                        return None, "Download failed: File is empty"
                 else:
-                    return "Download failed: File not created"
+                    return None, "Download failed: File not created"
             else:
-                return "Failed to download video file"
+                return None, "Failed to download video file"
         else:
-            return "Post is not a video"
+            return None, "Post is not a video"
             
     except instaloader.exceptions.InstaloaderException as e:
-        return f"Instagram error: {str(e)}"
+        return None, f"Instagram error: {str(e)}"
     except Exception as e:
-        return f"Unexpected error: {str(e)}\nType: {type(e)}"
+        return None, f"Unexpected error: {str(e)}"
 
 def main():
     st.set_page_config(
         page_title="Instagram Reel Downloader",
-        page_icon="📱"
+        page_icon="📱",
+        initial_sidebar_state="collapsed"
     )
     
     st.title('📱 Instagram Reel Downloader')
     
-    # Show current working directory and permissions
-    st.info(f"Working directory: {os.getcwd()}")
-    if os.path.exists("reels"):
-        st.info(f"Reels directory permissions: {oct(os.stat('reels').st_mode)[-3:]}")
+    st.markdown("""
+    ### Instructions:
+    1. Enter the Instagram reel URL
+    2. Click 'Download Reel'
+    3. After successful download, click 'Download Video' button to save the file
+    """)
     
     # Input for Reel URL
     url = st.text_input('Enter the Instagram Reel URL', 
@@ -166,20 +143,19 @@ def main():
     # Download button
     if st.button('Download Reel', type='primary'):
         if url:
-            with st.spinner('Downloading reel...'):
-                result = download_reel(url)
-                st.write(result)
+            with st.spinner('Processing...'):
+                file_path, message = download_reel(url)
+                st.write(message)
                 
-                # Show directory contents
-                if os.path.exists("reels"):
-                    st.write("\nContents of reels directory:")
-                    for root, dirs, files in os.walk("reels"):
-                        level = root.replace("reels", "").count(os.sep)
-                        indent = " " * 4 * level
-                        st.write(f"{indent}{os.path.basename(root)}/")
-                        subindent = " " * 4 * (level + 1)
-                        for f in files:
-                            st.write(f"{subindent}{f}")
+                # If download was successful, show download button
+                if file_path and os.path.exists(file_path):
+                    with open(file_path, 'rb') as file:
+                        st.download_button(
+                            label="Download Video",
+                            data=file,
+                            file_name=os.path.basename(file_path),
+                            mime="video/mp4"
+                        )
         else:
             st.warning('Please enter a URL.')