Update app.py

app.py

@@ -1,3 +1,4 @@
+import logging
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 import instaloader
@@ -5,8 +6,6 @@ import os
 from dotenv import load_dotenv
 import re
 from typing import Optional, List
-import logging
-
 
 # Load environment variables
 load_dotenv()
@@ -15,6 +14,10 @@ load_dotenv()
 INSTAGRAM_USERNAME = os.getenv('INSTAGRAM_USERNAME')
 INSTAGRAM_PASSWORD = os.getenv('INSTAGRAM_PASSWORD')
 
+# Configure logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 app = FastAPI()
 
 class DownloadRequest(BaseModel):
@@ -75,18 +78,27 @@ async def download_media(request: DownloadRequest):
             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)
-        logging.info(loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD))
+        # Load session
+        if os.path.exists(SESSION_FILE):
+            loader.load_session_from_file(INSTAGRAM_USERNAME, SESSION_FILE)
+        else:
+            # Log before login attempt
+            logger.info("Attempting to log in with username: %s", INSTAGRAM_USERNAME)
+            
+            # Login
+            if not all([INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD]):
+                raise HTTPException(status_code=500, detail="Missing Instagram credentials")
+            loader.login(INSTAGRAM_USERNAME, INSTAGRAM_PASSWORD)
+            
+            # Log after successful login
+            logger.info("Login successful for username: %s", INSTAGRAM_USERNAME)
+            loader.save_session_to_file(SESSION_FILE)
 
         # Get post
         post = instaloader.Post.from_shortcode(loader.context, shortcode)
-        
+
         urls = []
-        
+
         if post.is_video:
             media_type = "video"
             urls.append(post.video_url)
@@ -96,9 +108,9 @@ async def download_media(request: DownloadRequest):
                 # Multiple images
                 for node in post.get_sidecar_nodes():
                     urls.append(node.display_url)
-            else:
-                # Single image
-                urls.append(post.url)
+                else:
+                    # Single image
+                    urls.append(post.url)
 
         return DownloadResponse(
             success=True,
@@ -109,8 +121,12 @@ async def download_media(request: DownloadRequest):
         )
 
     except instaloader.exceptions.InstaloaderException as e:
+        # Log Instaloader exception
+        logger.error("InstaloaderException: %s", str(e))
         raise HTTPException(status_code=400, detail=str(e))
     except Exception as e:
+        # Log general exception
+        logger.error("Exception: %s", str(e))
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.get("/api/health")