Update utils/instaloader_utils.py

utils/instaloader_utils.py

@@ -1,13 +1,12 @@
 import instaloader
-from typing import List, Dict, Optional
-import instaloader.exceptions
+from typing import List, Dict
+from fastapi import HTTPException
+import requests
 
 # Initialize Instaloader
 L = instaloader.Instaloader()
 
-
-
-async def fetch_user_posts(username: str, max_posts: int = 20):
+async def fetch_user_posts(username: str, max_posts: int = 20) -> List[Dict]:
     try:
         print(f"Fetching posts for user: {username}")
         profile = instaloader.Profile.from_username(L.context, username)
@@ -17,7 +16,7 @@ async def fetch_user_posts(username: str, max_posts: int = 20):
             if count >= max_posts:
                 break
             posts.append({
-                "caption": post.caption,
+                "caption": post.caption or "No Caption",
                 "likes": post.likes,
                 "comments": post.comments,
                 "date": post.date_utc.isoformat(),
@@ -25,12 +24,13 @@ async def fetch_user_posts(username: str, max_posts: int = 20):
             })
 
         if not posts:
-            raise HTTPException(status_code=404, detail="No posts found.")
-        
-        return {"username": username, "posts": posts}
+            print("No posts found.")
+            return []
+
+        return posts
 
     except instaloader.exceptions.ProfileNotExistsException:
-        raise HTTPException(status_code=404, detail="Profile does not exist.")
+        raise HTTPException(status_code=404, detail=f"Profile '{username}' does not exist.")
     except instaloader.exceptions.ConnectionException:
         raise HTTPException(status_code=503, detail="Instagram connection failed. Try again later.")
     except Exception as e:
@@ -64,7 +64,8 @@ def find_similar_accounts(username: str, rapidapi_key: str) -> List[str]:
         print(f"API request failed: {e}")
         return []
 
-def fetch_competitors_posts(username: str, rapidapi_key: str, max_posts: int = 50) -> List[Dict]:
+
+async def fetch_competitors_posts(username: str, rapidapi_key: str, max_posts: int = 50) -> List[Dict]:
     """
     Fetch posts for similar accounts (competitors) using the RapidAPI endpoint.
     """
@@ -78,7 +79,8 @@ def fetch_competitors_posts(username: str, rapidapi_key: str, max_posts: int = 5
     all_posts = []
     for account in similar_accounts:
         print(f"Fetching posts for competitor: {account}")
-        competitor_posts = fetch_user_posts(account, rapidapi_key, max_posts)
-        all_posts.extend(competitor_posts)
-
-    return all_posts
+        try:
+            competitor_posts = await fetch_user_posts(account, max_posts)
+            all_posts.extend(competitor_posts)
+        except HTTPException as e:
+            print(f"Error fetching posts for {account}: {e.detail}")