Update app.py

app.py

@@ -13,11 +13,112 @@ load_dotenv()
 
 # Get credentials from environment variables
 INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
+INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
 # Configure logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+
+# Check if credentials are available
+if not INSTAGRAM_USERNAME or not INSTAGRAM_PASSWORD:
+    logger.error("Instagram credentials are not set in the environment variables.")
+    raise RuntimeError("Instagram credentials missing.")
+
+app = FastAPI(title="Instagram Post Downloader")
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class PostRequest(BaseModel):
+    url: str
+
+class DownloadResponse(BaseModel):
+    download_url: str
+
+def extract_shortcode(url: str) -> str:
+    """Extract Instagram post shortcode from URL"""
+    url = url.strip()
+    parts = url.split('/')
+    
+    if 'instagram.com' not in parts:
+        raise ValueError("Invalid Instagram URL")
+    
+    for i, part in enumerate(parts):
+        if part in ['p', 'reel', 'tv'] and (i+1) < len(parts):
+            shortcode = parts[i+1].split('?')[0]
+            return shortcode
+    
+    if 'instagram.com/p/' in url:
+        return url.split('/p/')[1].split('/')[0].split('?')[0]
+    
+    raise ValueError("Invalid Instagram URL")
+
+@app.post("/dl", response_model=DownloadResponse)
+async def download_post(request: PostRequest):
+    """Download Instagram post and return media URL"""
+    try:
+        shortcode = extract_shortcode(request.url)
+        logger.info(f"Extracted shortcode: {shortcode}")
+    except ValueError as e:
+        logger.error(f"Invalid URL: {request.url}")
+        raise HTTPException(status_code=400, detail=str(e))
+
+    L = instaloader.Instaloader()
+    
+    # Configure retry parameters
+    max_retries = 3
+    retry_delay = random.uniform(1, 3)
+    
+    for attempt in range(max_retries):
+        try:
+            # Login to Instagram
+            L.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+            logger.info("Successfully logged in to Instagram")
+            
+            # Fetch post by shortcode
+            post = instaloader.Post.from_shortcode(L.context, shortcode)
+            logger.info(f"Fetched post: {post}")
+            break
+            
+        except instaloader.exceptions.ConnectionException as e:
+            logger.error(f"Connection error (attempt {attempt + 1}): {e}")
+            if attempt == max_retries - 1:
+                raise HTTPException(status_code=500, detail="Failed to connect to Instagram after multiple attempts")
+            time.sleep(retry_delay)
+            retry_delay *= 2  # Exponential backoff
+            
+        except instaloader.exceptions.InvalidArgumentException:
+            logger.error(f"Post not found for shortcode: {shortcode}")
+            raise HTTPException(status_code=404, detail="Post not found")
+            
+        except instaloader.exceptions.InstaloaderException as e:
+            logger.error(f"Instaloader error: {e}")
+            raise HTTPException(status_code=500, detail="Error fetching post")
+            
+        except Exception as e:
+            logger.error(f"Unexpected error: {e}")
+            raise HTTPException(status_code=500, detail="Internal server error")
+
+    # Get download URL based on media type
+    if post.is_video:
+        download_url = post.video_url
+    else:
+        download_url = post.url
+
+    if not download_url:
+        logger.error("No media found in post")
+        raise HTTPException(status_code=404, detail="No media found in post")
+
+    return {"download_url": download_url}
+
+
 # Directly embed session data
 SESSION_DATA = {
     "sessionid": "49379440378%3AAMKUUgo79iWhmt%3A1%3AAYcuPYtZFJAwzPlT0u_xuXkaiBHbDJ-8igmXkPBmzQ",