Update app.py

app.py

@@ -1,10 +1,9 @@
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+import streamlit as st
 import instaloader
 import os
 from dotenv import load_dotenv
 import re
-from typing import Optional, List
+import requests
 
 # Load environment variables
 load_dotenv()
@@ -13,27 +12,13 @@ load_dotenv()
 INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
 INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
-app = FastAPI()
-
-class DownloadRequest(BaseModel):
-    url: str
-
-class DownloadResponse(BaseModel):
-    success: bool
-    message: str
-    urls: Optional[List[str]] = None
-    type: Optional[str] = None
-    shortcode: Optional[str] = None
-
-def clean_url(url: str) -> Optional[str]:
+def clean_url(url):
     """Clean Instagram URL and extract shortcode"""
     try:
         cleaned_url = url.split('?')[0].split('#')[0].strip('/')
         patterns = [
             r'instagram.com/reel/([A-Za-z0-9_-]+)',
             r'instagram.com/p/([A-Za-z0-9_-]+)',
-            r'instagram.com/stories/([A-Za-z0-9_-]+)',
-            r'instagram.com/tv/([A-Za-z0-9_-]+)',
             r'/([A-Za-z0-9_-]{11})[/]?$'
         ]
         
@@ -47,69 +32,132 @@ def clean_url(url: str) -> Optional[str]:
             return segments[-1]
                 
         return None
-    except Exception:
+    except Exception as e:
+        st.error(f"Error cleaning URL: {str(e)}")
         return None
 
-@app.get('/')
-def main():
-    return {"appStatus": "running"}
-    
-
-@app.post("/api/download", response_model=DownloadResponse)
-async def download_media(request: DownloadRequest):
+def download_file(url, path):
+    """Download file from URL with progress bar"""
     try:
-        # Extract shortcode
-        shortcode = clean_url(request.url)
-        if not shortcode:
-            raise HTTPException(status_code=400, detail="Invalid URL format")
-
-        # Initialize loader
-        loader = instaloader.Instaloader(
-            download_videos=True,
-            download_video_thumbnails=False,
-            download_geotags=False,
-            download_comments=False,
-            save_metadata=False,
-            compress_json=False
-        )
-
-        # Login
-        if not all([INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD]):
-            raise HTTPException(status_code=500, detail="Missing Instagram credentials")
+        response = requests.get(url, stream=True)
+        response.raise_for_status()
         
-        loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+        file_size = int(response.headers.get('content-length', 0))
+        progress_bar = st.progress(0)
+        
+        with open(path, 'wb') as f:
+            if file_size == 0:
+                f.write(response.content)
+            else:
+                downloaded = 0
+                for chunk in response.iter_content(chunk_size=8192):
+                    if chunk:
+                        f.write(chunk)
+                        downloaded += len(chunk)
+                        progress = int((downloaded / file_size) * 100)
+                        progress_bar.progress(progress)
+        
+        return True
+    except Exception as e:
+        st.error(f"Download error: {str(e)}")
+        return False
 
-        # Get post
-        post = instaloader.Post.from_shortcode(loader.context, shortcode)
+def download_reel(url):
+    """Download Instagram reel"""
+    if not url:
+        return None, "Please enter a URL"
+    
+    shortcode = clean_url(url)
+    if not shortcode:
+        return None, "Could not extract valid shortcode from URL"
+    
+    st.info(f"Extracted shortcode: {shortcode}")
+    
+    loader = instaloader.Instaloader(
+        download_videos=True,
+        download_video_thumbnails=False,
+        download_geotags=False,
+        download_comments=False,
+        save_metadata=False,
+        compress_json=False
+    )
+    
+    try:
+        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 None, "Missing Instagram credentials"
         
-        urls = []
+        download_path = os.path.join('reels', shortcode)
+        os.makedirs(download_path, exist_ok=True)
+        
+        st.info("Fetching post metadata...")
+        post = instaloader.Post.from_shortcode(loader.context, shortcode)
         
         if post.is_video:
-            media_type = "video"
-            urls.append(post.video_url)
-        else:
-            media_type = "image"
-            if post.typename == "GraphSidecar":
-                # Multiple images
-                for node in post.get_sidecar_nodes():
-                    urls.append(node.display_url)
+            video_url = post.video_url
+            video_path = os.path.join(download_path, f"{shortcode}.mp4")
+            st.info("Starting download...")
+            
+            if download_file(video_url, video_path):
+                if os.path.exists(video_path):
+                    size = os.path.getsize(video_path)
+                    if size > 0:
+                        return video_path, f"Download successful! Size: {size/1024/1024:.2f} MB"
+                    else:
+                        return None, "Download failed: File is empty"
+                else:
+                    return None, "Download failed: File not created"
             else:
-                # Single image
-                urls.append(post.url)
-
-        return DownloadResponse(
-            success=True,
-            message="Media URLs retrieved successfully",
-            urls=urls,
-            type=media_type,
-            shortcode=shortcode
-        )
-
+                return None, "Failed to download video file"
+        else:
+            return None, "Post is not a video"
+            
     except instaloader.exceptions.InstaloaderException as e:
-        raise HTTPException(status_code=400, detail=str(e))
+        return None, f"Instagram error: {str(e)}"
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        return None, f"Unexpected error: {str(e)}"
+
+def main():
+    st.set_page_config(
+        page_title="Instagram Reel Downloader",
+        page_icon="📱",
+        initial_sidebar_state="collapsed"
+    )
+    
+    st.title('📱 Instagram Reel Downloader')
+    
+    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', 
+                       placeholder='https://www.instagram.com/reel/XXXXXXXXXXXX')
+    
+    # Download button
+    if st.button('Download Reel', type='primary'):
+        if url:
+            with st.spinner('Processing...'):
+                file_path, message = download_reel(url)
+                st.write(message)
+                
+                # 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.')
 
-@app.get("/api/health")
-async def health_check():
-    return {"status": "healthy"}
+if __name__ == "__main__":
+    main()