Update app.py via AI Editor

app.py

@@ -260,11 +260,7 @@ def get_right_col_content(selected_type, shred_doc, pink_doc, pink_review_doc, r
                     },
                     multiple=True
                 ),
-                file_list_component(),
-                html.Div([
-                    html.Div("Loaded Shred Document:", className="mt-2 mb-1"),
-                    markdown_table_preview(shred_doc)
-                ], style={'marginBottom': '12px'})
+                file_list_component()
             ])
         )
     elif selected_type in doc_dependencies:
@@ -481,32 +477,35 @@ def combined_callback(
     loading_msg = ""
     generated_doc = dash.no_update
 
-    if trigger == "upload-document":
-        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
-                        )
-                    ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
-                )
-            logging.info("Shred document uploaded and stored.")
-            preview = markdown_table_preview(new_shred)
-            generated_doc = dash.no_update
-        else:
-            preview = markdown_table_preview(new_shred)
-            generated_doc = dash.no_update
+    # Only handle upload if upload-document is present in layout
+    if trigger == "upload-document" and upload_contents is not None and upload_filenames is not None:
+        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
+                    )
+                ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
+            )
+        logging.info("Shred document uploaded and stored.")
+        preview = markdown_table_preview(new_shred)
+        generated_doc = dash.no_update
+        return files_display, new_shred, preview, loading_msg, generated_doc
+
+    # Always show preview for current shred if nothing else triggered, for Shred only
+    if selected_type == "Shred" and not any(n_clicks):
+        preview = markdown_table_preview(new_shred)
         return files_display, new_shred, preview, loading_msg, generated_doc
 
+    # Handle generate document
     if any(n_clicks):
         logging.info(f"Generating document for type: {selected_type}")
         inputs = {
@@ -542,15 +541,27 @@ def combined_callback(
         t.start()
         t.join(timeout=60)
         generated_doc = result_holder.get('result', 'Error: No document generated.')
+        # Always update preview for all doc types
         if selected_type in spreadsheet_types:
             preview = markdown_table_preview(generated_doc)
         else:
             preview = markdown_narrative_preview(generated_doc)
         logging.info("Document preview updated.")
-        # Always update document-preview, even if file-list/store-shred do not change
         return files_display, new_shred, preview, "", generated_doc
 
-    # Always show preview for current shred if nothing else triggered
+    # For all other doc types, show preview of generated doc if present
+    if selected_type != "Shred":
+        if generated_doc not in (None, "", dash.no_update):
+            if selected_type in spreadsheet_types:
+                preview = markdown_table_preview(generated_doc)
+            else:
+                preview = markdown_narrative_preview(generated_doc)
+            return files_display, new_shred, preview, loading_msg, generated_doc
+
+    # Default: clear preview for non-shred if nothing else
+    if selected_type != "Shred":
+        return files_display, new_shred, html.Div(), loading_msg, generated_doc
+
     return files_display, new_shred, markdown_table_preview(new_shred), loading_msg, generated_doc
 
 @app.callback(