Update app.py via AI Editor

app.py

@@ -276,7 +276,6 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
     if not uploaded_files and doc_type == "Shred":
         return html.Div("Please upload a document before shredding."), ""
 
-    # Source selection
     source = radio_values[idx] if radio_values and len(radio_values) > idx else 'loaded'
     doc_content = None
 
@@ -286,13 +285,12 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             generated = generate_document(doc_type, file_contents)
             current_document = generated
             shredded_document = generated
-            preview = spreadsheet_preview(generated)
+            preview = spreadsheet_preview(generated, doc_type)
             return preview, "Shred generated"
         except Exception as e:
             logging.error(f"Error generating document: {str(e)}")
             return html.Div(f"Error generating document: {str(e)}"), "Error"
 
-    # For all other types
     if shredded_document is None:
         return html.Div("Please shred a document first."), ""
 
@@ -326,12 +324,12 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             generated = generate_document(doc_type, [doc_content, shredded_document])
             pink_review_document = generated
             current_document = generated
-            preview = spreadsheet_preview(generated)
+            preview = spreadsheet_preview(generated, doc_type)
             return preview, f"{doc_type} generated"
         elif doc_type in ["Red Review", "Gold Review", "Virtual Board", "LOE"]:
             generated = generate_document(doc_type, [doc_content, shredded_document])
             current_document = generated
-            preview = spreadsheet_preview(generated)
+            preview = spreadsheet_preview(generated, doc_type)
             return preview, f"{doc_type} generated"
         elif doc_type in ["Pink", "Red", "Gold"]:
             generated = generate_document(doc_type, [doc_content])
@@ -362,13 +360,22 @@ def update_uploaded_doc_name(contents, filename, id_dict):
         return filename, contents, "uploaded"
     return "", None, "loaded"
 
-def spreadsheet_preview(text):
-    try:
-        # Try to parse as CSV, fallback to plain text
-        df = pd.read_csv(StringIO(text))
-        return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
-    except Exception:
-        return html.Pre(text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
+def spreadsheet_preview(text, doc_type=None):
+    if doc_type is None:
+        doc_type = ""
+    if doc_type in spreadsheet_types:
+        try:
+            df = pd.read_csv(StringIO(text))
+            return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+        except Exception as e:
+            logging.error(f"Error parsing CSV for preview: {str(e)}")
+            return html.Pre("Unable to display spreadsheet preview. Raw output:\n" + text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
+    else:
+        try:
+            df = pd.read_csv(StringIO(text))
+            return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+        except Exception:
+            return html.Pre(text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
 
 def generate_document(document_type, file_contents):
     if document_type in spreadsheet_types:
@@ -460,7 +467,7 @@ Now, provide the updated {selected_doc_type}:
         current_document = response['choices'][0]['message']['content']
         logging.info("Document updated via chat successfully.")
         if selected_doc_type in spreadsheet_types:
-            preview = spreadsheet_preview(current_document)
+            preview = spreadsheet_preview(current_document, selected_doc_type)
         else:
             preview = dcc.Markdown(current_document)
         return f"Document updated based on: {chat_input}", preview