Update ui_components.py

ui_components.py

@@ -9,16 +9,11 @@ import logging
 import traceback
 from openai import OpenAI
 
-from config import AppConfig, CUSTOM_CSS, HF_HUB_AVAILABLE
+from config import AppConfig, CUSTOM_CSS
 from chat_handler import ChatHandler
 from server_manager import ServerManager
 from mcp_client import UniversalMCPClient
 
-# Import HuggingFace Hub for login functionality
-if HF_HUB_AVAILABLE:
-    from huggingface_hub import login, logout, whoami
-    from huggingface_hub.utils import HfHubHTTPError
-
 logger = logging.getLogger(__name__)
 
 class UIComponents:
@@ -78,8 +73,8 @@ class UIComponents:
         with gr.Sidebar(elem_id="main-sidebar"):
             gr.Markdown("# 🤗 gpt.gradio.app")
             
-            # HuggingFace Login Section
-            self._create_login_section()
+            # API key management section
+            self._create_api_key_section()
             
             # Provider and Model Selection with defaults
             self._create_provider_model_selection()
@@ -91,7 +86,7 @@ class UIComponents:
             with gr.Accordion("📚 Guide & Info", open=False):
                 gr.Markdown("""
                 ## 🎯 How To Use
-                1. **Login**: Login with your HuggingFace account for API access
+                1. **Add Your API Key**: Paste a valid Hugging Face Inference API token
                 2. **Add MCP Servers**: Connect to various AI tools on 🤗Hub
                 3. **Enable/Disable Servers**: Use checkboxes to control which servers are active
                 4. **Chat**: Interact with GPT-OSS and use connected MCP Servers
@@ -103,13 +98,21 @@ class UIComponents:
                 - **Media Support**: Automatic embedding of media -- images, audio, and video etc
                 """)
     
-    def _create_login_section(self):
-        """Create HuggingFace OAuth login section"""
+    def _create_api_key_section(self):
+        """Create secret input section for Hugging Face API keys"""
         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/"
+            gr.Markdown("""
+            **🔐 Hugging Face Inference Token**
+            
+            Paste your API token below. It stays in your browser session only and is not stored.
+            """)
+            self.api_key_box = gr.Textbox(
+                label="HF API Token",
+                placeholder="hf_...",
+                type="password",
+                value=os.getenv("HF_TOKEN", "")
             )
+            self.api_key_status = gr.Markdown("", visible=False, container=True)
     
     def _create_provider_model_selection(self):
         """Create provider and model selection dropdowns with defaults"""
@@ -232,24 +235,34 @@ class UIComponents:
     def _setup_event_handlers(self, chatbot: gr.Chatbot, demo: gr.Blocks):
         """Set up all event handlers"""
         
-        # OAuth profile handler
-        def handle_oauth_profile(profile: gr.OAuthProfile | None, token: gr.OAuthToken | None):
-            if profile is None:
-                return
-            
-            logger.info(f"👤 OAuth profile received for user: {profile.name}")
-            
-            if token and token.token:
-                logger.info("🔑 OAuth token received, updating HF client...")
-                os.environ["HF_TOKEN"] = token.token
-                try:
-                    self.mcp_client.hf_client = OpenAI(
-                        base_url="https://router.huggingface.co/v1",
-                        api_key=token.token
-                    )
-                    logger.info("✅ HuggingFace Inference client updated with OAuth token")
-                except Exception as e:
-                    logger.error(f"❌ Failed to update HF client: {e}")
+        def handle_api_key_update(api_key: str):
+            """Persist user-provided API key for the current session"""
+            if not api_key:
+                os.environ.pop("HF_TOKEN", None)
+                AppConfig.HF_TOKEN = None
+                self.mcp_client.hf_client = None
+                return gr.Markdown("⚠️ API token cleared. Add a token to enable calls.", visible=True)
+
+            token = api_key.strip()
+            os.environ["HF_TOKEN"] = token
+            AppConfig.HF_TOKEN = token
+
+            try:
+                self.mcp_client.hf_client = OpenAI(
+                    base_url="https://router.huggingface.co/v1",
+                    api_key=token
+                )
+                logger.info("✅ HuggingFace client configured from pasted token")
+                return gr.Markdown("✅ API token saved for this session.", visible=True)
+            except Exception as exc:
+                logger.error(f"❌ Failed to configure HF client with provided token: {exc}")
+                return gr.Markdown("❌ Invalid token. Please verify and try again.", visible=True)
+
+        def initialize_api_key_status():
+            token_present = bool(os.getenv("HF_TOKEN"))
+            if token_present:
+                return gr.Markdown("✅ API token detected from environment.", visible=True)
+            return gr.Markdown("", visible=False)
                     
         # Provider selection with auto-model loading
         def handle_provider_change(provider_id):
@@ -290,7 +303,7 @@ class UIComponents:
             if self.mcp_client.hf_client:
                 return f"✅ Ready! Using {active_params} active params, {context_length:,} token context"
             else:
-                return "❌ Please login first"
+                return "❌ Please add your Hugging Face API token"
         
         # Chat handlers
         def submit_message(message, history):
@@ -415,10 +428,15 @@ class UIComponents:
                 label=f"Active Servers ({len(server_choices)} loaded)"
             )
 
-        # Connect OAuth
+        self.api_key_box.input(
+            handle_api_key_update,
+            inputs=[self.api_key_box],
+            outputs=[self.api_key_status]
+        )
+
         demo.load(
-            fn=handle_oauth_profile,
-            outputs=[]
+            fn=initialize_api_key_status,
+            outputs=[self.api_key_status]
         )
         
         # Connect provider/model dropdowns with auto-selection on load