Add toggle_label field to AnalysisCardBase for custom subtitle expansion text (#5124)

Summary:
Pull Request resolved: #5124

Adds a serializable `toggle_label: str` field to `AnalysisCardBase` that allows
analyses to customize the subtitle expand/collapse toggle button text.

When non-empty, `toggle_label` replaces the default "See more" text with
a context-specific label (e.g., "Expand to see annotated parameters.").
The field is persisted to both SQA and JSON storage backends, and used by
the notebook HTML template for rendering.

Changes:
- Add `toggle_label` field + constructor param (default "") to AnalysisCardBase
- Update notebook `_to_html()` to use `self.toggle_label or "See more"`
- Add `toggle_label` nullable column to SQAAnalysisCard
- Update SQA encoder/decoder (all 8 card-type callsites)
- Update JSON encoder for both card and group dicts
- Unit tests

Differential Revision: D98738752

Author: shrutipatel31

ax/core/analysis_card.py

@@ -54,22 +54,24 @@
     display: none;
 }}
 </style>
-<div class="card">
+<div class="card" id="{card_id}">
     <div class="card-header">
         <b>{title_str}</b>
         <div class="card-subtitle">{subtitle_str}</div>
-        <button class="card-subtitle-toggle">See more</button>
+        <button class="card-subtitle-toggle">{toggle_text}</button>
     </div>
     <div class="card-body">
         {body_html}
     </div>
 </div>
 <script>
 (function() {{
-    var card = document.currentScript.previousElementSibling;
+    var card = document.getElementById('{card_id}');
+    if (!card) return;
     var subtitle = card.querySelector('.card-subtitle');
     var toggle = card.querySelector('.card-subtitle-toggle');
     if (subtitle && toggle) {{
+        var originalText = toggle.textContent;
         requestAnimationFrame(function() {{
             if (subtitle.scrollHeight > subtitle.clientHeight
                     || subtitle.scrollWidth > subtitle.clientWidth) {{
@@ -81,7 +83,7 @@
                 subtitle.style.maxHeight = '1.4em';
                 subtitle.style.whiteSpace = 'nowrap';
                 subtitle.style.textOverflow = 'ellipsis';
-                toggle.textContent = 'See more';
+                toggle.textContent = originalText;
             }} else {{
                 subtitle.style.maxHeight = 'none';
                 subtitle.style.whiteSpace = 'normal';
@@ -154,6 +156,7 @@ class AnalysisCardBase(SortableBase, ABC):
 
     title: str
     subtitle: str
+    subtitle_toggle_label: str
 
     _timestamp: datetime
 
@@ -163,6 +166,7 @@ def __init__(
         title: str,
         subtitle: str,
         timestamp: datetime | None = None,
+        subtitle_toggle_label: str = "",
     ) -> None:
         """
         Args:
@@ -175,10 +179,15 @@ def __init__(
             timestamp: The time at which the Analysis was computed. This can be
                 especially useful when querying the database for the most recently
                 produced artifacts.
+            subtitle_toggle_label: Custom label for the subtitle
+                expansion toggle. When non-empty, replaces the default
+                "See more" text. Persisted to storage and used by both
+                notebook and web UI rendering.
         """
         self.name = name
         self.title = title
         self.subtitle = subtitle
+        self.subtitle_toggle_label = subtitle_toggle_label
         self._timestamp = timestamp if timestamp is not None else datetime.now()
 
     @abstractmethod
@@ -258,10 +267,16 @@ def _repr_html_(self) -> str:
         return plotlyjs_script + self._to_html(depth=0)
 
     def _to_html(self, depth: int) -> str:
+        toggle_text = self.subtitle_toggle_label or "See more"
+        # Use id(self) so each card gets a unique HTML element ID even when
+        # multiple cards share the same name.
+        card_id = f"ax-card-{id(self)}"
         return html_card_template.format(
+            card_id=card_id,
             title_str=self.title,
             subtitle_str=self.subtitle,
             body_html=self._body_html(depth=depth),
+            toggle_text=toggle_text,
         )
 
 
@@ -282,6 +297,7 @@ def __init__(
         subtitle: str | None,
         children: Sequence[AnalysisCardBase],
         timestamp: datetime | None = None,
+        subtitle_toggle_label: str = "",
     ) -> None:
         """
         Args:
@@ -294,12 +310,15 @@ def __init__(
             timestamp: The time at which the Analysis was computed. This can be
                 especially useful when querying the database for the most recently
                 produced artifacts.
+            subtitle_toggle_label: Custom label for the subtitle expansion
+                toggle. When non-empty, replaces the default "See more" text
         """
         super().__init__(
             name=name,
             title=title,
             subtitle=subtitle if subtitle is not None else "",
             timestamp=timestamp,
+            subtitle_toggle_label=subtitle_toggle_label,
         )
 
         self.children = [
@@ -404,6 +423,7 @@ def __init__(
         df: pd.DataFrame,
         blob: str,
         timestamp: datetime | None = None,
+        subtitle_toggle_label: str = "",
     ) -> None:
         """
         Args:
@@ -417,12 +437,15 @@ def __init__(
             timestamp: The time at which the Analysis was computed. This can be
                 especially useful when querying the database for the most recently
                 produced artifacts.
+            subtitle_toggle_label: Custom label for the subtitle expansion
+                toggle. When non-empty, replaces the default "See more" text
         """
         super().__init__(
             name=name,
             title=title,
             subtitle=subtitle,
             timestamp=timestamp,
+            subtitle_toggle_label=subtitle_toggle_label,
         )
 
         self.df = df

ax/core/tests/test_analysis_card.py

@@ -14,35 +14,36 @@
 from plotly import graph_objects as go, io as pio
 
 
-class TestAnalysisCard(TestCase):
-    def test_hierarchy_str(self) -> None:
-        test_df = pd.DataFrame(
-            columns=["a", "b"],
-            data=[
-                [1, 2],
-                [3, 4],
-            ],
-        )
+DUMMY_DF: pd.DataFrame = pd.DataFrame(
+    columns=["a", "b"],
+    data=[[1, 2], [3, 4]],
+)
+
 
-        base_analysis_card = AnalysisCard(
+class TestAnalysisCard(TestCase):
+    def setUp(self) -> None:
+        super().setUp()
+        self.base_card = AnalysisCard(
             name="test_base_analysis_card",
             title="test_base_analysis_card_title",
             subtitle="test_subtitle",
-            df=test_df,
+            df=DUMMY_DF,
             blob="test blob",
         )
+
+    def test_hierarchy_str(self) -> None:
         markdown_analysis_card = MarkdownAnalysisCard(
             name="test_markdown_analysis_card",
             title="test_markdown_analysis_card_title",
             subtitle="test_subtitle",
-            df=test_df,
+            df=DUMMY_DF,
             blob="This is some **really cool** markdown",
         )
         plotly_analysis_card = PlotlyAnalysisCard(
             name="test_plotly_analysis_card",
             title="test_plotly_analysis_card_title",
             subtitle="test_subtitle",
-            df=test_df,
+            df=DUMMY_DF,
             blob=pio.to_json(go.Figure()),
         )
 
@@ -51,7 +52,7 @@ def test_hierarchy_str(self) -> None:
             name="small_group",
             title="Small Group",
             subtitle="This is a small group with just a few cards",
-            children=[base_analysis_card, markdown_analysis_card],
+            children=[self.base_card, markdown_analysis_card],
         )
         big_group = AnalysisCardGroup(
             name="big_group",
@@ -78,3 +79,47 @@ def test_not_applicable_card(self) -> None:
             blob="Explanation text.",
         )
         self.assertIn("Explanation text.", card._body_html(depth=0))
+
+    def test_subtitle_toggle_label_rendering(self) -> None:
+        """Verify subtitle_toggle_label controls toggle button text in HTML."""
+        for label, expected_text in (
+            ("", "See more"),
+            (
+                "Expand to see annotated parameters.",
+                "Expand to see annotated parameters.",
+            ),
+        ):
+            with self.subTest(label=label):
+                card = AnalysisCard(
+                    name="Test",
+                    title="Title",
+                    subtitle="A long subtitle",
+                    df=pd.DataFrame(),
+                    blob="blob",
+                    subtitle_toggle_label=label,
+                )
+                self.assertEqual(card.subtitle_toggle_label, label)
+                html = card._repr_html_()
+                self.assertIn(expected_text, html)
+
+    def test_analysis_card_group_html_does_not_render_toggle(self) -> None:
+        """AnalysisCardGroup._to_html uses html_group_card_template which renders
+        the subtitle as a plain <p> tag (no collapsible toggle). Verify the group's
+        own subtitle_toggle_label is stored but not rendered in the group header."""
+
+        group = AnalysisCardGroup(
+            name="G",
+            title="GT",
+            subtitle="GS",
+            children=[self.base_card],
+            subtitle_toggle_label="Custom toggle.",
+        )
+        self.assertEqual(group.subtitle_toggle_label, "Custom toggle.")
+
+        html = group._to_html(depth=0)
+
+        # The group template uses a plain <p> for subtitles, not the
+        # collapsible card-subtitle + toggle-button pattern.
+        self.assertNotIn("Custom toggle.", html)
+        self.assertIn('<p class="group-subtitle">', html)
+        self.assertIn("GS", html)

ax/storage/json_store/encoders.py

@@ -89,6 +89,7 @@ def analysis_card_to_dict(card: AnalysisCard) -> dict[str, Any]:
         "df": card.df,
         "blob": card.blob,
         "timestamp": card._timestamp,
+        "subtitle_toggle_label": card.subtitle_toggle_label,
     }
 
 
@@ -101,6 +102,7 @@ def analysis_card_group_to_dict(group: AnalysisCardGroup) -> dict[str, Any]:
         "subtitle": group.subtitle,
         "children": group.children,
         "timestamp": group._timestamp,
+        "subtitle_toggle_label": group.subtitle_toggle_label,
     }
 
 

ax/storage/json_store/tests/test_json_store.py

@@ -426,6 +426,7 @@
             subtitle="subtitle",
             df=pd.DataFrame({"a": [1, 2]}),
             blob="blob_str",
+            subtitle_toggle_label="Expand to see details.",
         ),
     ),
     (
@@ -486,6 +487,7 @@
             subtitle="na subtitle",
             df=pd.DataFrame(),
             blob="Not enough data.",
+            subtitle_toggle_label="Expand to see why.",
         ),
     ),
     (
@@ -510,6 +512,7 @@
                     blob="# md",
                 ),
             ],
+            subtitle_toggle_label="Expand to see children.",
         ),
     ),
 ]
@@ -649,6 +652,64 @@ def test_EncodeDecode(self) -> None:
                 else:
                     raise e
 
+    def test_EncodeDecodeAnalysisCardSubtitleToggleLabel(self) -> None:
+        """Verify decoding old JSON missing subtitle_toggle_label falls back to
+        the constructor default (backwards compatibility).
+        """
+
+        # GIVEN old AnalysisCard JSON missing the subtitle_toggle_label key
+        with self.subTest(msg="backward compatible - AnalysisCard"):
+            restored = object_from_json(
+                {
+                    "__type": "AnalysisCard",
+                    "name": "OldCard",
+                    "title": "T",
+                    "subtitle": "S",
+                    "df": {"__type": "DataFrame", "value": "{}"},
+                    "blob": "b",
+                    "timestamp": {
+                        "__type": "datetime",
+                        "value": "2025-01-01 00:00:00.000000",
+                    },
+                },
+                decoder_registry=CORE_DECODER_REGISTRY,
+                class_decoder_registry=CORE_CLASS_DECODER_REGISTRY,
+            )
+            self.assertEqual(restored.subtitle_toggle_label, "")
+
+        # GIVEN old AnalysisCardGroup JSON missing subtitle_toggle_label
+        with self.subTest(msg="backward compatible - AnalysisCardGroup"):
+            restored_group = object_from_json(
+                {
+                    "__type": "AnalysisCardGroup",
+                    "name": "OldGroup",
+                    "title": "GT",
+                    "subtitle": "GS",
+                    "children": [
+                        {
+                            "__type": "AnalysisCard",
+                            "name": "C",
+                            "title": "CT",
+                            "subtitle": "CS",
+                            "df": {"__type": "DataFrame", "value": "{}"},
+                            "blob": "b",
+                            "timestamp": {
+                                "__type": "datetime",
+                                "value": "2025-01-01 00:00:00.000000",
+                            },
+                        }
+                    ],
+                    "timestamp": {
+                        "__type": "datetime",
+                        "value": "2025-01-01 00:00:00.000000",
+                    },
+                },
+                decoder_registry=CORE_DECODER_REGISTRY,
+                class_decoder_registry=CORE_CLASS_DECODER_REGISTRY,
+            )
+            self.assertEqual(restored_group.subtitle_toggle_label, "")
+            self.assertEqual(restored_group.children[0].subtitle_toggle_label, "")
+
     def test_EncodeDecode_dataclass_with_initvar(self) -> None:
         @dataclasses.dataclass
         class TestDataclass: