From f35ccbbed52b1cf3acce8c39e6fb32c0017303d3 Mon Sep 17 00:00:00 2001
From: LUIS NOVO <lfnovo@gmail.com>
Date: Tue, 10 Jun 2025 11:55:58 -0300
Subject: [PATCH] chore: improve ux in the model provider page

---
 pages/7_🤖_Models.py | 25 +++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/pages/7_🤖_Models.py b/pages/7_🤖_Models.py
index 0736775..66dc766 100644
--- a/pages/7_🤖_Models.py
+++ b/pages/7_🤖_Models.py
@@ -78,6 +78,13 @@ st.divider()
 # Helper function to add model with auto-save
 def add_model_form(model_type, container_key):
     available_providers = esperanto_available_providers.get(model_type, [])
+    # Sort providers alphabetically for easier navigation
+    available_providers.sort()
+
+    # Remove perplexity from available_providers if it exists
+    if "perplexity" in available_providers:
+        available_providers.remove("perplexity")
+
     if not available_providers:
         st.info(f"No providers available for {model_type}")
         return
@@ -91,15 +98,11 @@ def add_model_form(model_type, container_key):
             key=f"provider_{model_type}_{container_key}",
         )
 
-        if model_type == "text_to_speech" and provider == "gemini":
-            model_name = "gemini-default"
-            st.markdown("Gemini models are pre-configured. Using the default model.")
-        else:
-            model_name = st.text_input(
-                "Model Name",
-                key=f"name_{model_type}_{container_key}",
-                help="gpt-4o-mini, claude, gemini, llama3, etc",
-            )
+        model_name = st.text_input(
+            "Model Name",
+            key=f"name_{model_type}_{container_key}",
+            help="gpt-4o-mini, claude, gemini, llama3, etc. For azure, use the deployment_name as the model_name",
+        )
 
         if st.form_submit_button("Add Model"):
             if model_name:
@@ -125,10 +128,12 @@ def handle_default_selection(
         setattr(default_models, key, selected_model.id)
         default_models.update()
         model_manager.refresh_defaults()
+        st.toast(f"Default {model_type} model set to {selected_model.name}")
     elif not selected_model and current_value:
         setattr(default_models, key, None)
         default_models.update()
         model_manager.refresh_defaults()
+        st.toast(f"Default {model_type} model removed")
 
     if caption:
         st.caption(caption)
@@ -150,7 +155,7 @@ for model in all_models:
 
 st.markdown("""
 **Model Management Guide:** For optimal performance, refer to [Which model to choose?](https://github.com/lfnovo/open-notebook/blob/main/docs/models.md) 
-You can test models in the [Transformations](https://try-it-out.open-notebook.com) page.
+You can test models in the [Transformations](Transformations) page.
 """)
 
 # Language Models Section