Update app.py via AI Editor

app.py

@@ -545,6 +545,8 @@ def update_right_col(selected_type):
 )
 def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, selected_type,
                     filenames_lists, shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board, prev_generated):
+    logging.info(f"unified_handler triggered. Trigger: {callback_context.triggered}")
+
     stores = [shred, pink, pink_review, red, red_review, gold, gold_review, loe, virtual_board]
     file_list = []
     uploaded_doc_name_shred = dash.no_update
@@ -568,6 +570,8 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
     }
 
     loader_trigger = time.time()
+
+    # Handle file upload
     if "upload-doc-type" in trigger:
         for i, (contents, filenames) in enumerate(zip(contents_lists, filenames_lists)):
             if contents and filenames:
@@ -580,6 +584,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                 content = contents[-1]
                 name = filenames[-1]
                 file_text = process_document(content, name)
+                logging.info(f"Processed uploaded file for subtype '{subtype}': {name}")
                 set_uploaded_doc(subtype, file_text, name)
                 stores[idx] = file_text
                 if subtype == "shred":
@@ -596,6 +601,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                         ], id={'type': 'file-row', 'index': name}, align="center", className="mb-1")
                     ]
                     uploaded_doc_name_shred = name
+                # Preview logic with fallback
                 if subtype == "shred":
                     preview = markdown_table_preview(file_text)
                 elif subtype in ["pink", "red", "gold"]:
@@ -604,8 +610,10 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                     preview = markdown_table_preview(file_text)
                 generated_doc = dash.no_update
                 download_data = dash.no_update
+                logging.info("Returning after file upload.")
                 return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, download_data, loader_trigger]
 
+    # Handle document generation
     if any(n_clicks_generate_list):
         logging.info(f"Generating document for type: {selected_type}")
         inputs = {
@@ -628,6 +636,7 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             context += f"\n\n{inputs.get('shred','')}"
         full_prompt = f"{prompt}\n\n{context}"
         result_holder = {}
+
         def gemini_thread():
             try:
                 model = genai.GenerativeModel(GEMINI_MODEL)
@@ -641,13 +650,18 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
         t.start()
         t.join(timeout=60)
         generated_doc = result_holder.get('result', 'Error: No document generated.')
+        logging.info(f"Gemini API returned: {generated_doc[:500]}")  # LOGGING LINE
+
         set_generated_doc(selected_type, generated_doc)
+        # Preview with fallback
         if selected_type in spreadsheet_types:
             preview = markdown_table_preview(generated_doc)
         else:
             preview = markdown_narrative_preview(generated_doc)
+        logging.info("Returning after document generation.")
         return stores + [file_list, uploaded_doc_name_shred, preview, generated_doc, dash.no_update, loader_trigger]
 
+    # Handle download
     if "btn-download" in trigger:
         doc = prev_generated
         if not doc:
@@ -673,7 +687,8 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
                 doc = stores[8]
         if not doc:
             logging.warning("No generated document to download.")
-            return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, dash.no_update, loader_trigger]
+            preview = html.Div("No document loaded or generated.")
+            return stores + [file_list, uploaded_doc_name_shred, preview, dash.no_update, dash.no_update, loader_trigger]
         if selected_type in spreadsheet_types:
             xlsx_io = markdown_tables_to_xlsx(doc)
             logging.info("Spreadsheet document prepared for download.")
@@ -685,6 +700,13 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             download_data = dcc.send_bytes(content, filename=f"{selected_type}_output.md")
             return stores + [file_list, uploaded_doc_name_shred, dash.no_update, dash.no_update, download_data, loader_trigger]
 
+    # Fallback: test preview with hardcoded table if Shred and no doc
+    if selected_type == "Shred" and not stores[0]:
+        logging.info("Testing preview with hardcoded markdown table (Shred, no doc).")
+        preview = markdown_table_preview("| Col1 | Col2 |\n|---|---|\n| a | b |")
+        return stores + [file_list, uploaded_doc_name_shred, preview, dash.no_update, dash.no_update, loader_trigger]
+
+    # Default preview logic
     doc = prev_generated
     if not doc:
         doc = get_generated_doc(selected_type)
@@ -715,7 +737,9 @@ def unified_handler(contents_lists, n_clicks_generate_list, n_clicks_download, s
             doc = doc_store.get(selected_type, "")
             preview = markdown_narrative_preview(doc)
         else:
-            preview = html.Div("No document loaded.")
+            logging.warning("No document content found for preview.")
+            preview = html.Div("No document loaded or generated.")
+
     return stores + [file_list, uploaded_doc_name_shred, preview, doc, dash.no_update, loader_trigger]
 
 @app.callback(