Update app.py

app.py

@@ -1,9 +1,10 @@
-import streamlit as st
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 import instaloader
 import os
 from dotenv import load_dotenv
 import re
-import requests
+from typing import Optional, List
 
 # Load environment variables
 load_dotenv()
@@ -12,13 +13,27 @@ load_dotenv()
 INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
 INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
-def clean_url(url):
+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]:
     """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})[/]?$'
         ]
         
@@ -32,132 +47,69 @@ def clean_url(url):
             return segments[-1]
                 
         return None
-    except Exception as e:
-        st.error(f"Error cleaning URL: {str(e)}")
+    except Exception:
         return None
 
-def download_file(url, path):
-    """Download file from URL with progress bar"""
-    try:
-        response = requests.get(url, stream=True)
-        response.raise_for_status()
-        
-        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
-
-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
-    )
+@app.get('/')
+def main():
+    return {"appStatus": "running"}
     
+
+@app.post("/api/download", response_model=DownloadResponse)
+async def download_media(request: DownloadRequest):
     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"
-        
-        download_path = os.path.join('reels', shortcode)
-        os.makedirs(download_path, exist_ok=True)
+        # 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")
         
-        st.info("Fetching post metadata...")
+        loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+
+        # Get post
         post = instaloader.Post.from_shortcode(loader.context, shortcode)
         
+        urls = []
+        
         if post.is_video:
-            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:
-                return None, "Failed to download video file"
+            media_type = "video"
+            urls.append(post.video_url)
         else:
-            return None, "Post is not a video"
-            
+            media_type = "image"
+            if post.typename == "GraphSidecar":
+                # Multiple images
+                for node in post.get_sidecar_nodes():
+                    urls.append(node.display_url)
+            else:
+                # Single image
+                urls.append(post.url)
+
+        return DownloadResponse(
+            success=True,
+            message="Media URLs retrieved successfully",
+            urls=urls,
+            type=media_type,
+            shortcode=shortcode
+        )
+
     except instaloader.exceptions.InstaloaderException as e:
-        return None, f"Instagram error: {str(e)}"
+        raise HTTPException(status_code=400, detail=str(e))
     except Exception as 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.')
+        raise HTTPException(status_code=500, detail=str(e))
 
-if __name__ == "__main__":
-    main()
+@app.get("/api/health")
+async def health_check():
+    return {"status": "healthy"}