Add checkbox in Details to show only differences

app.py

@@ -105,6 +105,7 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
             sample_idx = gr.Number(
                 label="Sample Index", info="Index of the sample to be displayed", value=0, minimum=0, visible=False
             )
+            details_show_only_differences = gr.Checkbox(label="Show Only Differences", value=False, info="Options")
             details = gr.HTML()
             details_dataframe_1 = gr.Dataframe(visible=False)
             details_dataframe_2 = gr.Dataframe(visible=False)
@@ -197,9 +198,14 @@ with gr.Blocks(fill_height=True, fill_width=True) as demo:
         outputs=sample_idx,
     )
     gr.on(
-        triggers=[details_dataframe_1.change, details_dataframe_2.change, sample_idx.change],
+        triggers=[
+            details_dataframe_1.change,
+            details_dataframe_2.change,
+            sample_idx.change,
+            details_show_only_differences.change,
+        ],
         fn=display_details,
-        inputs=[sample_idx, details_dataframe_1, details_dataframe_2],
+        inputs=[sample_idx, details_show_only_differences, details_dataframe_1, details_dataframe_2],
         outputs=details,
     )
     clear_details_btn.click(

src/details.py

@@ -66,7 +66,7 @@ async def load_details_dataframes(subtask, *model_ids):
     return result
 
 
-def display_details(sample_idx, *dfs):
+def display_details(sample_idx, show_only_differences, *dfs):
     rows = [df.iloc[sample_idx] for df in dfs if "model_name" in df.columns and sample_idx < len(df)]
     if not rows:
         return
@@ -82,10 +82,14 @@ def display_details(sample_idx, *dfs):
             return row
 
     df = df.apply(wrap, axis=1)
+    if show_only_differences:
+        any_difference = df.ne(df.iloc[:, 0], axis=0).any(axis=1)
     # Style
     return (
         df.style.format(escape="html", na_rep="")
         # .hide(axis="index")
+        # Hide non-different rows
+        .hide([row for row in df.index if show_only_differences and not any_difference[row]])
         .to_html()
     )