Create main.py

main.py

@@ -0,0 +1,110 @@
+# main.py
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import instaloader
+import os
+from dotenv import load_dotenv
+import re
+from typing import Optional, List
+
+# Load environment variables
+load_dotenv()
+
+# Get credentials from environment variables
+INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
+INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
+
+app = FastAPI(title="Instagram Downloader API")
+
+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'/([A-Za-z0-9_-]{11})[/]?$'
+        ]
+        
+        for pattern in patterns:
+            match = re.search(pattern, cleaned_url)
+            if match:
+                return match.group(1)
+        
+        segments = cleaned_url.split('/')
+        if segments:
+            return segments[-1]
+                
+        return None
+    except Exception:
+        return None
+
+@app.post("/api/download", response_model=DownloadResponse)
+async def download_media(request: DownloadRequest):
+    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")
+        
+        loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+
+        # Get post
+        post = instaloader.Post.from_shortcode(loader.context, shortcode)
+        
+        urls = []
+        
+        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)
+            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:
+        raise HTTPException(status_code=400, detail=str(e))
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/api/health")
+async def health_check():
+    return {"status": "healthy"}