Update app.py via AI Editor

app.py

@@ -299,16 +299,13 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
     doc_type = btn_ids[idx]['index']
     document_type = doc_type
 
-    if not uploaded_files and doc_type == "Shred":
-        logging.info("No uploaded files for Shred. Aborting.")
-        return html.Div("Please upload a document before shredding."), ""
-
-    source = radio_values[idx] if radio_values and len(radio_values) > idx else 'loaded'
-    doc_content = None
-
+    # SHRED CASE: Use uploaded_files directly, no radio or upload needed
     if doc_type == "Shred":
+        if not uploaded_files:
+            logging.info("No uploaded files for Shred. Aborting.")
+            return html.Div("Please upload a document before shredding."), ""
         file_contents = list(uploaded_files.values())
-        logging.info(f"Calling OpenAI for Shred with {len(file_contents)} files.")
+        logging.info(f"Calling OpenAI for Shred with {len(file_contents)} files: {list(uploaded_files.keys())}")
         try:
             generated = generate_document(doc_type, file_contents)
             current_document = generated
@@ -320,10 +317,19 @@ def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, uplo
             logging.error(f"Error generating document: {str(e)}")
             return html.Div(f"Error generating document: {str(e)}"), "Error"
 
+    # For other doc types, handle radio/upload logic
     if shredded_document is None:
         logging.info("Shredded document is None. Aborting.")
         return html.Div("Please shred a document first."), ""
 
+    # Defensive: radio_values, upload_contents, upload_filenames may be None or short
+    source = None
+    if radio_values and len(radio_values) > idx:
+        source = radio_values[idx]
+    else:
+        source = 'loaded'
+
+    doc_content = None
     if source == 'uploaded':
         if upload_contents and len(upload_contents) > idx and upload_contents[idx] and upload_filenames and len(upload_filenames) > idx and upload_filenames[idx]:
             doc_content = process_document(upload_contents[idx], upload_filenames[idx])