Update app.py via AI Editor

app.py

@@ -23,7 +23,8 @@ CLAUDE3_SONNET_MODEL = "claude-3-7-sonnet-20250219"
 CLAUDE3_MAX_CONTEXT_TOKENS = 200_000
 CLAUDE3_MAX_OUTPUT_TOKENS = 64_000
 
-uploaded_documents = {}  # filename: content
+uploaded_documents = {}          # filename: content for RFP/SOW/PWS/RFI
+uploaded_proposals = {}          # filename: content for Proposal
 shredded_document = None
 generated_response = None
 
@@ -60,18 +61,29 @@ def anthropic_stream_generate(prompt):
         logging.error("Error during anthropic streaming request: %s", e)
         return f"Error during streaming: {e}"
 
-def process_document(action, selected_filename=None, chat_input=None):
+def process_document(action, selected_filename=None, chat_input=None, selected_proposal=None):
     global shredded_document, generated_response
     logging.info(f"Process document called with action: {action}")
 
     doc_content = None
-    if selected_filename and selected_filename in uploaded_documents:
-        doc_content = uploaded_documents[selected_filename]
-    elif uploaded_documents:
-        doc_content = next(iter(uploaded_documents.values()))
-        selected_filename = next(iter(uploaded_documents.keys()))
-    else:
-        doc_content = None
+    # Actions "shred" and "generate" use the RFP/SOW/PWS/RFI document dropdown.
+    # If needed, proposal document available in selected_proposal.
+    if action in ["shred", "generate"]:
+        if selected_filename and selected_filename in uploaded_documents:
+            doc_content = uploaded_documents[selected_filename]
+        elif uploaded_documents:
+            doc_content = next(iter(uploaded_documents.values()))
+            selected_filename = next(iter(uploaded_documents.keys()))
+        else:
+            doc_content = None
+    elif action == "proposal":
+        if selected_proposal and selected_proposal in uploaded_proposals:
+            doc_content = uploaded_proposals[selected_proposal]
+        elif uploaded_proposals:
+            doc_content = next(iter(uploaded_proposals.values()))
+            selected_proposal = next(iter(uploaded_proposals.keys()))
+        else:
+            doc_content = None
 
     if action == 'shred':
         if not doc_content:
@@ -131,6 +143,10 @@ def process_document(action, selected_filename=None, chat_input=None):
         t.join()
         return generated_response
 
+    elif action == 'proposal':
+        if not doc_content:
+            return "No proposal document uploaded."
+        return dcc.Markdown(doc_content, style={"whiteSpace": "pre-wrap", "wordWrap": "break-word"})
     elif action == 'compliance':
         return "Compliance checking not implemented yet."
     elif action == 'recover':
@@ -141,15 +157,28 @@ def process_document(action, selected_filename=None, chat_input=None):
         return "LOE estimation not implemented yet."
     return "Action not implemented yet."
 
-def get_uploaded_doc_list():
-    if not uploaded_documents:
+def get_uploaded_doc_list(docdict):
+    if not docdict:
         return html.Div("No documents uploaded.", style={"wordWrap": "break-word"})
     doc_list = []
-    for filename in uploaded_documents:
+    for filename in docdict:
         doc_list.append(
             dbc.ListGroupItem([
                 html.Span(filename, style={"wordWrap": "break-word"}),
-                dbc.Button("Delete", id={'type': 'delete-doc-btn', 'index': filename}, size="sm", color="danger", className="float-end ms-2")
+                dbc.Button("Delete", id={'type': 'delete-doc-btn', 'index': filename, 'group': 'rfp'}, size="sm", color="danger", className="float-end ms-2")
+            ], className="d-flex justify-content-between align-items-center")
+        )
+    return dbc.ListGroup(doc_list, flush=True)
+
+def get_uploaded_proposal_list(docdict):
+    if not docdict:
+        return html.Div("No proposal documents uploaded.", style={"wordWrap": "break-word"})
+    doc_list = []
+    for filename in docdict:
+        doc_list.append(
+            dbc.ListGroupItem([
+                html.Span(filename, style={"wordWrap": "break-word"}),
+                dbc.Button("Delete", id={'type': 'delete-proposal-btn', 'index': filename, 'group': 'proposal'}, size="sm", color="danger", className="float-end ms-2")
             ], className="d-flex justify-content-between align-items-center")
         )
     return dbc.ListGroup(doc_list, flush=True)
@@ -158,38 +187,15 @@ app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
             dbc.Card([
-                dbc.CardHeader(html.H3("Navigation")),
-                dbc.CardBody([
-                    dbc.Button("Shred RFP/PWS/SOW/RFI", id="shred-btn", className="mb-2 w-100 btn-primary"),
-                    dbc.Button("Generate Proposal Response", id="generate-btn", className="mb-2 w-100 btn-secondary"),
-                    dbc.Button("Check Compliance", id="compliance-btn", className="mb-2 w-100 btn-tertiary"),
-                    dbc.Button("Recover Document", id="recover-btn", className="mb-2 w-100 btn-tertiary"),
-                    dbc.Button("Virtual Board", id="board-btn", className="mb-2 w-100 btn-tertiary"),
-                    dbc.Button("Estimate LOE", id="loe-btn", className="mb-2 w-100 btn-tertiary"),
-                ])
-            ], className="mb-3"),
-            dbc.Card([
-                dbc.CardHeader(html.H5("Uploaded Documents")),
+                dbc.CardHeader(html.H3("Documents")),
                 dbc.CardBody([
                     html.Div(id='uploaded-doc-list')
                 ])
-            ])
-        ], width=3, style={'minWidth': '260px'}),
-        dbc.Col([
+            ], className="mb-3"),
+
             dbc.Card([
-                dbc.CardHeader(html.H2("RFP Proposal Assistant", style={'wordWrap': 'break-word'})),
+                dbc.CardHeader(html.H5("RFP/SOW/PWS/RFI")),
                 dbc.CardBody([
-                    dbc.Form([
-                        dbc.Textarea(id="chat-input", placeholder="Enter additional instructions...", style={"width":"100%", "wordWrap": "break-word"}, className="mb-2"),
-                    ]),
-                    html.Div([
-                        dbc.Button("Shred", id="shred-action-btn", className="mr-2 btn-primary"),
-                        dbc.Button("Generate", id="generate-action-btn", className="mr-2 btn-secondary"),
-                        dbc.Button("Check Compliance", id="compliance-action-btn", className="mr-2 btn-tertiary"),
-                        dbc.Button("Recover", id="recover-action-btn", className="mr-2 btn-tertiary"),
-                        dbc.Button("Virtual Board", id="board-action-btn", className="mr-2 btn-tertiary"),
-                        dbc.Button("LOE", id="loe-action-btn", className="btn-tertiary"),
-                    ], className="mt-3 mb-3"),
                     dcc.Dropdown(
                         id='select-document-dropdown',
                         options=[{'label': fn, 'value': fn} for fn in uploaded_documents.keys()],
@@ -215,7 +221,58 @@ app.layout = dbc.Container([
                         },
                         multiple=False
                     ),
-                    html.Div(id='output-document-upload'),
+                    html.Div(id='output-document-upload')
+                ])
+            ], className="mb-3"),
+
+            dbc.Card([
+                dbc.CardHeader(html.H5("Proposal")),
+                dbc.CardBody([
+                    dcc.Dropdown(
+                        id='select-proposal-dropdown',
+                        options=[{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()],
+                        placeholder="Select a proposal document",
+                        value=next(iter(uploaded_proposals), None),
+                        style={"marginBottom": "10px"}
+                    ),
+                    dcc.Upload(
+                        id='upload-proposal',
+                        children=html.Div([
+                            'Drag and Drop or ',
+                            html.A('Select Files')
+                        ]),
+                        style={
+                            'width': '100%',
+                            'height': '60px',
+                            'lineHeight': '60px',
+                            'borderWidth': '1px',
+                            'borderStyle': 'dashed',
+                            'borderRadius': '5px',
+                            'textAlign': 'center',
+                            'margin': '10px'
+                        },
+                        multiple=False
+                    ),
+                    html.Div(id='output-proposal-upload')
+                ])
+            ])
+        ], width=3, style={'minWidth': '260px'}),
+
+        dbc.Col([
+            dbc.Card([
+                dbc.CardHeader(html.H2("RFP Proposal Assistant", style={'wordWrap': 'break-word'})),
+                dbc.CardBody([
+                    dbc.Form([
+                        dbc.Textarea(id="chat-input", placeholder="Enter additional instructions...", style={"width":"100%", "wordWrap": "break-word"}, className="mb-2"),
+                    ]),
+                    html.Div([
+                        dbc.Button("Shred", id="shred-action-btn", className="mr-2 btn-primary"),
+                        dbc.Button("Generate", id="generate-action-btn", className="mr-2 btn-secondary"),
+                        dbc.Button("Check Compliance", id="compliance-action-btn", className="mr-2 btn-tertiary"),
+                        dbc.Button("Recover", id="recover-action-btn", className="mr-2 btn-tertiary"),
+                        dbc.Button("Virtual Board", id="board-action-btn", className="mr-2 btn-tertiary"),
+                        dbc.Button("LOE", id="loe-action-btn", className="btn-tertiary"),
+                    ], className="mt-3 mb-3"),
                     dcc.Loading(
                         id="loading",
                         type="default",
@@ -232,32 +289,59 @@ app.layout = dbc.Container([
     Output('uploaded-doc-list', 'children'),
     Output('select-document-dropdown', 'options'),
     Output('select-document-dropdown', 'value'),
+    Output('output-document-upload', 'children'),
+    Output('select-proposal-dropdown', 'options'),
+    Output('select-proposal-dropdown', 'value'),
+    Output('output-proposal-upload', 'children'),
     Input('upload-document', 'contents'),
     State('upload-document', 'filename'),
-    Input({'type': 'delete-doc-btn', 'index': dash.ALL}, 'n_clicks'),
-    State('uploaded-doc-list', 'children'),
+    Input({'type': 'delete-doc-btn', 'index': dash.ALL, 'group': 'rfp'}, 'n_clicks'),
     State('select-document-dropdown', 'value'),
+    Input('upload-proposal', 'contents'),
+    State('upload-proposal', 'filename'),
+    Input({'type': 'delete-proposal-btn', 'index': dash.ALL, 'group': 'proposal'}, 'n_clicks'),
+    State('select-proposal-dropdown', 'value'),
     prevent_initial_call=True
 )
-def update_uploaded_docs(content, filename, delete_clicks, children, selected_doc):
+def update_uploaded_docs(
+    rfp_content, rfp_filename, rfp_delete_clicks, selected_doc,
+    proposal_content, proposal_filename, proposal_delete_clicks, selected_proposal
+):
     ctx = callback_context
     triggered = ctx.triggered
     changed_id = ""
-    if triggered:
-        changed_id = triggered[0]['prop_id'].split('.')[0]
-    # Handle upload
-    if content is not None and filename:
-        content_type, content_string = content.split(',')
+    output_document_upload = ""
+    output_proposal_upload = ""
+
+    # --- Handle RFP/SOW/PWS/RFI document upload ---
+    if rfp_content is not None and rfp_filename:
+        content_type, content_string = rfp_content.split(',')
         decoded = base64.b64decode(content_string)
         text = decode_document(decoded)
         if text is not None:
-            uploaded_documents[filename] = text
-            logging.info(f"Document uploaded: {filename}")
+            uploaded_documents[rfp_filename] = text
+            output_document_upload = html.Div(f"Document '{rfp_filename}' uploaded successfully.", style={"wordWrap": "break-word"})
+            logging.info(f"Document uploaded: {rfp_filename}")
         else:
-            logging.error(f"Failed to decode uploaded document: {filename}")
-    # Handle delete
-    if delete_clicks:
-        for i, n_click in enumerate(delete_clicks):
+            output_document_upload = html.Div("Error: Could not decode document. Please upload a valid text file.", style={"wordWrap": "break-word"})
+            logging.error(f"Failed to decode uploaded document: {rfp_filename}")
+
+    # --- Handle Proposal document upload ---
+    if proposal_content is not None and proposal_filename:
+        content_type, content_string = proposal_content.split(',')
+        decoded = base64.b64decode(content_string)
+        text = decode_document(decoded)
+        if text is not None:
+            uploaded_proposals[proposal_filename] = text
+            output_proposal_upload = html.Div(f"Proposal '{proposal_filename}' uploaded successfully.", style={"wordWrap": "break-word"})
+            logging.info(f"Proposal uploaded: {proposal_filename}")
+        else:
+            output_proposal_upload = html.Div("Error: Could not decode proposal. Please upload a valid text file.", style={"wordWrap": "break-word"})
+            logging.error(f"Failed to decode uploaded proposal: {proposal_filename}")
+
+    # --- Handle RFP/SOW/PWS/RFI document delete ---
+    if rfp_delete_clicks:
+        for i, n_click in enumerate(rfp_delete_clicks):
             if n_click:
                 btn_id = ctx.inputs_list[2][i]['id']
                 del_filename = btn_id['index']
@@ -267,27 +351,32 @@ def update_uploaded_docs(content, filename, delete_clicks, children, selected_do
                     if selected_doc == del_filename:
                         selected_doc = next(iter(uploaded_documents), None)
                     break
-    options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
-    value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
-    return get_uploaded_doc_list(), options, value
+    # --- Handle Proposal document delete ---
+    if proposal_delete_clicks:
+        for i, n_click in enumerate(proposal_delete_clicks):
+            if n_click:
+                btn_id = ctx.inputs_list[6][i]['id']
+                del_filename = btn_id['index']
+                if del_filename in uploaded_proposals:
+                    del uploaded_proposals[del_filename]
+                    logging.info(f"Proposal deleted: {del_filename}")
+                    if selected_proposal == del_filename:
+                        selected_proposal = next(iter(uploaded_proposals), None)
+                    break
 
-@app.callback(
-    Output('output-document-upload', 'children'),
-    Input('upload-document', 'contents'),
-    State('upload-document', 'filename'),
-    prevent_initial_call=True
-)
-def handle_upload(content, filename):
-    logging.info("Upload callback triggered (output-document-upload).")
-    if content is not None and filename:
-        content_type, content_string = content.split(',')
-        decoded = base64.b64decode(content_string)
-        text = decode_document(decoded)
-        if text is None:
-            return html.Div("Error: Could not decode document. Please upload a valid text file.", style={"wordWrap": "break-word"})
-        else:
-            return html.Div(f"Document '{filename}' uploaded successfully.", style={"wordWrap": "break-word"})
-    return ""
+    doc_options = [{'label': fn, 'value': fn} for fn in uploaded_documents.keys()]
+    doc_value = selected_doc if selected_doc in uploaded_documents else (next(iter(uploaded_documents), None) if uploaded_documents else None)
+    proposal_options = [{'label': fn, 'value': fn} for fn in uploaded_proposals.keys()]
+    proposal_value = selected_proposal if selected_proposal in uploaded_proposals else (next(iter(uploaded_proposals), None) if uploaded_proposals else None)
+    return (
+        get_uploaded_doc_list(uploaded_documents),
+        doc_options,
+        doc_value,
+        output_document_upload,
+        proposal_options,
+        proposal_value,
+        output_proposal_upload
+    )
 
 @app.callback(
     Output('output-data-upload', 'children'),
@@ -301,9 +390,10 @@ def handle_upload(content, filename):
     ],
     State('chat-input', 'value'),
     State('select-document-dropdown', 'value'),
+    State('select-proposal-dropdown', 'value'),
     prevent_initial_call=True
 )
-def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input, selected_filename):
+def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input, selected_filename, selected_proposal):
     ctx = callback_context
     if not ctx.triggered:
         logging.info("No action triggered yet.")