Update app.py via AI Editor

app.py

@@ -295,8 +295,8 @@ def get_right_col_content(selected_type,
                 loaded_preview = markdown_narrative_preview(red_doc) if red_doc else html.Div("No Red document loaded.")
             elif src == "red_review":
                 label = "Red Review"
-                store_var = red_review_doc
-                loaded_preview = markdown_table_preview(red_review_doc) if red_review_doc else html.Div("No Red Review document loaded.")
+                store_var = red_doc
+                loaded_preview = markdown_narrative_preview(red_doc) if red_doc else html.Div("No Red Review document loaded.")
             elif src == "gold":
                 label = "Gold Document"
                 store_var = gold_doc
@@ -580,7 +580,6 @@ def master_callback(
     global uploaded_files, uploaded_doc_contents
     ctx = callback_context
 
-    # Get lengths for all wildcard outputs for correct no_update list length
     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,
@@ -640,10 +639,12 @@ def master_callback(
     # --- 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(
@@ -661,7 +662,8 @@ def master_callback(
             elif not isinstance(existing_files, list):
                 existing_files = [existing_files]
             outputs[0] = existing_files + new_files
-            outputs[1] = current_shred
+            # 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\""):
@@ -704,7 +706,6 @@ def master_callback(
                 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]}")
-                    # For wildcard outputs, always return a list with the correct length
                     name_list = ["" for _ in range(len(contents))]
                     content_list = [None for _ in range(len(contents))]
                     radio_list = ["loaded" for _ in range(len(contents))]
@@ -716,7 +717,6 @@ def master_callback(
                     outputs[radio_idx] = radio_list
                     return outputs
 
-    # --- Document generation/chat/download logic ---
     def gemini_generate(prompt, max_tokens=32798, temperature=0.4):
         result_holder = {}
         def run_gemini():
@@ -798,11 +798,11 @@ Now, generate the {document_type}:
                 return loaded_var
 
         if doc_type == "Shred":
-            if not uploaded_files:
+            if not store_shred:
                 outputs[20] = html.Div("Please upload a document before shredding.")
                 outputs[21] = ""
                 return outputs
-            file_contents = list(uploaded_files.values())
+            file_contents = [store_shred]
             try:
                 generated = generate_document(doc_type, file_contents)
                 outputs[1] = generated