Update app.py

app.py

@@ -1,15 +1,29 @@
+from fastapi import FastAPI, HTTPException
 import instaloader
+from pydantic import BaseModel
+from typing import List, Dict
 
-def fetch_instagram_posts(username: str, max_posts: int = 10):
+app = FastAPI()
+
+# Pydantic model for the response
+class PostResponse(BaseModel):
+    caption: str
+    likes: int
+    comments: int
+    posted_on: str
+    image_url: str
+
+@app.get("/fetch-posts", response_model=List[PostResponse])
+async def fetch_instagram_posts(username: str, max_posts: int = 10):
     """
     Fetch Instagram posts from a public profile.
 
     Args:
         username (str): The username of the Instagram profile.
-        max_posts (int): The maximum number of posts to fetch.
+        max_posts (int): The maximum number of posts to fetch (default is 10).
 
     Returns:
-        None
+        List[PostResponse]: A list of posts with caption, likes, comments, posted_on, and image_url.
     """
     # Initialize Instaloader
     L = instaloader.Instaloader()
@@ -19,20 +33,22 @@ def fetch_instagram_posts(username: str, max_posts: int = 10):
         print(f"Fetching posts for user: {username}")
         profile = instaloader.Profile.from_username(L.context, username)
 
+        # List to store posts
+        posts = []
+
         # Counter to limit the number of posts
         count = 0
 
         # Loop through the posts
         for post in profile.get_posts():
-            print(f"\nPost {count + 1}:")
-            print(f"Caption: {post.caption or 'No Caption'}")
-            print(f"Likes: {post.likes}")
-            print(f"Comments: {post.comments}")
-            print(f"Posted on: {post.date_utc}")
-            print(f"Image URL: {post.url}")
-
-            # Save the post image locally
-            L.download_post(post, target=profile.username)
+            post_data = {
+                "caption": post.caption or "No Caption",
+                "likes": post.likes,
+                "comments": post.comments,
+                "posted_on": post.date_utc.strftime("%Y-%m-%d %H:%M:%S"),
+                "image_url": post.url,
+            }
+            posts.append(post_data)
 
             # Increment the counter
             count += 1
@@ -40,19 +56,16 @@ def fetch_instagram_posts(username: str, max_posts: int = 10):
                 break
 
         print(f"\nFetched {count} posts for user: {username}.")
+        return posts
 
     except instaloader.exceptions.ProfileNotExistsException:
-        print(f"Error: Profile '{username}' does not exist.")
+        raise HTTPException(status_code=404, detail=f"Profile '{username}' does not exist.")
     except instaloader.exceptions.ConnectionException:
-        print("Error: Unable to connect to Instagram. Check your internet connection.")
+        raise HTTPException(status_code=503, detail="Unable to connect to Instagram. Check your internet connection.")
     except Exception as e:
-        print(f"Unexpected error: {e}")
-
+        raise HTTPException(status_code=500, detail=f"Unexpected error: {e}")
 
+# Run the FastAPI app
 if __name__ == "__main__":
-    # Hardcode the username and number of posts to fetch
-    username = "bionsdesignco"  # Replace with the desired Instagram username
-    max_posts = 5  # Replace with the desired number of posts to fetch
-
-    # Fetch posts
-    fetch_instagram_posts(username, max_posts)
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)