debug

app/api.py

@@ -1,20 +1,17 @@
 # app/api.py
 from fastapi import APIRouter, HTTPException, status, Depends, WebSocket, WebSocketDisconnect
-# Remove JSONResponse if not explicitly needed
 import logging
 
 from . import schemas, crud, auth, models
 from .websocket import manager
-# --- Use the new dependency ---
 from .dependencies import get_required_current_user
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
 
-# --- (register and login endpoints remain the same) ---
-@router.post("/register", ...) # Keep as is
+# --- FIX THE DECORATORS HERE ---
+@router.post("/register", status_code=status.HTTP_201_CREATED, response_model=models.User) # <-- FIX HERE
 async def register_user(user_in: schemas.UserCreate):
-    # ... same logic ...
     existing_user = await crud.get_user_by_email(user_in.email)
     if existing_user:
         raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="Email already registered")
@@ -26,29 +23,25 @@ async def register_user(user_in: schemas.UserCreate):
     if not created_user: raise HTTPException(status_code=500, detail="Failed to retrieve created user")
     return models.User(id=created_user.id, email=created_user.email)
 
-@router.post("/login", ...) # Keep as is
+@router.post("/login", response_model=schemas.Token) # <-- FIX HERE
 async def login_for_access_token(form_data: schemas.UserLogin):
-     # ... same logic ...
     user = await crud.get_user_by_email(form_data.email)
     if not user or not auth.verify_password(form_data.password, user.hashed_password):
         raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Incorrect email or password", headers={"WWW-Authenticate": "Bearer"})
     access_token = auth.create_session_token(user_id=user.id)
     return {"access_token": access_token, "token_type": "bearer"}
+# --- END FIXES ---
 
-# --- UPDATE this endpoint ---
 @router.get("/users/me", response_model=models.User)
 async def read_users_me(current_user: models.User = Depends(get_required_current_user)):
     """
     Returns the current authenticated user's details based on the
     Authorization: Bearer <token> header.
     """
-    # The dependency now handles getting the user from the header token
     return current_user
 
-# --- (websocket endpoint remains the same) ---
 @router.websocket("/ws/{user_id_token}")
 async def websocket_endpoint(websocket: WebSocket, user_id_token: str):
-    # ... same logic ...
     user_id = await auth.get_user_id_from_token(user_id_token)
     if user_id is None:
         logger.warning(f"WebSocket connection rejected: Invalid token {user_id_token}")