Update app.py

app.py

@@ -223,6 +223,26 @@ def load_journal() -> dict:
 # ------------------------------
 # 5) Small utilities (thumbs, resize, prefix/suffix, names)
 # ------------------------------
+# put near your other small utils
+def _normalize_table_value(table_value):
+    """Return a list-of-lists no matter what Gradio gives us."""
+    if table_value is None:
+        return []
+    # handle pandas.DataFrame without importing pandas explicitly
+    if hasattr(table_value, "to_dict") and hasattr(table_value, "values"):
+        try:
+            return table_value.reset_index(drop=True).values.tolist()
+        except Exception:
+            try:
+                return table_value.values.tolist()
+            except Exception:
+                return []
+    # sometimes Gradio hands a dict with "data"
+    if isinstance(table_value, dict) and "data" in table_value:
+        return table_value["data"] or []
+    # already a list (or something list-like)
+    return table_value
+
 def sanitize_basename(s: str) -> str:
     s = (s or "").strip() or "forgecaptions"
     return re.sub(r"[^A-Za-z0-9._-]+", "_", s)[:120]
@@ -477,7 +497,7 @@ def _rows_to_table(rows: List[dict]) -> list:
     return [[r.get("filename",""), r.get("caption","")] for r in (rows or [])]
 
 def _table_to_rows(table_value: Any, rows: List[dict]) -> List[dict]:
-    tbl = table_value or []
+    tbl = _normalize_table_value(table_value)
     new = []
     for i, r in enumerate(rows or []):
         r = dict(r)
@@ -488,7 +508,7 @@ def _table_to_rows(table_value: Any, rows: List[dict]) -> List[dict]:
     return new
 
 def export_csv_from_table(table_value: Any, dataset_name: str) -> str:
-    data = table_value or []
+    data = _normalize_table_value(table_value)
     name = sanitize_basename(dataset_name)
     out = f"/tmp/{name}_{int(time.time())}.csv"
     with open(out, "w", newline="", encoding="utf-8") as f:
@@ -520,7 +540,7 @@ def export_excel_with_thumbs(table_value: Any, session_rows: List[dict], thumb_p
         raise RuntimeError("Excel export requires 'openpyxl' in requirements.txt.") from e
 
     caption_by_file = {}
-    for row in (table_value or []):
+    for row in _normalize_table_value(table_value):
         if not row:
             continue
         fn = str(row[0]) if len(row) > 0 else ""
@@ -560,7 +580,7 @@ def export_txt_zip(table_value: Any, dataset_name: str) -> str:
     """
     Create one .txt per caption, zip them.
     """
-    data = table_value or []
+    data = _normalize_table_value(table_value)
     # wipe old
     for fn in os.listdir(TXT_EXPORT_DIR):
         try: