Update app.py via AI Editor

app.py

@@ -449,6 +449,9 @@ 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'),
@@ -577,11 +580,64 @@ def master_callback(
     global uploaded_files, uploaded_doc_contents
     ctx = callback_context
 
-    # Outputs: file-list, store-shred, ...names for uploads, ...upload contents, ...radio values, document-preview, loading-output, store-pink, store-pink-review, store-red, store-red-review, store-gold, store-gold-review, store-loe, store-virtual-board, chat-output, download-document
-    outputs = [dash.no_update] * 32
+    # 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,
+        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 ---
-    # Handle Shred upload
     if ctx.triggered and ctx.triggered[0]['prop_id'].startswith('upload-document'):
         new_files = []
         if upload_contents is not None:
@@ -608,7 +664,6 @@ def master_callback(
             outputs[1] = current_shred
         return outputs
 
-    # Handle file removal
     if ctx.triggered and ctx.triggered[0]['prop_id'].startswith("{\"type\":\"remove-file\""):
         try:
             import json
@@ -635,7 +690,6 @@ def master_callback(
         outputs[1] = remove_current_shred
         return outputs
 
-    # Handle all doc-type uploads for preview and radio
     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),
@@ -650,13 +704,20 @@ 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]}")
-                    outputs[name_idx] = [filenames[i] if j==i else "" for j in range(len(contents))]
-                    outputs[content_idx] = [content if j==i else None for j in range(len(contents))]
-                    outputs[radio_idx] = ["uploaded" if j==i else "loaded" for j in range(len(contents))]
+                    # 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))]
+                    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
 
     # --- Document generation/chat/download logic ---
-    def gemini_generate(prompt, max_tokens=4096, temperature=0.25):
+    def gemini_generate(prompt, max_tokens=32798, temperature=0.4):
         result_holder = {}
         def run_gemini():
             try:
@@ -710,7 +771,6 @@ Now, generate the {document_type}:
         logging.info(f"Generating document for type: {document_type} using Gemini.")
         return gemini_generate(prompt, max_tokens=4096, temperature=0.25)
 
-    # Document generation triggers
     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:
@@ -894,7 +954,6 @@ Now, generate the {document_type}:
         outputs[21] = ""
         return outputs
 
-    # Chat triggers
     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] = ""
@@ -956,7 +1015,6 @@ Now, provide the updated {chat_doc_type}:
             outputs[30] = f"Error updating document: {str(e)}"
             return outputs
 
-    # Download trigger
     if ctx.triggered and ctx.triggered[0]['prop_id'] == "btn-download.n_clicks":
         doc_map = {
             "Shred": dl_store_shred,