Update app.py

app.py

@@ -70,12 +70,22 @@ def load_review_table():
 
 
 def _empty_table():
-    return [[None] * 8]
+    import pandas as pd
+    return pd.DataFrame(
+        [["", "", "", 0, "", "yes", "", ""]],
+        columns=["#", "Topic Label", "Top Evidence", "Sentences", "Papers", "Approve", "Rename To", "Reasoning"],
+    )
 
 
 def _format_table(data, mode):
+    import pandas as pd
     rows = list(map(lambda item: _format_row(item, mode), data))
-    return rows if rows else _empty_table()
+    if not rows:
+        return _empty_table()
+    return pd.DataFrame(
+        rows,
+        columns=["#", "Topic Label", "Top Evidence", "Sentences", "Papers", "Approve", "Rename To", "Reasoning"],
+    )
 
 
 def _format_row(item, mode):
@@ -136,31 +146,55 @@ def get_download_files():
 
 # ── Table-to-theme-map parser ──────────────────────────────────────────────────
 def parse_table_to_message(table_data):
-    """Convert review table edits into a structured message for the agent."""
-    if not table_data:
+    """Convert review table edits into a structured message for the agent.
+    Handles both pandas DataFrame (from gr.Dataframe) and list of lists."""
+    import pandas as pd
+
+    # Normalise to list of lists regardless of input type
+    if table_data is None:
+        return "Submit Review: No table data provided."
+    if isinstance(table_data, pd.DataFrame):
+        if table_data.empty:
+            return "Submit Review: Table is empty, nothing to review."
+        rows = table_data.values.tolist()
+    else:
+        rows = list(table_data) if table_data else []
+
+    if not rows:
         return "Submit Review: No table data provided."
 
-    approved = list(filter(lambda row: len(row) >= 6 and str(row[5]).lower() in ("yes", "y", "1", "true"), table_data))
-    rejected = list(filter(lambda row: len(row) >= 6 and str(row[5]).lower() in ("no", "n", "0", "false"), table_data))
+    approved = list(filter(
+        lambda row: len(row) >= 6 and str(row[5]).strip().lower() in ("yes", "y", "1", "true"),
+        rows,
+    ))
+    rejected = list(filter(
+        lambda row: len(row) >= 6 and str(row[5]).strip().lower() in ("no", "n", "0", "false"),
+        rows,
+    ))
 
     theme_groups = {}
     list(map(
-        lambda row: theme_groups.setdefault(str(row[6]) if len(row) > 6 and row[6] else str(row[1]), []).append(int(row[0]) if str(row[0]).isdigit() else row[0]),
+        lambda row: theme_groups.setdefault(
+            str(row[6]).strip() if len(row) > 6 and row[6] and str(row[6]).strip() else str(row[1]),
+            []
+        ).append(int(row[0]) if str(row[0]).isdigit() else str(row[0])),
         approved,
     ))
 
     theme_map_str = json.dumps(theme_groups)
 
+    reasoning_lines = list(filter(None, list(map(
+        lambda row: f"  - Topic {row[0]} ({row[1]}): {row[7]}" if len(row) > 7 and str(row[7]).strip() else "",
+        approved,
+    ))))
+
     msg = (
         f"Submit Review received.\n\n"
         f"Approved topics: {len(approved)}\n"
         f"Rejected topics: {len(rejected)}\n\n"
         f"Theme groupings (RENAME TO → [topic_ids]):\n{theme_map_str}\n\n"
-        f"Researcher reasoning summary:\n"
-        + "\n".join(list(map(
-            lambda row: f"  - Topic {row[0]} ({row[1]}): {row[7]}" if len(row) > 7 and row[7] else "",
-            approved,
-        )))
+        f"Researcher reasoning:\n"
+        + ("\n".join(reasoning_lines) if reasoning_lines else "  (no reasoning provided)")
         + "\n\nPlease proceed to the next phase based on these decisions."
     )
     return msg
@@ -372,6 +406,7 @@ if __name__ == "__main__":
     demo.launch(
         server_name="0.0.0.0",
         server_port=7860,
+        ssr_mode=False,
         share=False,
         theme=gr.themes.Base(
             primary_hue="emerald",