Update auth.py

auth.py

@@ -6,11 +6,9 @@ from urllib.parse import quote_plus
 from typing import List, Optional, Any
 
 from dotenv import load_dotenv
-from fastapi import APIRouter, HTTPException, Depends, Request, status, UploadFile, File, Form
+from fastapi import APIRouter, HTTPException, Depends, Request, UploadFile, File, Form
 from fastapi.responses import JSONResponse
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
-from slowapi import Limiter, _rate_limit_exceeded_handler
-from slowapi.util import get_remote_address
 from jose import JWTError, jwt
 from passlib.context import CryptContext
 from pydantic import BaseModel, EmailStr, Field, validator
@@ -30,9 +28,6 @@ client = MongoClient(MONGO_URL)
 db = client.users_database
 users_collection = db.users
 
-# Rate limiter (applied to auth endpoints)
-limiter = Limiter(key_func=get_remote_address, default_limits=["200 per day", "50 per hour"])
-
 # OAuth2 setup
 oauth2_scheme = OAuth2PasswordBearer(tokenUrl="token")
 
@@ -168,7 +163,6 @@ def save_avatar_file(file: UploadFile) -> str:
 # ----- Auth Endpoints -----
 
 @router.post("/signup", response_model=Token)
-@limiter.limit("5/minute")
 async def signup(
     request: Request,
     name: str = Form(...),
@@ -176,7 +170,6 @@ async def signup(
     password: str = Form(...),
     avatar: Optional[UploadFile] = File(None)
 ):
-    # Validate input using the User model
     try:
         _ = User(name=name, email=email, password=password)
     except Exception as e:
@@ -190,7 +183,7 @@ async def signup(
         "name": name,
         "email": email,
         "hashed_password": hashed_password,
-        "chat_histories": []  # Initialize an empty array for chat histories and future data.
+        "chat_histories": []
     }
     if avatar:
         avatar_path = save_avatar_file(avatar)
@@ -204,7 +197,6 @@ async def signup(
     }
 
 @router.post("/login", response_model=LoginResponse)
-@limiter.limit("10/minute")
 async def login(request: Request, form_data: OAuth2PasswordRequestForm = Depends()):
     user = authenticate_user(form_data.username, form_data.password)
     if not user:
@@ -220,7 +212,6 @@ async def login(request: Request, form_data: OAuth2PasswordRequestForm = Depends
     }
 
 @router.get("/user/data")
-@limiter.limit("20/minute")
 async def get_user_data(request: Request, current_user: dict = Depends(get_current_user)):
     return {
         "name": current_user["name"],
@@ -230,7 +221,6 @@ async def get_user_data(request: Request, current_user: dict = Depends(get_curre
     }
 
 @router.put("/user/update")
-@limiter.limit("10/minute")
 async def update_user(
     request: Request,
     name: Optional[str] = Form(None),
@@ -262,8 +252,6 @@ async def update_user(
     return {"message": "User updated successfully"}
 
 @router.post("/logout")
-@limiter.limit("20/minute")
 async def logout(request: Request, current_user: dict = Depends(get_current_user)):
     logger.info(f"User logged out: {current_user['email']}")
     return {"message": "User logged out successfully"}
-