Update app.py

app.py

@@ -238,16 +238,16 @@ def render_sample_view(df_filtered: pd.DataFrame, index: int):
     meta_raw = row.get("metadata", "")
     meta_raw = meta_raw if isinstance(meta_raw, str) else str(meta_raw)
 
-    # Gallery (label, image)
+    # Gallery: Gradio expects (image, caption) per item
     gallery_items = [
-        ("Marked / Original", marked),
-        ("Original", original),
-        ("Solution", sol),
-        ("Mask", mask),
-        ("Cell map", cell_map),
+        (marked, "Marked / Original"),
+        (original, "Original"),
+        (sol, "Solution"),
+        (mask, "Mask"),
+        (cell_map, "Cell map"),
     ]
     # remove None images
-    gallery_items = [(t, img) for (t, img) in gallery_items if img is not None]
+    gallery_items = [(img, cap) for (img, cap) in gallery_items if img is not None]
 
     status_md = f"**Sample** `{sid}`  \n**Index** `{index}` / `{len(df_filtered)-1}`"
     return index, status_md, instruction, gallery_items, meta_json, meta_raw