Update app.py via AI Editor

app.py

@@ -30,6 +30,7 @@ MAX_OUTPUT_TOKENS = 65536
 
 uploaded_documents = {}
 uploaded_documents_fileid = {}
+uploaded_documents_bytes = {}
 proposals = {}
 proposals_fileid = {}
 shredded_documents = {}
@@ -149,7 +150,7 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
 
     doc_content = None
     doc_fileid = None
-    if action in ["shred", "compliance", "virtual_board"]:
+    if action in ["shred", "compliance", "virtual_board", "proposal"]:
         if selected_filename and selected_filename in uploaded_documents:
             doc_content = uploaded_documents[selected_filename]
             doc_fileid = uploaded_documents_fileid.get(selected_filename)
@@ -223,9 +224,10 @@ def process_document(action, selected_filename=None, chat_input=None, rfp_decode
             return "No RFP/SOW/PWS/RFI document selected.", None, None, None, None
         rfp_filename = selected_filename
         rfp_fileid = uploaded_documents_fileid.get(selected_filename)
-        if not rfp_fileid and rfp_decoded_bytes is not None:
+        # Try to upload to Gemini if fileid missing and we have bytes
+        if not rfp_fileid and rfp_filename in uploaded_documents_bytes:
             try:
-                fileid = upload_to_gemini_file(rfp_decoded_bytes, rfp_filename)
+                fileid = upload_to_gemini_file(uploaded_documents_bytes[rfp_filename], rfp_filename)
                 if fileid:
                     uploaded_documents_fileid[rfp_filename] = fileid
                     rfp_fileid = fileid
@@ -458,7 +460,7 @@ def master_callback(
     proposal_delete_clicks = safe_get_n_clicks(ctx, 9)
     uploaded_rfp_decoded_bytes = None
 
-    global gemini_lock
+    global gemini_lock, uploaded_documents_bytes
 
     if triggered_id == 'cancel-action-btn':
         output_data_upload = html.Div("[Cancelled by user]\n", style={"wordWrap": "break-word"})
@@ -484,6 +486,7 @@ def master_callback(
             fileid = upload_to_gemini_file(decoded, rfp_filename)
         if text is not None:
             uploaded_documents[rfp_filename] = text
+            uploaded_documents_bytes[rfp_filename] = decoded
             if fileid:
                 uploaded_documents_fileid[rfp_filename] = fileid
             logging.info(f"Document uploaded: {rfp_filename}")
@@ -518,6 +521,8 @@ def master_callback(
                         except Exception as e:
                             logging.warning(f"Failed to delete Gemini file {del_filename}: {e}")
                         del uploaded_documents_fileid[del_filename]
+                    if del_filename in uploaded_documents_bytes:
+                        del uploaded_documents_bytes[del_filename]
                     logging.info(f"Document deleted: {del_filename}")
                 if del_filename in shredded_documents:
                     del shredded_documents[del_filename]
@@ -609,9 +614,11 @@ def master_callback(
                 action_name = "loe"
             else:
                 action_name = None
+            # For proposal, always use the selected doc_value and its bytes if available
+            selected_bytes = uploaded_documents_bytes.get(doc_value, None)
             if action_name:
                 result, _, _, generated_filename, generated_docx_bytes = process_document(
-                    action_name, doc_value, chat_input, uploaded_rfp_decoded_bytes, None
+                    action_name, doc_value, chat_input, selected_bytes, None
                 )
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
         finally: