Update app.py via AI Editor

app.py

@@ -161,7 +161,7 @@ def gemini_generate_content(prompt, file_id=None, chat_input=None, file_ids=None
             content_list.extend(files)
             content_list.append("\n\n")
         content_list.append(prompt)
-        logging.info(f"Prompt sent to Gemini: {prompt[:500]}...")  # log first 500 chars
+        logging.info(f"Prompt sent to Gemini: {prompt[:500]}...")
         model = genai.GenerativeModel(GEMINI_MODEL)
         response = model.generate_content(
             contents=content_list,
@@ -634,6 +634,28 @@ app.clientside_callback(
     Input('dummy-url', 'pathname')
 )
 
+@app.callback(
+    Output('documents-list', 'children'),
+    Output('select-document-dropdown', 'options'),
+    Output('select-document-dropdown', 'value'),
+    Output('proposals-list', 'children'),
+    Output('select-proposal-dropdown', 'options'),
+    Output('select-proposal-dropdown', 'value'),
+    Output('preview-window-state', 'data'),
+    Input('session-id-store', 'data'),
+    prevent_initial_call=False
+)
+def refresh_session_files(session_id):
+    sid = get_session_id(session_id)
+    sess_data = get_session_data(sid)
+    doc_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["uploaded_documents"].keys()]
+    doc_value = next(iter(sess_data["uploaded_documents"]), None) if sess_data["uploaded_documents"] else None
+    proposal_options = [{'label': truncate_filename(fn), 'value': fn} for fn in sess_data["proposals"].keys()]
+    proposal_value = next(iter(sess_data["proposals"]), None) if sess_data["proposals"] else None
+    documents_list = get_documents_list(sess_data["uploaded_documents"], sess_data["shredded_documents"])
+    proposals_list = get_proposals_list(sess_data["proposals"])
+    return documents_list, doc_options, doc_value, proposals_list, proposal_options, proposal_value, "expanded"
+
 @app.callback(
     Output('output-data-upload', 'children'),
     Output('documents-list', 'children'),