Update app.py

app.py

@@ -40,7 +40,7 @@ class PostRequest(BaseModel):
     url: str
 
 class DownloadResponse(BaseModel):
-    download_url: str
+    download_urls: List[str]
 
 def extract_shortcode(url: str) -> str:
     """Extract Instagram post shortcode from URL"""
@@ -113,19 +113,27 @@ async def download_post(request: PostRequest):
             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
+    download_urls = []
+    if post.typename == 'GraphSidecar':
+        # Handle carousel posts
+        for node in post.get_sidecar_nodes():
+            if node.is_video:
+                download_urls.append(node.video_url)
+            else:
+                download_urls.append(node.display_url)
     else:
-        download_url = post.url
+        if post.is_video:
+            download_urls.append(post.video_url)
+        else:
+            download_urls.append(post.url)
 
-    if not download_url:
+    if not download_urls:
         logger.error("No media found in post")
         raise HTTPException(status_code=404, detail="No media found in post")
 
-    return {"download_url": download_url}
+    return {"download_urls": download_urls}
 
 
 @app.get("/api/health")
 async def health_check():
-    return {"status": "healthy"}
+    return {"status": "healthy"}