Update app.py

app.py

@@ -207,10 +207,11 @@ def respond(
 # -----------------------------
 # Use model helper
 # -----------------------------
-def use_model(fk, current_model_label):
+def use_model(fk, old_fk):
     """
-    fk: full key "Category / Family / Model"
-    current_model_label: the Markdown component instance to update
+    fk: full key "Category / Family / Model" (from gr.State(full_key))
+    old_fk: previous active model key (from active_model_state)
+    Returns: (new_active_key, current_model_label_text)
     """
     try:
         models_local = merge_models()
@@ -219,24 +220,20 @@ def use_model(fk, current_model_label):
 
         if not meta_loc_tuple:
             debug(f"use_model: key not found: {fk}")
-            return fk, current_model_label.update(
-                value="**Current model:** _none selected_"
-            )
+            return old_fk, "**Current model:** _none selected_"
 
         meta_loc, _, _, mname = meta_loc_tuple
         emoji_local = meta_loc.get("emoji", "✨")
         label_text = f"**Current model:** {emoji_local} {mname}"
 
         debug(f"use_model: selected {fk}")
-        return fk, current_model_label.update(value=label_text)
+        return fk, label_text
 
     except Exception as e:
         import traceback
         tb = traceback.format_exc()
         debug(f"ERROR in use_model:\n{tb}")
-        return fk, current_model_label.update(
-            value="**Current model:** _error occurred (see Debug Log)_"
-        )
+        return old_fk, "**Current model:** _error occurred (see Debug Log)_"
 
 
 # -----------------------------
@@ -277,11 +274,12 @@ def build_model_tree(
                                 # Wire button -> use_model
                                 use_btn.click(
                                     use_model,
-                                    inputs=[gr.State(full_key), current_model_label],
+                                    inputs=[gr.State(full_key), active_model_state],
                                     outputs=[active_model_state, current_model_label],
                                 )
 
 
+
 # -----------------------------
 # Build the UI
 # -----------------------------