Update service/auth_service.py

service/auth_service.py

@@ -54,29 +54,29 @@ class AuthService:
             raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(e))
     
     @staticmethod
-    def get_google_oauth_url(final_redirect: str = None):
+    def get_google_oauth_url(redirect_to: str = None):
         try:
-            # We append the final destination as a query parameter directly 
-            # to the callback URL that Supabase will use.
+            # This is your FastAPI backend callback URL
             base_callback = "https://hammad712-digitalbillboard.hf.space/auth/callback"
             
-            # If final_redirect is 'digital-billboard://', 
-            # target becomes '.../callback?final_dest=digital-billboard://'
-            target_redirect = base_callback
-            if final_redirect:
-                target_redirect = f"{base_callback}?final_dest={final_redirect}"
+            # If redirect_to is 'digital-billboard://', we bake it into the callback URL
+            # so it comes back to us in the 'google_callback' function later.
+            if redirect_to:
+                computed_callback = f"{base_callback}?final_dest={redirect_to}"
+            else:
+                computed_callback = base_callback
     
             response = supabase.auth.sign_in_with_oauth({
                 "provider": "google",
                 "options": {
-                    "redirect_to": target_redirect  # This is the key change
+                    "redirect_to": computed_callback # Supabase will return the user here
                 }
             })
             return response.url
         except Exception as e:
             logger.error("Failed to generate Google URL: %s", str(e))
             raise HTTPException(status_code=500, detail="OAuth Init Failed")
-
+        
     @staticmethod
     def exchange_code(auth_code: str):
         try: