Update ui_components.py

ui_components.py

@@ -404,7 +404,14 @@ class UIComponents:
                 return gr.Dropdown(choices=[], value=None), "⚪ Select provider first", gr.Group(visible=False)
             
             available_models = AppConfig.get_available_models_for_provider(provider_id)
-            model_choices = [(AppConfig.AVAILABLE_MODELS[model]["name"], model) for model in available_models]
+            def _label(model_id: str) -> str:
+                info = AppConfig.AVAILABLE_MODELS.get(model_id, {})
+                name = info.get("name", model_id)
+                vision = info.get("supports_vision", False)
+                family = "GPT‑OSS" if AppConfig.is_gpt_oss_model(model_id) else ("Qwen VL" if vision else "Model")
+                cap = "Supports Vision" if vision else "Text Only"
+                return f"{name} · {cap}"
+            model_choices = [(_label(model), model) for model in available_models]
             
             # Auto-select 120b model if available
             default_model = "openai/gpt-oss-120b" if "openai/gpt-oss-120b" in available_models else (available_models[0] if available_models else None)
@@ -435,9 +442,10 @@ class UIComponents:
             model_info = AppConfig.AVAILABLE_MODELS.get(model_id, {})
             context_length = model_info.get("context_length", 128000)
             active_params = model_info.get("active_params", "N/A")
+            cap = "Supports Vision" if model_info.get("supports_vision", False) else "Text Only"
             
             if self.mcp_client.hf_client:
-                status = f"✅ Ready! Using {active_params} active params, {context_length:,} token context"
+                status = f"✅ Ready! Using {active_params} active params, {context_length:,} token context · {cap}"
             else:
                 status = "❌ Please add your Hugging Face API token"