Update ui_components.py

ui_components.py

@@ -50,9 +50,10 @@ class UIComponents:
     
     def create_interface(self) -> gr.Blocks:
         """Create the main Gradio interface with improved layout"""
+        theme = self._load_theme()
         with gr.Blocks(
             title="Universal MCP Client - HF Inference Powered", 
-            theme=getattr(gr.themes, AppConfig.GRADIO_THEME.title())(),
+            theme=theme,
             fill_height=True,
             css=CUSTOM_CSS
         ) as demo:
@@ -67,6 +68,16 @@ class UIComponents:
             self._setup_event_handlers(chatbot, demo)
             
         return demo
+
+    def _load_theme(self):
+        """Load the preferred Gradio theme with fallback handling"""
+        try:
+            return gr.themes.Theme.from_hub(AppConfig.GRADIO_THEME)
+        except Exception as exc:
+            logger.warning(f"Custom theme load failed: {exc}; using fallback theme")
+            fallback_name = getattr(AppConfig, "GRADIO_FALLBACK_THEME", "ocean")
+            fallback_attr = getattr(gr.themes, fallback_name.title(), gr.themes.Ocean)
+            return fallback_attr()
     
     def _create_sidebar(self):
         """Create the sidebar with login, provider/model selection, and server management"""
@@ -102,9 +113,7 @@ class UIComponents:
         """Create secret input section for Hugging Face API keys"""
         with gr.Group(elem_classes="login-section"):
             gr.Markdown("""
-            **🔐 Hugging Face Inference Token**
-            
-            Paste your API token below. It stays in your browser session only and is not stored.
+            **🔐 HF Token**
             """)
             self.api_key_box = gr.Textbox(
                 label="HF API Token",