Update app.py

app.py

@@ -70,7 +70,7 @@ def flatten_models(model_tree):
 
 
 # -----------------------------
-# Add a new model
+# Add a new model (from the box)
 # -----------------------------
 def add_model_box(
     category,
@@ -179,21 +179,17 @@ def respond(
 def build_model_tree(
     models,
     active_model_state,
-    current_model_label,
-    info_markdowns,
-    use_buttons
+    current_model_label
 ):
     """
     models: merged models dict (Category -> Family -> Model -> meta)
     active_model_state: gr.State storing current active full key
     current_model_label: gr.Markdown for 'Current model: ...'
-    info_markdowns: dict full_key -> gr.Markdown (Model Info)
-    use_buttons: dict full_key -> gr.Button (Use this model)
     """
-
     flat = flatten_models(models)
 
-    # We’ll build the UI, and inside each loop create small closures
+    # We will keep references to inner model-info components
+    # to wire callbacks in a simpler way.
     for category, families in models.items():
         with gr.Accordion(category, open=False):
             for family, model_dict in families.items():
@@ -202,69 +198,61 @@ def build_model_tree(
                         emoji = meta.get("emoji", "✨")
                         full_key = f"{category} / {family} / {model_name}"
 
-                        # Model button row
-                        model_button = gr.Button(f"{emoji} {model_name}", size="sm")
-
-                        # Model info area
-                        info_md = gr.Markdown(visible=False)
-                        info_markdowns[full_key] = info_md
-
-                        # Use button
-                        use_btn = gr.Button("Use this model", visible=False, size="sm")
-                        use_buttons[full_key] = use_btn
-
-                        # ---- Click model button → show this info, hide others
-                        def show_info(clicked_key, fk=full_key):
-                            models_local = merge_models()
-                            flat_local = flatten_models(models_local)
-
-                            updates = {}
-                            for key, (meta_loc, _, _, _) in flat_local.items():
-                                md_out = info_markdowns.get(key)
-                                btn_out = use_buttons.get(key)
-                                if md_out is None or btn_out is None:
-                                    continue
-
-                                if key == fk:
+                        # Model accordion
+                        with gr.Accordion(f"{emoji} {model_name}", open=False):
+                            # Nested Model Info accordion
+                            with gr.Accordion("Model Info", open=False):
+                                # Filled dynamically via a function
+                                info_md = gr.Markdown("")
+                                use_btn = gr.Button("Use this model", size="sm")
+
+                                # Function to populate Model Info
+                                def load_model_info(fk=full_key):
+                                    models_local = merge_models()
+                                    flat_local = flatten_models(models_local)
+                                    meta_loc_tuple = flat_local.get(fk)
+                                    if not meta_loc_tuple:
+                                        return gr.Markdown.update(
+                                            value="No info available for this model."
+                                        )
+                                    meta_loc, _, _, _ = meta_loc_tuple
                                     text = (
                                         f"**Model ID:** `{meta_loc['id']}`  \n"
                                         f"**Description:** {meta_loc['description']}  \n"
                                         f"[Model card]({meta_loc['link']})"
                                     )
-                                    updates[md_out] = gr.Markdown.update(value=text, visible=True)
-                                    updates[btn_out] = gr.Button.update(visible=True)
-                                else:
-                                    updates[md_out] = gr.Markdown.update(visible=False)
-                                    updates[btn_out] = gr.Button.update(visible=False)
-                            return updates
-
-                        model_button.click(
-                            show_info,
-                            inputs=active_model_state,
-                            outputs=list(info_markdowns.values()) + list(use_buttons.values()),
-                        )
-
-                        # ---- Click "Use this model" → set active model + label
-                        def use_model(fk=full_key):
-                            models_local = merge_models()
-                            flat_local = flatten_models(models_local)
-                            meta_loc_tuple = flat_local.get(fk)
-
-                            if not meta_loc_tuple:
-                                return fk, gr.Markdown.update(
-                                    value="**Current model:** _none selected_"
+                                    return gr.Markdown.update(value=text)
+
+                                # Populate Model Info when this accordion is opened
+                                # Gradio doesn't have a direct 'on open' event,
+                                # so we just pre-fill it once on load:
+                                info_md.load(
+                                    load_model_info,
+                                    inputs=None,
+                                    outputs=info_md,
                                 )
 
-                            meta_loc, _, _, mname = meta_loc_tuple
-                            emoji_local = meta_loc.get("emoji", "✨")
-                            label_text = f"**Current model:** {emoji_local} {mname}"
-                            return fk, gr.Markdown.update(value=label_text)
-
-                        use_btn.click(
-                            use_model,
-                            inputs=None,
-                            outputs=[active_model_state, current_model_label],
-                        )
+                                # Function to use this model
+                                def use_model(fk=full_key):
+                                    models_local = merge_models()
+                                    flat_local = flatten_models(models_local)
+                                    meta_loc_tuple = flat_local.get(fk)
+
+                                    if not meta_loc_tuple:
+                                        return fk, gr.Markdown.update(
+                                            value="**Current model:** _none selected_"
+                                        )
+
+                                    meta_loc, _, _, mname = meta_loc_tuple
+                                    emoji_local = meta_loc.get("emoji", "✨")
+                                    label_text = f"**Current model:** {emoji_local} {mname}"
+                                    return fk, gr.Markdown.update(value=label_text)
+
+                                use_btn.click(
+                                    use_model,
+                                    inputs=None,
+                                    outputs=[active_model_state, current_model_label],
+                                )
 
 
 # -----------------------------
@@ -330,10 +318,6 @@ with gr.Blocks() as demo:
         # Current model label under the Add box
         current_model_label = gr.Markdown("**Current model:** _none selected_")
 
-        # Structures to hold info areas
-        info_markdowns = {}
-        use_buttons = {}
-
         gr.Markdown("### Models")
 
         # Build nested accordions for models
@@ -341,8 +325,6 @@ with gr.Blocks() as demo:
             models_tree,
             active_model_state=active_model_key,
             current_model_label=current_model_label,
-            info_markdowns=info_markdowns,
-            use_buttons=use_buttons,
         )
 
     # Main chat interface