from fastapi import APIRouter
from pydantic import BaseModel

auth_router = APIRouter()

class LoginRequest(BaseModel):
    username: str
    password: str

USER_DATA = {
    "admin": "123",
    "sri": "123",
    "mugu": "123",
    "praveen": "123"
}

@auth_router.post("/login")
def login(request: LoginRequest):
    username = request.username
    password = request.password
    if username in USER_DATA and USER_DATA[username] == password:
        return {"success": True, "message": "Login successful"}
    else:
        return {"success": False, "message": "Invalid username or password"}