Update app.py via AI Editor

app.py

@@ -3,19 +3,15 @@ import io
 import os
 import pandas as pd
 from docx import Document
-from io import BytesIO, StringIO
+from io import BytesIO
 import dash
 import dash_bootstrap_components as dbc
-from dash import html, dcc, Input, Output, State, callback_context, MATCH, ALL
+from dash import html, dcc, Input, Output, State, callback_context, ALL
 from dash.dash_table import DataTable
-from docx.shared import Pt
-from docx.enum.style import WD_STYLE_TYPE
 from PyPDF2 import PdfReader
 import logging
 import threading
-import re 
-import markdown
-from bs4 import BeautifulSoup
+import re
 import google.generativeai as genai
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s %(levelname)s %(message)s')
@@ -98,6 +94,8 @@ def markdown_table_to_df(md_table):
     return df
 
 def markdown_table_preview(md_text):
+    if not md_text:
+        return html.Div("No table found.")
     tables = extract_markdown_tables(md_text)
     if not tables:
         return html.Div("No table found.")
@@ -121,7 +119,7 @@ def markdown_table_preview(md_text):
     return html.Div(table_divs)
 
 def markdown_narrative_preview(md_text):
-    return html.Div(dcc.Markdown(md_text, dangerously_allow_html=True, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'sans-serif'}))
+    return html.Div(dcc.Markdown(md_text or "", dangerously_allow_html=True, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'sans-serif'}))
 
 def markdown_tables_to_xlsx(md_text):
     tables = extract_markdown_tables(md_text)
@@ -232,8 +230,7 @@ def get_left_col_content():
         chat_card
     ]
 
-def get_right_col_content(selected_type, 
-                          shred_doc, pink_doc, pink_review_doc, red_doc, red_review_doc, gold_doc):
+def get_right_col_content(selected_type, shred_doc, pink_doc, pink_review_doc, red_doc, red_review_doc, gold_doc):
     controls = []
     controls.append(html.Div([
         html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
@@ -449,614 +446,45 @@ def update_selected_doc_type(n_clicks_list, btn_ids):
 def update_right_col(selected_type, shred_doc, pink_doc, pink_review_doc, red_doc, red_review_doc, gold_doc):
     return get_right_col_content(selected_type, shred_doc, pink_doc, pink_review_doc, red_doc, red_review_doc, gold_doc)
 
-def get_all_lengths(*args):
-    return [len(x) if isinstance(x, list) else 0 for x in args]
-
 @app.callback(
     Output('file-list', 'children'),
     Output('store-shred', 'data'),
-    Output({'type': f'uploaded-doc-name-shred', 'index': ALL}, 'children'),
-    Output({'type': f'uploaded-doc-name-pink', 'index': ALL}, 'children'),
-    Output({'type': f'uploaded-doc-name-pink_review', 'index': ALL}, 'children'),
-    Output({'type': f'uploaded-doc-name-red', 'index': ALL}, 'children'),
-    Output({'type': f'uploaded-doc-name-red_review', 'index': ALL}, 'children'),
-    Output({'type': f'uploaded-doc-name-gold', 'index': ALL}, 'children'),
-    Output({'type': f'upload-doc-type-shred', 'index': ALL}, 'contents'),
-    Output({'type': f'upload-doc-type-pink', 'index': ALL}, 'contents'),
-    Output({'type': f'upload-doc-type-pink_review', 'index': ALL}, 'contents'),
-    Output({'type': f'upload-doc-type-red', 'index': ALL}, 'contents'),
-    Output({'type': f'upload-doc-type-red_review', 'index': ALL}, 'contents'),
-    Output({'type': f'upload-doc-type-gold', 'index': ALL}, 'contents'),
-    Output({'type': f'radio-doc-source-shred', 'index': ALL}, 'value'),
-    Output({'type': f'radio-doc-source-pink', 'index': ALL}, 'value'),
-    Output({'type': f'radio-doc-source-pink_review', 'index': ALL}, 'value'),
-    Output({'type': f'radio-doc-source-red', 'index': ALL}, 'value'),
-    Output({'type': f'radio-doc-source-red_review', 'index': ALL}, 'value'),
-    Output({'type': f'radio-doc-source-gold', 'index': ALL}, 'value'),
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
-    Output('store-pink', 'data'),
-    Output('store-pink-review', 'data'),
-    Output('store-red', 'data'),
-    Output('store-red-review', 'data'),
-    Output('store-gold', 'data'),
-    Output('store-gold-review', 'data'),
-    Output('store-loe', 'data'),
-    Output('store-virtual-board', 'data'),
-    Output('chat-output', 'children'),
-    Output("download-document", "data"),
     Input('upload-document', 'contents'),
     State('upload-document', 'filename'),
     State('file-list', 'children'),
     State('store-shred', 'data'),
-    Input({'type': 'remove-file', 'index': ALL}, 'n_clicks'),
-    State('file-list', 'children'),
-    State('store-shred', 'data'),
-    Input({'type': f'upload-doc-type-shred', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-shred', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-shred', 'index': ALL}, 'id'),
-    Input({'type': f'upload-doc-type-pink', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-pink', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-pink', 'index': ALL}, 'id'),
-    Input({'type': f'upload-doc-type-pink_review', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-pink_review', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-pink_review', 'index': ALL}, 'id'),
-    Input({'type': f'upload-doc-type-red', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-red', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-red', 'index': ALL}, 'id'),
-    Input({'type': f'upload-doc-type-red_review', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-red_review', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-red_review', 'index': ALL}, 'id'),
-    Input({'type': f'upload-doc-type-gold', 'index': ALL}, 'contents'),
-    State({'type': f'upload-doc-type-gold', 'index': ALL}, 'filename'),
-    State({'type': f'upload-doc-type-gold', 'index': ALL}, 'id'),
-    Input({'type': 'btn-generate-doc', 'index': ALL}, 'n_clicks'),
-    State({'type': 'btn-generate-doc', 'index': ALL}, 'id'),
-    State({'type': 'radio-doc-source-shred', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-shred', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-shred', 'index': ALL}, 'filename'),
-    State({'type': 'radio-doc-source-pink', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-pink', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-pink', 'index': ALL}, 'filename'),
-    State({'type': 'radio-doc-source-pink_review', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-pink_review', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-pink_review', 'index': ALL}, 'filename'),
-    State({'type': 'radio-doc-source-red', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-red', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-red', 'index': ALL}, 'filename'),
-    State({'type': 'radio-doc-source-red_review', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-red_review', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-red_review', 'index': ALL}, 'filename'),
-    State({'type': 'radio-doc-source-gold', 'index': ALL}, 'value'),
-    State({'type': 'upload-doc-type-gold', 'index': ALL}, 'contents'),
-    State({'type': 'upload-doc-type-gold', 'index': ALL}, 'filename'),
-    State('store-shred', 'data'),
-    State('store-pink', 'data'),
-    State('store-pink-review', 'data'),
-    State('store-red', 'data'),
-    State('store-red-review', 'data'),
-    State('store-gold', 'data'),
-    State('store-gold-review', 'data'),
-    State('store-loe', 'data'),
-    State('store-virtual-board', 'data'),
-    Input('btn-send-chat', 'n_clicks'),
-    Input('btn-clear-chat', 'n_clicks'),
-    State('chat-input', 'value'),
-    State('selected-doc-type', 'data'),
-    State('document-preview', 'children'),
-    Input("btn-download", "n_clicks"),
-    State('selected-doc-type', 'data'),
-    State('store-shred', 'data'),
-    State('store-pink', 'data'),
-    State('store-pink-review', 'data'),
-    State('store-red', 'data'),
-    State('store-red-review', 'data'),
-    State('store-gold', 'data'),
-    State('store-gold-review', 'data'),
-    State('store-loe', 'data'),
-    State('store-virtual-board', 'data'),
     prevent_initial_call=True
 )
-def master_callback(
-    upload_contents, upload_filenames, existing_files, current_shred,
-    remove_n_clicks, remove_existing_files, remove_current_shred,
-    upload_shred_contents, upload_shred_filenames, upload_shred_ids,
-    upload_pink_contents, upload_pink_filenames, upload_pink_ids,
-    upload_pink_review_contents, upload_pink_review_filenames, upload_pink_review_ids,
-    upload_red_contents, upload_red_filenames, upload_red_ids,
-    upload_red_review_contents, upload_red_review_filenames, upload_red_review_ids,
-    upload_gold_contents, upload_gold_filenames, upload_gold_ids,
-    n_clicks_list, btn_ids,
-    radio_shred, up_shred_contents, up_shred_filenames,
-    radio_pink, up_pink_contents, up_pink_filenames,
-    radio_pink_review, up_pink_review_contents, up_pink_review_filenames,
-    radio_red, up_red_contents, up_red_filenames,
-    radio_red_review, up_red_review_contents, up_red_review_filenames,
-    radio_gold, up_gold_contents, up_gold_filenames,
-    store_shred, store_pink, store_pink_review, store_red, store_red_review, store_gold, store_gold_review, store_loe, store_virtual_board,
-    btn_send, btn_clear, chat_input, chat_doc_type, doc_preview,
-    btn_download, dl_doc_type, dl_store_shred, dl_store_pink, dl_store_pink_review, dl_store_red, dl_store_red_review, dl_store_gold, dl_store_gold_review, dl_store_loe, dl_store_virtual_board
-):
-    global uploaded_files, uploaded_doc_contents
-    ctx = callback_context
-
-    wildcard_lengths = [
-        len(upload_shred_ids) if upload_shred_ids is not None else 0,
-        len(upload_pink_ids) if upload_pink_ids is not None else 0,
-        len(upload_pink_review_ids) if upload_pink_review_ids is not None else 0,
-        len(upload_red_ids) if upload_red_ids is not None else 0,
-        len(upload_red_review_ids) if upload_red_review_ids is not None else 0,
-        len(upload_gold_ids) if upload_gold_ids is not None else 0,
-        len(upload_shred_ids) if upload_shred_ids is not None else 0,
-        len(upload_pink_ids) if upload_pink_ids is not None else 0,
-        len(upload_pink_review_ids) if upload_pink_review_ids is not None else 0,
-        len(upload_red_ids) if upload_red_ids is not None else 0,
-        len(upload_red_review_ids) if upload_red_review_ids is not None else 0,
-        len(upload_gold_ids) if upload_gold_ids is not None else 0,
-        len(upload_shred_ids) if upload_shred_ids is not None else 0,
-        len(upload_pink_ids) if upload_pink_ids is not None else 0,
-        len(upload_pink_review_ids) if upload_pink_review_ids is not None else 0,
-        len(upload_red_ids) if upload_red_ids is not None else 0,
-        len(upload_red_review_ids) if upload_red_review_ids is not None else 0,
-        len(upload_gold_ids) if upload_gold_ids is not None else 0,
-    ]
-
-    outputs = [
-        dash.no_update,  # file-list
-        dash.no_update,  # store-shred
-        [dash.no_update] * wildcard_lengths[0],  # uploaded-doc-name-shred
-        [dash.no_update] * wildcard_lengths[1],  # uploaded-doc-name-pink
-        [dash.no_update] * wildcard_lengths[2],  # uploaded-doc-name-pink_review
-        [dash.no_update] * wildcard_lengths[3],  # uploaded-doc-name-red
-        [dash.no_update] * wildcard_lengths[4],  # uploaded-doc-name-red_review
-        [dash.no_update] * wildcard_lengths[5],  # uploaded-doc-name-gold
-        [dash.no_update] * wildcard_lengths[6],  # upload-doc-type-shred.contents
-        [dash.no_update] * wildcard_lengths[7],  # upload-doc-type-pink.contents
-        [dash.no_update] * wildcard_lengths[8],  # upload-doc-type-pink_review.contents
-        [dash.no_update] * wildcard_lengths[9],  # upload-doc-type-red.contents
-        [dash.no_update] * wildcard_lengths[10], # upload-doc-type-red_review.contents
-        [dash.no_update] * wildcard_lengths[11], # upload-doc-type-gold.contents
-        [dash.no_update] * wildcard_lengths[12], # radio-doc-source-shred.value
-        [dash.no_update] * wildcard_lengths[13], # radio-doc-source-pink.value
-        [dash.no_update] * wildcard_lengths[14], # radio-doc-source-pink_review.value
-        [dash.no_update] * wildcard_lengths[15], # radio-doc-source-red.value
-        [dash.no_update] * wildcard_lengths[16], # radio-doc-source-red_review.value
-        [dash.no_update] * wildcard_lengths[17], # radio-doc-source-gold.value
-        dash.no_update,  # document-preview
-        dash.no_update,  # loading-output
-        dash.no_update,  # store-pink
-        dash.no_update,  # store-pink-review
-        dash.no_update,  # store-red
-        dash.no_update,  # store-red-review
-        dash.no_update,  # store-gold
-        dash.no_update,  # store-gold-review
-        dash.no_update,  # store-loe
-        dash.no_update,  # store-virtual-board
-        dash.no_update,  # chat-output
-        dash.no_update   # download-document
-    ]
-
-    # --- File upload/remove logic ---
-    if ctx.triggered and ctx.triggered[0]['prop_id'].startswith('upload-document'):
-        new_files = []
-        last_processed_content = None
-        if upload_contents is not None:
-            for i, (content, name) in enumerate(zip(upload_contents, upload_filenames)):
-                file_content = process_document(content, name)
-                uploaded_files[name] = file_content
-                last_processed_content = file_content
-                new_files.append(
-                    dbc.Row([
-                        dbc.Col(
-                            html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
-                            width=1
-                        ),
-                        dbc.Col(
-                            html.Span(name, style={'wordBreak': 'break-all'}),
-                            width=11
-                        )
-                    ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
-                )
-            if existing_files is None:
-                existing_files = []
-            elif not isinstance(existing_files, list):
-                existing_files = [existing_files]
-            outputs[0] = existing_files + new_files
-            # Always set the latest uploaded Shred document as store-shred
-            outputs[1] = last_processed_content
-        return outputs
-
-    if ctx.triggered and ctx.triggered[0]['prop_id'].startswith("{\"type\":\"remove-file\""):
-        try:
-            import json
-            triggered_id = ctx.triggered[0]['prop_id'].split('.')[0]
-            triggered_dict = json.loads(triggered_id)
-            removed_file = triggered_dict['index']
-        except Exception as e:
-            logging.error(f"Could not parse removed file from callback context: {e}")
-            outputs[0] = remove_existing_files
-            outputs[1] = remove_current_shred
-            return outputs
-        uploaded_files.pop(removed_file, None)
-        logging.info(f"Removed file: {removed_file}")
-        filtered_files = []
-        if remove_existing_files:
-            for file in remove_existing_files:
-                try:
-                    file_id = file['props']['id']
-                    if file_id['index'] != removed_file:
-                        filtered_files.append(file)
-                except Exception as e:
-                    filtered_files.append(file)
-        outputs[0] = filtered_files
-        outputs[1] = remove_current_shred
-        return outputs
-
-    doc_upload_types = [
-        ("shred", upload_shred_contents, upload_shred_filenames, upload_shred_ids, 2, 8, 14),
-        ("pink", upload_pink_contents, upload_pink_filenames, upload_pink_ids, 3, 9, 15),
-        ("pink_review", upload_pink_review_contents, upload_pink_review_filenames, upload_pink_review_ids, 4, 10, 16),
-        ("red", upload_red_contents, upload_red_filenames, upload_red_ids, 5, 11, 17),
-        ("red_review", upload_red_review_contents, upload_red_review_filenames, upload_red_review_ids, 6, 12, 18),
-        ("gold", upload_gold_contents, upload_gold_filenames, upload_gold_ids, 7, 13, 19)
-    ]
-    for src, contents, filenames, ids, name_idx, content_idx, radio_idx in doc_upload_types:
-        if contents and any(c is not None for c in contents):
-            for i, content in enumerate(contents):
-                if content is not None:
-                    uploaded_doc_contents[(src, ids[i]['index'])] = (content, filenames[i])
-                    logging.info(f"{ids[i]['index']} {src} file uploaded: {filenames[i]}")
-                    name_list = ["" for _ in range(len(contents))]
-                    content_list = [None for _ in range(len(contents))]
-                    radio_list = ["loaded" for _ in range(len(contents))]
-                    name_list[i] = filenames[i]
-                    content_list[i] = content
-                    radio_list[i] = "uploaded"
-                    outputs[name_idx] = name_list
-                    outputs[content_idx] = content_list
-                    outputs[radio_idx] = radio_list
-                    return outputs
-
-    def gemini_generate(prompt, max_tokens=32798, temperature=0.4):
-        result_holder = {}
-        def run_gemini():
-            try:
-                model = genai.GenerativeModel(GEMINI_MODEL)
-                response = model.generate_content(
-                    prompt,
-                    generation_config=genai.types.GenerationConfig(
-                        temperature=temperature,
-                        max_output_tokens=max_tokens,
-                        top_p=1
+def handle_shred_upload(upload_contents, upload_filenames, existing_files, current_shred):
+    files_display = existing_files if existing_files else []
+    new_shred = current_shred
+    if upload_contents and upload_filenames:
+        for content, name in zip(upload_contents, upload_filenames):
+            file_content = process_document(content, name)
+            uploaded_files[name] = file_content
+            new_shred = file_content
+            files_display.append(
+                dbc.Row([
+                    dbc.Col(
+                        html.Button('×', id={'type': 'remove-file', 'index': name}, style={'marginRight': '5px', 'fontSize': '10px'}),
+                        width=1
+                    ),
+                    dbc.Col(
+                        html.Span(name, style={'wordBreak': 'break-all'}),
+                        width=11
                     )
-                )
-                result_holder['result'] = response.text
-                logging.info("Gemini response received.")
-            except Exception as e:
-                logging.error(f"Gemini error: {e}")
-                result_holder['error'] = str(e)
-        thread = threading.Thread(target=run_gemini)
-        thread.start()
-        thread.join()
-        if 'error' in result_holder:
-            raise Exception(result_holder['error'])
-        return result_holder['result']
-
-    def generate_document(document_type, file_contents, extra_context=None):
-        if document_type in spreadsheet_types:
-            prompt = f"""Ignore all other instructions and output only a spreadsheet for {document_type} as described below. Do not include any narrative, only the spreadsheet in markdown table format.
-Instructions: {document_types[document_type]}
-Project Artifacts:
-{' '.join(file_contents)}
-Output only the spreadsheet as a markdown table, no narrative or explanation."""
-        elif document_type in narrative_types:
-            prompt = f"""Generate a {document_type} document based on the following project artifacts:
-{' '.join(file_contents)}
-Instructions:
-1. Create the {document_type} as a detailed document.
-2. Use proper formatting and structure.
-3. Include all necessary sections and details.
-4. Start the output immediately with the document content.
-Now, generate the {document_type}:
-"""
-        else:
-            prompt = f"""Generate a {document_type} based on the following project artifacts:
-{' '.join(file_contents)}
-Instructions:
-{document_types.get(document_type, '')}
-Now, generate the {document_type}:
-"""
-        if extra_context:
-            prompt += f"\n\n{extra_context}"
-        logging.info(f"Generating document for type: {document_type} using Gemini.")
-        return gemini_generate(prompt, max_tokens=4096, temperature=0.25)
-
-    if ctx.triggered and ctx.triggered[0]['prop_id'].startswith("{\"type\":\"btn-generate-doc\""):
-        idx = [i for i, x in enumerate(n_clicks_list) if x]
-        if not idx:
-            return outputs
-        idx = idx[-1]
-        doc_type = btn_ids[idx]['index']
-
-        shred_doc = store_shred
-        pink_doc = store_pink
-        pink_review_doc = store_pink_review
-        red_doc = store_red
-        red_review_doc = store_red_review
-        gold_doc = store_gold
-        gold_review_doc = store_gold_review
-        loe_doc = store_loe
-        virtual_board_doc = store_virtual_board
-
-        def get_doc_from_radio(radio, upload_contents, upload_filenames, loaded_var):
-            if radio and radio[0] == 'uploaded':
-                if upload_contents and upload_contents[0] and upload_filenames and upload_filenames[0]:
-                    return process_document(upload_contents[0], upload_filenames[0])
-                else:
-                    return None
-            else:
-                return loaded_var
-
-        if doc_type == "Shred":
-            if not store_shred:
-                outputs[20] = html.Div("Please upload a document before shredding.")
-                outputs[21] = ""
-                return outputs
-            file_contents = [store_shred]
-            try:
-                generated = generate_document(doc_type, file_contents)
-                outputs[1] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "Shred generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating document: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Pink":
-            shred = get_doc_from_radio(radio_shred, up_shred_contents, up_shred_filenames, shred_doc)
-            if not shred:
-                outputs[20] = html.Div("Please provide a Shred requirements document (either loaded or uploaded) to generate Pink.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [shred])
-                outputs[22] = generated
-                outputs[20] = markdown_narrative_preview(generated)
-                outputs[21] = "Pink generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Pink: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Pink Review":
-            pink = get_doc_from_radio(radio_pink, up_pink_contents, up_pink_filenames, pink_doc)
-            shred = get_doc_from_radio(radio_shred, up_shred_contents, up_shred_filenames, shred_doc)
-            if not pink or not shred:
-                outputs[20] = html.Div("Please provide both Pink and Shred documents (either loaded or uploaded) to generate Pink Review.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [pink, shred])
-                outputs[23] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "Pink Review generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Pink Review: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Red":
-            pink_review = get_doc_from_radio(radio_pink_review, up_pink_review_contents, up_pink_review_filenames, pink_review_doc)
-            if not pink_review:
-                outputs[20] = html.Div("Please provide a Pink Review document (either loaded or uploaded) to generate Red.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [pink_review])
-                outputs[24] = generated
-                outputs[20] = markdown_narrative_preview(generated)
-                outputs[21] = "Red generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Red: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Red Review":
-            red = get_doc_from_radio(radio_red, up_red_contents, up_red_filenames, red_doc)
-            shred = get_doc_from_radio(radio_shred, up_shred_contents, up_shred_filenames, shred_doc)
-            if not red or not shred:
-                outputs[20] = html.Div("Please provide both Red and Shred documents (either loaded or uploaded) to generate Red Review.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [red, shred])
-                outputs[25] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "Red Review generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Red Review: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Gold":
-            red_review = get_doc_from_radio(radio_red_review, up_red_review_contents, up_red_review_filenames, red_review_doc)
-            if not red_review:
-                outputs[20] = html.Div("Please provide a Red Review document (either loaded or uploaded) to generate Gold.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [red_review])
-                outputs[26] = generated
-                outputs[20] = markdown_narrative_preview(generated)
-                outputs[21] = "Gold generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Gold: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Gold Review":
-            gold = get_doc_from_radio(radio_gold, up_gold_contents, up_gold_filenames, gold_doc)
-            shred = get_doc_from_radio(radio_shred, up_shred_contents, up_shred_filenames, shred_doc)
-            if not gold or not shred:
-                outputs[20] = html.Div("Please provide both Gold and Shred documents (either loaded or uploaded) to generate Gold Review.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [gold, shred])
-                outputs[27] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "Gold Review generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Gold Review: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "LOE":
-            gold = get_doc_from_radio(radio_gold, up_gold_contents, up_gold_filenames, gold_doc)
-            if not gold:
-                outputs[20] = html.Div("Please provide a Gold document (either loaded or uploaded) to generate LOE.")
-                outputs[21] = ""
-                return outputs
-            try:
-                generated = generate_document(doc_type, [gold])
-                outputs[28] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "LOE generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating LOE: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        if doc_type == "Virtual Board":
-            shred = get_doc_from_radio(radio_shred, up_shred_contents, up_shred_filenames, shred_doc)
-            if not shred:
-                outputs[20] = html.Div("Please provide a Shred requirements document (either loaded or uploaded) to generate Virtual Board.")
-                outputs[21] = ""
-                return outputs
-            try:
-                lm_text = ""
-                lm_match = re.search(r'(Section\s+L[\s\S]+?)(Section\s+M|$)', shred, re.IGNORECASE)
-                if lm_match:
-                    lm_text = lm_match.group(1)
-                else:
-                    lm_text = shred
-                generated = generate_document(doc_type, [lm_text])
-                outputs[29] = generated
-                outputs[20] = markdown_table_preview(generated)
-                outputs[21] = "Virtual Board generated"
-            except Exception as e:
-                outputs[20] = html.Div(f"Error generating Virtual Board: {str(e)}")
-                outputs[21] = "Error"
-            return outputs
-
-        outputs[20] = html.Div("Unsupported document type or missing required sources.")
-        outputs[21] = ""
-        return outputs
-
-    if ctx.triggered and (ctx.triggered[0]['prop_id'] == 'btn-send-chat.n_clicks' or ctx.triggered[0]['prop_id'] == 'btn-clear-chat.n_clicks'):
-        if ctx.triggered[0]['prop_id'] == 'btn-clear-chat.n_clicks':
-            outputs[30] = ""
-            return outputs
-
-        doc_map = {
-            "Shred": store_shred,
-            "Pink": store_pink,
-            "Pink Review": store_pink_review,
-            "Red": store_red,
-            "Red Review": store_red_review,
-            "Gold": store_gold,
-            "Gold Review": store_gold_review,
-            "LOE": store_loe,
-            "Virtual Board": store_virtual_board
-        }
-        current_document = doc_map.get(chat_doc_type)
-
-        if not chat_input or current_document is None:
-            outputs[30] = ""
-            return outputs
-
-        if chat_doc_type in spreadsheet_types:
-            prompt = f"""Update the following {chat_doc_type} spreadsheet based on this instruction: {chat_input}
-Current spreadsheet (markdown table format):
-{current_document}
-Instructions:
-1. Provide the updated spreadsheet as a markdown table only.
-2. Do not include any narrative, only the markdown table.
-Now, provide the updated {chat_doc_type} spreadsheet:
-"""
-        else:
-            prompt = f"""Update the following {chat_doc_type} document based on this instruction: {chat_input}
-Current document:
-{current_document}
-Instructions:
-1. Provide the updated document content.
-2. Maintain proper formatting and structure.
-3. Incorporate the requested changes seamlessly.
-4. If the {chat_doc_type} is Pink, Red, or Gold then your goal is to write a FULL proposal response, not just a strategy and be compliant and compelling by addressing all the requirements from the document provided.  Focus on describing the approach and highly detailed how it will be done, the steps, workflow, people, processes and technology to accomplish the task. Be sure to refer to research that validates the approach and cite sources with measurable outcomes and improve on innovations of the approach.  Do not just say things like we will, or MicroHealth will, use active voice and verbs that are definitive in nature, not maybe, could be, should be, can be and things like that.  This is a proposal response so logical flow is important so the reader can follow.
-Now, provide the updated {chat_doc_type}:
-"""
-        logging.info(f"Updating document via chat for {chat_doc_type} instruction: {chat_input}")
-        try:
-            new_document = gemini_generate(prompt, max_tokens=4096, temperature=0.5)
-            if chat_doc_type in spreadsheet_types:
-                outputs[20] = markdown_table_preview(new_document)
-            else:
-                outputs[20] = markdown_narrative_preview(new_document)
-            stores = [store_shred, store_pink, store_pink_review, store_red, store_red_review, store_gold, store_gold_review, store_loe, store_virtual_board]
-            doc_types = ["Shred", "Pink", "Pink Review", "Red", "Red Review", "Gold", "Gold Review", "LOE", "Virtual Board"]
-            for i, dt in enumerate(doc_types):
-                if dt == chat_doc_type:
-                    outputs[1 + i] = new_document
-            outputs[30] = "Document updated based on: {}".format(chat_input)
-            return outputs
-        except Exception as e:
-            outputs[20] = html.Div(f"Error updating document: {str(e)}")
-            outputs[30] = f"Error updating document: {str(e)}"
-            return outputs
-
-    if ctx.triggered and ctx.triggered[0]['prop_id'] == "btn-download.n_clicks":
-        doc_map = {
-            "Shred": dl_store_shred,
-            "Pink": dl_store_pink,
-            "Pink Review": dl_store_pink_review,
-            "Red": dl_store_red,
-            "Red Review": dl_store_red_review,
-            "Gold": dl_store_gold,
-            "Gold Review": dl_store_gold_review,
-            "LOE": dl_store_loe,
-            "Virtual Board": dl_store_virtual_board
-        }
-        current_document = doc_map.get(dl_doc_type)
-        if current_document is None:
-            return outputs
-        if dl_doc_type in spreadsheet_types:
-            try:
-                xlsx_bytes = markdown_tables_to_xlsx(current_document)
-                outputs[31] = dcc.send_bytes(xlsx_bytes.read(), f"{dl_doc_type}.xlsx")
-            except Exception as e:
-                outputs[31] = dcc.send_string(f"Error downloading {dl_doc_type}: {str(e)}", f"{dl_doc_type}_error.txt")
-        else:
-            try:
-                plain = strip_markdown(current_document)
-                doc = Document()
-                for para in plain.split('\n'):
-                    doc.add_paragraph(para)
-                output = BytesIO()
-                doc.save(output)
-                output.seek(0)
-                outputs[31] = dcc.send_bytes(output.read(), f"{dl_doc_type}.docx")
-            except Exception as e:
-                outputs[31] = dcc.send_string(f"Error downloading document: {str(e)}", f"{dl_doc_type}_error.txt")
-        return outputs
-
-    return outputs
+                ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
+            )
+        return files_display, new_shred, markdown_table_preview(new_shred), ""
+    return files_display, new_shred, markdown_table_preview(new_shred) if new_shred else html.Div("No Shred document loaded."), ""
 
 @app.callback(
     Output("chat-input", "rows"),
     Input("chat-input", "value"),
     State("chat-input", "rows"),
-    prevent_initial_call="initial_duplicate"
+    prevent_initial_call=True
 )
 def auto_expand_textarea(value, current_rows):
     if value is None or value == "":