Update app.py via AI Editor

app.py

@@ -4,6 +4,7 @@ import io
 import dash
 from dash import dcc, html, Input, Output, State, callback_context, ALL
 import dash_bootstrap_components as dbc
+import dash.dash_table as dt
 import pandas as pd
 import logging
 from docx import Document
@@ -843,7 +844,32 @@ def unified_master_callback(
             if triggered_id == "shred-action-btn":
                 action_name = "shred"
                 result, generated_filename, generated_xlsx_bytes, _, _ = process_document(sess_data, action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None)
-                output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
+                # Try to parse as markdown table, show DataTable if possible, else fallback to Markdown
+                try:
+                    df = parse_markdown_table(result)
+                    output_data_upload = dt.DataTable(
+                        columns=[{"name": i, "id": i} for i in df.columns],
+                        data=df.to_dict('records'),
+                        style_table={
+                            "overflowX": "auto",
+                            "overflowY": "auto",
+                            "maxHeight": "60vh",
+                            "minWidth": "100%"
+                        },
+                        style_cell={
+                            "whiteSpace": "normal",
+                            "height": "auto",
+                            "textAlign": "left",
+                            "wordBreak": "break-word"
+                        },
+                        style_header={
+                            "fontWeight": "bold"
+                        },
+                        page_action="none"
+                    )
+                except Exception as e:
+                    logging.warning(f"Failed to parse markdown as table for preview: {e}")
+                    output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
             elif triggered_id == "compliance-action-btn":
                 action_name = "compliance"
                 result, generated_filename, generated_xlsx_bytes, _, _ = process_document(