Update app.py

app.py

@@ -1,6 +1,6 @@
 # api/main.py
 from fastapi import FastAPI, HTTPException, Depends
-from pydantic import BaseModel
+from pydantic import BaseModel, constr
 from typing import List, Dict
 import logging
 import requests
@@ -39,19 +39,37 @@ class AnalyzePostRequest(BaseModel):
     hashtags: str
     image_url: str
 
+RATE_LIMIT_DELAY = 5  # Delay in seconds between API calls
+LAST_REQUEST_TIME = 0
+
 @app.post("/fetch-posts")
 async def fetch_posts(user: UserRequest):
     """
     Fetch posts from a given Instagram profile (public data only).
     """
+    global LAST_REQUEST_TIME
+
     username = user.username
-    logging.info(f"Fetching posts for user: {username}")
+    logger.info(f"Fetching posts for user: {username}")
+
+    # Rate limiting
+    current_time = time.time()
+    if current_time - LAST_REQUEST_TIME < RATE_LIMIT_DELAY:
+        raise HTTPException(
+            status_code=status.HTTP_429_TOO_MANY_REQUESTS,
+            detail="Please wait a few seconds before making another request."
+        )
+    LAST_REQUEST_TIME = current_time
 
     try:
         # Fetch user's posts
         user_posts = fetch_user_posts(username)
         if not user_posts:
-            raise HTTPException(status_code=404, detail="No posts found for the user.")
+            logger.warning(f"No posts found for user: {username}")
+            raise HTTPException(
+                status_code=status.HTTP_404_NOT_FOUND,
+                detail="No posts found for the user."
+            )
 
         # Fetch competitors' posts
         competitors_posts = fetch_competitors_data(username)
@@ -60,12 +78,29 @@ async def fetch_posts(user: UserRequest):
         all_posts = user_posts + competitors_posts
 
         # Save data to the database
-        save_to_db(all_posts)
+        if not save_to_db(all_posts):
+            logger.error("Failed to save data to the database.")
+            raise HTTPException(
+                status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+                detail="Failed to save data to the database."
+            )
+
+        # Return success response
+        return {
+            "status": "success",
+            "data": all_posts,
+            "message": f"Successfully fetched {len(all_posts)} posts."
+        }
 
-        return {"status": "success", "data": all_posts}
+    except HTTPException as e:
+        # Re-raise HTTPException to return specific error messages
+        raise e
     except Exception as e:
-        logging.error(f"Error fetching posts: {e}")
-        raise HTTPException(status_code=500, detail=str(e))
+        logger.error(f"Unexpected error fetching posts: {e}")
+        raise HTTPException(
+            status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
+            detail="An unexpected error occurred. Please try again later."
+        )
 
 @app.post("/analyze")
 async def analyze(user: UserRequest, db: Session = Depends(get_db)):