Update app.py

app.py

@@ -20,7 +20,6 @@ INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 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.")
@@ -46,21 +45,18 @@ class DownloadResponse(BaseModel):
 def extract_shortcode(url: str) -> str:
     """Extract Instagram post shortcode from URL"""
     url = url.strip()
-    # Correct the check for 'instagram.com'
     if 'instagram.com' not in url:
         raise ValueError("Invalid Instagram URL")
-    
     parts = url.split('/')
     for i, part in enumerate(parts):
-        if part in ['p', 'reel', 'tv'] and (i+1) < len(parts):
-            shortcode = parts[i+1].split('?')[0]
+        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"""
@@ -72,41 +68,47 @@ async def download_post(request: PostRequest):
         raise HTTPException(status_code=400, detail=str(e))
 
     L = instaloader.Instaloader()
-    
-    try:
-        # Login to Instagram
-        L.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
-        logger.info("Successfully logged in to Instagram")
-    except instaloader.exceptions.InstaloaderException as e:
-        logger.error(f"Login error: {e}")
-        raise HTTPException(status_code=500, detail="Error logging in to Instagram")
-    
+    session_file = f"{INSTAGRAM_USERNAME}.session"
+
+    if os.path.exists(session_file):
+        try:
+            L.load_session_from_file(INSTAGRAM_USERNAME, session_file)
+            logger.info("Loaded session from file.")
+        except FileNotFoundError:
+            pass
+    else:
+        try:
+            # Login to Instagram
+            L.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+            logger.info("Successfully logged in to Instagram")
+            L.save_session_to_file(session_file)
+            logger.info("Saved session to file.")
+        except instaloader.exceptions.InstaloaderException as e:
+            logger.error(f"Login error: {e}")
+            raise HTTPException(status_code=500, detail="Error logging in to Instagram")
+
     # Configure retry parameters
     max_retries = 3
-    retry_delay = random.uniform(30, 60)
-    
+    retry_delay = random.uniform(1, 3)
+
     for attempt in range(max_retries):
         try:
             # 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")
@@ -126,4 +128,4 @@ async def download_post(request: PostRequest):
 
 @app.get("/api/health")
 async def health_check():
-    return {"status": "healthy"}
+    return {"status": "healthy"}