Update app.py via AI Editor

app.py

@@ -401,6 +401,7 @@ def get_proposals_list(proposaldict):
     return dbc.ListGroup(doc_list, flush=True)
 
 app.layout = dbc.Container([
+    dcc.Store(id='preview-window-state', data='expanded'),
     dbc.Row([
         dbc.Col([
             dbc.Card([
@@ -486,7 +487,10 @@ app.layout = dbc.Container([
                     dcc.Loading(
                         id="loading",
                         type="default",
-                        children=html.Div(id="output-data-upload"),
+                        children=html.Div(
+                            id="output-preview-container",
+                            children=html.Div(id="output-data-upload"),
+                        ),
                         style={"textAlign": "center"}
                     )
                 ])
@@ -495,6 +499,48 @@ app.layout = dbc.Container([
     ], style={'marginTop':'20px'})
 ], fluid=True)
 
+@app.callback(
+    Output('preview-window-state', 'data'),
+    [
+        Input('shred-action-btn', 'n_clicks'),
+        Input('proposal-action-btn', 'n_clicks'),
+        Input('compliance-action-btn', 'n_clicks'),
+        Input('recover-action-btn', 'n_clicks'),
+        Input('board-action-btn', 'n_clicks'),
+        Input('loe-action-btn', 'n_clicks')
+    ],
+    prevent_initial_call=True
+)
+def shrink_preview_window_on_action(
+    shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks
+):
+    ctx = callback_context
+    if ctx.triggered:
+        return "shrunk"
+    return dash.no_update
+
+@app.callback(
+    Output('output-preview-container', 'style'),
+    [Input('preview-window-state', 'data')]
+)
+def update_preview_window_style(state):
+    if state == "shrunk":
+        return {
+            "maxHeight": "8em",
+            "overflow": "hidden",
+            "transition": "max-height 0.4s",
+            "border": "1px solid #ddd",
+            "borderRadius": "4px"
+        }
+    else:
+        return {
+            "maxHeight": "none",
+            "overflow": "visible",
+            "transition": "max-height 0.4s",
+            "border": "1px solid #ddd",
+            "borderRadius": "4px"
+        }
+
 @app.callback(
     Output('output-data-upload', 'children'),
     Output('documents-list', 'children'),
@@ -503,6 +549,7 @@ app.layout = dbc.Container([
     Output('proposals-list', 'children'),
     Output('select-proposal-dropdown', 'options'),
     Output('select-proposal-dropdown', 'value'),
+    Output('preview-window-state', 'data'),
     [
         Input('shred-action-btn', 'n_clicks'),
         Input('proposal-action-btn', 'n_clicks'),
@@ -518,7 +565,8 @@ app.layout = dbc.Container([
         Input({'type': 'delete-proposal-btn', 'index': ALL, 'group': 'proposal'}, 'n_clicks'),
         State('select-proposal-dropdown', 'value'),
         State('chat-input', 'value'),
-        Input('cancel-action-btn', 'n_clicks')
+        Input('cancel-action-btn', 'n_clicks'),
+        State('preview-window-state', 'data')
     ],
     prevent_initial_call=True
 )
@@ -526,7 +574,7 @@ def master_callback(
     shred_clicks, proposal_clicks, compliance_clicks, recover_clicks, board_clicks,
     rfp_content, rfp_filename, doc_delete_clicks, selected_doc,
     proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal,
-    chat_input, cancel_clicks
+    chat_input, cancel_clicks, preview_window_state
 ):
     ctx = callback_context
     triggered_id = ctx.triggered[0]['prop_id'].split('.')[0] if ctx.triggered else None
@@ -543,6 +591,13 @@ def master_callback(
 
     global gemini_lock, uploaded_documents_bytes
 
+    # If an action button is clicked, immediately shrink window, do nothing else (let other callback run)
+    if triggered_id in [
+        'shred-action-btn', 'proposal-action-btn', 'compliance-action-btn',
+        'recover-action-btn', 'board-action-btn', 'loe-action-btn'
+    ] and preview_window_state != "shrunk":
+        return dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, dash.no_update, "shrunk"
+
     if triggered_id == 'cancel-action-btn':
         output_data_upload = html.Div("[Cancelled by user]\n", style={"wordWrap": "break-word"})
         doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
@@ -554,7 +609,8 @@ def master_callback(
         return (
             output_data_upload,
             documents_list, doc_options, doc_value,
-            proposals_list, proposal_options, proposal_value
+            proposals_list, proposal_options, proposal_value,
+            "expanded"
         )
 
     if triggered_id == 'upload-document' and rfp_content is not None and rfp_filename:
@@ -646,7 +702,8 @@ def master_callback(
             return (
                 output_data_upload,
                 documents_list, doc_options, doc_value,
-                proposals_list, proposal_options, proposal_value
+                proposals_list, proposal_options, proposal_value,
+                "expanded"
             )
         try:
             if triggered_id == "shred-action-btn":
@@ -676,7 +733,8 @@ def master_callback(
         return (
             output_data_upload,
             documents_list, doc_options, doc_value,
-            proposals_list, proposal_options, proposal_value
+            proposals_list, proposal_options, proposal_value,
+            "expanded"
         )
 
     if triggered_id in ['proposal-action-btn', 'recover-action-btn', 'loe-action-btn']:
@@ -686,7 +744,8 @@ def master_callback(
             return (
                 output_data_upload,
                 documents_list, doc_options, doc_value,
-                proposals_list, proposal_options, proposal_value
+                proposals_list, proposal_options, proposal_value,
+                "expanded"
             )
         try:
             if triggered_id == "proposal-action-btn":
@@ -720,7 +779,8 @@ def master_callback(
         return (
             output_data_upload,
             documents_list, doc_options, doc_value,
-            proposals_list, proposal_options, proposal_value
+            proposals_list, proposal_options, proposal_value,
+            "expanded"
         )
 
     doc_value = doc_value if doc_value in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
@@ -728,7 +788,8 @@ def master_callback(
     return (
         output_data_upload,
         documents_list, doc_options, doc_value,
-        proposals_list, proposal_options, proposal_value
+        proposals_list, proposal_options, proposal_value,
+        "expanded"
     )
 
 if __name__ == '__main__':