Update app.py

app.py

@@ -16,6 +16,7 @@ from utils.instaloader_utils import fetch_user_posts, fetch_competitors_data
 import torch
 from torchvision import transforms
 from transformers import ResNetForImageClassification
+import re
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -47,9 +48,17 @@ logger = logging.getLogger(__name__)
 RATE_LIMIT_DELAY = 5  # Delay in seconds between API calls
 LAST_REQUEST_TIME = 0
 
-# Pydantic model for request validation
 class UserRequest(BaseModel):
-    username: constr(min_length=1, max_length=30, regex="^[a-zA-Z0-9._]+$")  # Instagram username format
+    username: str = Field(..., min_length=1, max_length=30)
+
+    @validator("username")
+    def validate_username(cls, value):
+        """
+        Validate that the username contains only alphanumeric characters, dots, and underscores.
+        """
+        if not re.match(r"^[a-zA-Z0-9._]+$", value):
+            raise ValueError("Username can only contain alphanumeric characters, dots, and underscores.")
+        return value
 
 @app.post("/fetch-posts")
 async def fetch_posts(user: UserRequest):