Update ui_components.py

ui_components.py

@@ -106,15 +106,13 @@ class UIComponents:
     def _create_login_section(self):
         """Create HuggingFace OAuth login section"""
         with gr.Group(elem_classes="login-section"):
-            # Build a redirect URL that matches this Space's host to satisfy HF OAuth rules
-            # Prefer using SPACE_HOST provided by Hugging Face Spaces; fall back to default behavior
-            space_host = os.getenv("SPACE_HOST")
-            login_kwargs: Dict[str, Any] = {"value": "Sign in for Inference"}
-            if space_host:
-                # Use a callback path on the same host as recommended by HF docs
-                login_kwargs["redirect_url"] = f"https://{space_host}/login/callback"
-
-            self.login_button = gr.LoginButton(**login_kwargs)
+            # Let Gradio determine the correct redirect URL when running on Hugging Face Spaces.
+            # Overriding redirect_url can easily cause a host/scheme mismatch (e.g., double https:// or wrong domain),
+            # which results in: "Error 400: Invalid redirect_uri, must start with https://<space>.hf.space".
+            #
+            # On Spaces, Gradio auto-configures OAuth to the Space subdomain, so we intentionally do NOT pass redirect_url.
+            # If running locally, OAuth via HF won't work; use HF_TOKEN instead.
+            self.login_button = gr.LoginButton(value="Sign in for Inference")
     
     def _create_provider_model_selection(self):
         """Create provider and model selection dropdowns with defaults"""