Update routes/auth_routes.py

routes/auth_routes.py

@@ -52,7 +52,7 @@ async def get_google_auth_url(redirect_to: str = None):
 @router.get("/callback")
 async def google_callback(request: Request):
     code = request.query_params.get("code")
-    # We now look for 'final_dest' which we baked into the URL
+    # We retrieve the 'final_dest' we tucked away in the previous step
     final_dest = request.query_params.get("final_dest")
 
     if not code:
@@ -60,13 +60,14 @@ async def google_callback(request: Request):
 
     access_token = AuthService.exchange_code(code)
 
-    # If the URL contains our mobile scheme, go to the app. 
-    # Otherwise, go to Vercel.
+    # THE DECISION:
     if final_dest and "digital-billboard" in final_dest:
-        # Standard Expo redirect format
+        # Redirect to Mobile App
+        # Format: digital-billboard://--/auth-callback#access_token=...
         redirect_url = f"digital-billboard://--/auth-callback#access_token={access_token}"
     else:
+        # Redirect to Vercel Web
         redirect_url = f"{settings.FRONTEND_URL}/dashboard#access_token={access_token}"
 
-    logger.info("AUTH_COMPLETE: Redirecting to %s", redirect_url)
+    logger.info("CALLBACK_COMPLETE: Redirecting to %s", redirect_url)
     return RedirectResponse(url=redirect_url)