Update app.py

app.py

@@ -1,45 +1,102 @@
 import streamlit as st
 import instaloader
 import os
+from dotenv import load_dotenv
 
-def download_reel(url, username, password):
-    # Set up Instaloader with login credentials
-    loader = instaloader.Instaloader()
-    loader.login(username, password)
+# Load environment variables
+load_dotenv()
 
-    # Extract shortcode from URL
-    shortcode = url.split('/')[-2]
+# Get credentials from environment variables
+INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
+INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
-    # Download the reel
+def initialize_instaloader():
+    """Initialize and login to Instaloader"""
     try:
+        loader = instaloader.Instaloader()
+        if INSTAGRAM_USERNAME and INSTAGRAM_PASSWORD:
+            loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+            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 Exception as e:
+        st.error(f"Error initializing Instaloader: {str(e)}")
+        return None
+
+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]
+    except Exception:
+        return None
+
+def download_reel(url):
+    """Download Instagram reel"""
+    if not url:
+        return "Please enter a URL"
+    
+    loader = initialize_instaloader()
+    if not loader:
+        return "Failed to initialize downloader"
+    
+    shortcode = extract_shortcode(url)
+    if not shortcode:
+        return "Invalid URL format"
+    
+    try:
+        # Create download directory
+        os.makedirs("reels", exist_ok=True)
+        
+        # Download the post
         post = instaloader.Post.from_shortcode(loader.context, shortcode)
         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)}"
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        return f"An unexpected error occurred: {str(e)}"
 
 def main():
-    st.title('Instagram Reel Downloader')
-
+    st.set_page_config(
+        page_title="Instagram Reel Downloader",
+        page_icon="📱"
+    )
+    
+    st.title('📱 Instagram Reel Downloader')
+    
+    # Add instructions
+    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.
+    """)
+    
     # Input for Reel URL
-    url = st.text_input('Enter the Instagram Reel URL')
-
-    # Handle form submission
-    if st.button('Download Reel'):
+    url = st.text_input('Enter the Instagram Reel URL', 
+                       placeholder='https://www.instagram.com/reel/...')
+    
+    # Download button
+    if st.button('Download Reel', type='primary'):
         if url:
-            # Instagram login credentials (remember to handle this securely)
-            username = 'chrunosc'
-            password = 'Chrunosc58'
-            
-            # Create the directory if it doesn't exist
-            if not os.path.exists("reels"):
-                os.makedirs("reels")
-            
-            # Download the reel
-            result = download_reel(url, username, password)
-            st.write(result)
+            with st.spinner('Downloading reel...'):
+                result = download_reel(url)
+                if "successfully" in result:
+                    st.success(result)
+                else:
+                    st.error(result)
         else:
             st.warning('Please enter a URL.')
 
 if __name__ == "__main__":
-    main()
+    main()