Update app.py via AI Editor

app.py

@@ -617,7 +617,7 @@ def update_preview_window_style(state):
         State('chat-input', 'value'),
         Input('cancel-action-btn', 'n_clicks'),
         State('preview-window-state', 'data'),
-        Input('session-cleanup', 'data')
+        Input('session-cleanup', 'clear_data')
     ],
     prevent_initial_call=True
 )
@@ -625,7 +625,7 @@ def master_callback(
     shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks,
     rfp_content, rfp_filename, doc_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
-    chat_input, cancel_clicks, preview_window_state, session_cleanup_data
+    chat_input, cancel_clicks, preview_window_state, session_cleanup_clear
 ):
     ctx = callback_context
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
@@ -642,8 +642,8 @@ def master_callback(
     proposal_delete_clicks = safe_get_n_clicks(ctx, 12)
     uploaded_rfp_decoded_bytes = None
 
-    # Session cleanup: on session end, clear all data and temp dir
-    if triggered_id == 'session-cleanup':
+    # Session cleanup: only clear all data and temp dir if session-cleanup's clear_data is triggered (i.e., browser close)
+    if triggered_id == 'session-cleanup' and session_cleanup_clear:
         sid = get_session_id()
         cleanup_session_tmpdir(sid)
         if sid in SESSION_DATA:
@@ -801,7 +801,6 @@ def master_callback(
                 output_data_upload = dcc.Markdown(result, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
         finally:
             lock.release()
-        # Update dropdowns and lists after action (in case doc/proposal was auto-created)
         doc_options = [{'label': fn, 'value': fn} for fn in session_dict['uploaded_documents'].keys()]
         doc_value = doc_value if doc_value in session_dict['uploaded_documents'] else (next(iter(session_dict['uploaded_documents']), None) if session_dict['uploaded_documents'] else None)
         proposal_options = [{'label': fn, 'value': fn} for fn in session_dict['proposals'].keys()]