Update ui_components.py

ui_components.py

@@ -106,10 +106,15 @@ class UIComponents:
     def _create_login_section(self):
         """Create HuggingFace OAuth login section"""
         with gr.Group(elem_classes="login-section"):
-            self.login_button = gr.LoginButton(
-                value="Sign in for Inference",
-                redirect_url="https://gradio-chat-gradio-app-hfips.hf.space/"
-            )
+            # 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)
     
     def _create_provider_model_selection(self):
         """Create provider and model selection dropdowns with defaults"""
@@ -475,4 +480,4 @@ class UIComponents:
         self.remove_all_btn.click(
             handle_remove_all,
             outputs=[self.server_checkboxes, self.add_server_output]
-        )
+        )