Make display_details robust

app.py

@@ -206,11 +206,12 @@ def load_details_dataframes(subtask, *model_ids):
     return [load_details_dataframe(model_id, subtask) for model_id in model_ids]
 
 
-def display_details(df_1, df_2, sample_idx):
-    s_1 = df_1.iloc[sample_idx]
-    s_2 = df_2.iloc[sample_idx]
+def display_details(sample_idx, *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
     # Pop model_name and add it to the column name
-    df = pd.concat([s.rename(s.pop("model_name")) for s in [s_1, s_2] if "model_name" in s], axis="columns")
+    df = pd.concat([row.rename(row.pop("model_name")) for row in rows], axis="columns")
     return (
         df.style
         .format(na_rep="")
@@ -309,12 +310,12 @@ with gr.Blocks(fill_height=True) as demo:
         outputs=[details_dataframe_1, details_dataframe_2],
     ).then(
         fn=display_details,
-        inputs=[details_dataframe_1, details_dataframe_2, sample_idx],
+        inputs=[sample_idx, details_dataframe_1, details_dataframe_2],
         outputs=details,
     )
     sample_idx.change(
         fn=display_details,
-        inputs=[details_dataframe_1, details_dataframe_2, sample_idx],
+        inputs=[sample_idx, details_dataframe_1, details_dataframe_2],
         outputs=details,
     )