Update app.py via AI Editor

app.py

@@ -23,7 +23,7 @@ CLAUDE3_SONNET_MODEL = "claude-3-7-sonnet-20250219"
 CLAUDE3_MAX_CONTEXT_TOKENS = 200_000
 CLAUDE3_MAX_OUTPUT_TOKENS = 64_000
 
-uploaded_document = None
+uploaded_documents = {}  # filename: content
 shredded_document = None
 generated_response = None
 
@@ -60,31 +60,33 @@ def anthropic_stream_generate(prompt):
         logging.error("Error during anthropic streaming request: %s", e)
         return f"Error during streaming: {e}"
 
-def process_document(content, filename, action):
-    global uploaded_document, shredded_document, generated_response
+def process_document(action, selected_filename=None, chat_input=None):
+    global shredded_document, generated_response
     logging.info(f"Process document called with action: {action}")
 
-    if action == 'upload':
-        if content is None:
-            return "No content uploaded."
-        content_type, content_string = content.split(',')
-        decoded = base64.b64decode(content_string)
-        text = decode_document(decoded)
-        if text is None:
-            return "Error: Could not decode document. Please upload a valid text file."
-        uploaded_document = text
-        logging.info("Document uploaded successfully.")
-        return "Document uploaded successfully."
+    # Select document content to use, if needed
+    doc_content = None
+    if selected_filename and selected_filename in uploaded_documents:
+        doc_content = uploaded_documents[selected_filename]
+    elif uploaded_documents:
+        # Default to the first document if not specified
+        doc_content = next(iter(uploaded_documents.values()))
+        selected_filename = next(iter(uploaded_documents.keys()))
+    else:
+        doc_content = None
 
-    elif action == 'shred':
-        if not uploaded_document:
+    if action == 'shred':
+        if not doc_content:
             logging.warning("No uploaded document found for shredding.")
             return "No document uploaded."
         prompt = (
             "Analyze the following RFP/PWS/SOW/RFI and generate a requirements spreadsheet. "
             "Identify requirements by action words like 'shall', 'will', 'perform', etc. Organize by PWS section and requirement. "
-            "Do not write as if responding to the proposal:\n\n" + uploaded_document
+            "Do not write as if responding to the proposal.\n"
         )
+        if chat_input:
+            prompt += f"User additional instructions: {chat_input}\n"
+        prompt += f"\nFile Name: {selected_filename}\n\n{doc_content}"
         def thread_shred():
             global shredded_document
             shredded_document = ""
@@ -109,8 +111,11 @@ def process_document(content, filename, action):
         prompt = (
             "Create a highly detailed proposal response based on the following PWS requirements. "
             "Be compliant and compelling. Focus on describing the approach, steps, workflow, people, processes, and technology. "
-            "Refer to research that validates the approach and cite sources with measurable outcomes:\n\n" + shredded_document
+            "Refer to research that validates the approach and cite sources with measurable outcomes.\n"
         )
+        if chat_input:
+            prompt += f"User additional instructions: {chat_input}\n"
+        prompt += f"\nFile Name: {selected_filename}\n\n{shredded_document}"
         def thread_generate():
             global generated_response
             generated_response = ""
@@ -128,8 +133,29 @@ def process_document(content, filename, action):
         t.join()
         return generated_response
 
+    elif action == 'compliance':
+        return "Compliance checking not implemented yet."
+    elif action == 'recover':
+        return "Recovery not implemented yet."
+    elif action == 'board':
+        return "Virtual board not implemented yet."
+    elif action == 'loe':
+        return "LOE estimation not implemented yet."
     return "Action not implemented yet."
 
+def get_uploaded_doc_list():
+    if not uploaded_documents:
+        return html.Div("No documents uploaded.", style={"wordWrap": "break-word"})
+    doc_list = []
+    for filename in uploaded_documents:
+        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")
+            ], className="d-flex justify-content-between align-items-center")
+        )
+    return dbc.ListGroup(doc_list, flush=True)
+
 app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
@@ -143,6 +169,12 @@ app.layout = dbc.Container([
                     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.CardBody([
+                    html.Div(id='uploaded-doc-list')
+                ])
             ])
         ], width=3, style={'minWidth': '260px'}),
         dbc.Col([
@@ -151,7 +183,6 @@ app.layout = dbc.Container([
                 dbc.CardBody([
                     dbc.Form([
                         dbc.Textarea(id="chat-input", placeholder="Enter additional instructions...", style={"width":"100%", "wordWrap": "break-word"}, className="mb-2"),
-                        dbc.Button("Send", id="chat-send-btn", className="mt-2 btn-primary")
                     ]),
                     html.Div([
                         dbc.Button("Shred", id="shred-action-btn", className="mr-2 btn-primary"),
@@ -161,6 +192,13 @@ app.layout = dbc.Container([
                         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()],
+                        placeholder="Select a document to work with",
+                        value=next(iter(uploaded_documents), None),
+                        style={"marginBottom": "10px"}
+                    ),
                     dcc.Upload(
                         id='upload-document',
                         children=html.Div([
@@ -192,6 +230,51 @@ app.layout = dbc.Container([
     ], style={'marginTop':'20px'})
 ], fluid=True)
 
+@app.callback(
+    Output('uploaded-doc-list', 'children'),
+    Output('select-document-dropdown', 'options'),
+    Output('select-document-dropdown', 'value'),
+    Input('upload-document', 'contents'),
+    State('upload-document', 'filename'),
+    Input({'type': 'delete-doc-btn', 'index': dash.ALL}, 'n_clicks'),
+    State('uploaded-doc-list', 'children'),
+    State('select-document-dropdown', 'value'),
+    prevent_initial_call=True
+)
+def update_uploaded_docs(content, filename, delete_clicks, children, selected_doc):
+    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(',')
+        decoded = base64.b64decode(content_string)
+        text = decode_document(decoded)
+        if text is not None:
+            uploaded_documents[filename] = text
+            logging.info(f"Document uploaded: {filename}")
+        else:
+            logging.error(f"Failed to decode uploaded document: {filename}")
+    # Handle delete
+    if delete_clicks:
+        for i, n_click in enumerate(delete_clicks):
+            if n_click:
+                # The delete button's id is a dict with 'index' = filename
+                btn_id = ctx.inputs_list[2][i]['id']
+                del_filename = btn_id['index']
+                if del_filename in uploaded_documents:
+                    del uploaded_documents[del_filename]
+                    logging.info(f"Document deleted: {del_filename}")
+                    # If the selected doc is deleted, choose another
+                    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
+
 @app.callback(
     Output('output-document-upload', 'children'),
     Input('upload-document', 'contents'),
@@ -199,10 +282,16 @@ app.layout = dbc.Container([
     prevent_initial_call=True
 )
 def handle_upload(content, filename):
-    logging.info("Upload callback triggered.")
-    if content is not None:
-        result = process_document(content, filename, 'upload')
-        return html.Div(result, style={"wordWrap": "break-word"})
+    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:
+            # Already handled and added in update_uploaded_docs
+            return html.Div(f"Document '{filename}' uploaded successfully.", style={"wordWrap": "break-word"})
     return ""
 
 @app.callback(
@@ -213,12 +302,13 @@ def handle_upload(content, filename):
         Input('compliance-action-btn', 'n_clicks'),
         Input('recover-action-btn', 'n_clicks'),
         Input('board-action-btn', 'n_clicks'),
-        Input('loe-action-btn', 'n_clicks')
+        Input('loe-action-btn', 'n_clicks'),
     ],
     State('chat-input', 'value'),
+    State('select-document-dropdown', 'value'),
     prevent_initial_call=True
 )
-def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input):
+def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input, selected_filename):
     ctx = callback_context
     if not ctx.triggered:
         logging.info("No action triggered yet.")
@@ -226,17 +316,17 @@ def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_cli
     button_id = ctx.triggered[0]['prop_id'].split('.')[0]
     logging.info(f"Button pressed: {button_id}")
     if button_id == 'shred-action-btn':
-        return process_document(None, None, 'shred')
+        return process_document('shred', selected_filename, chat_input)
     elif button_id == 'generate-action-btn':
-        return process_document(None, None, 'generate')
+        return process_document('generate', selected_filename, chat_input)
     elif button_id == 'compliance-action-btn':
-        return "Compliance checking not implemented yet."
+        return process_document('compliance', selected_filename, chat_input)
     elif button_id == 'recover-action-btn':
-        return "Recovery not implemented yet."
+        return process_document('recover', selected_filename, chat_input)
     elif button_id == 'board-action-btn':
-        return "Virtual board not implemented yet."
+        return process_document('board', selected_filename, chat_input)
     elif button_id == 'loe-action-btn':
-        return "LOE estimation not implemented yet."
+        return process_document('loe', selected_filename, chat_input)
     return "Action not implemented yet."
 
 if __name__ == '__main__':