Update app.py via AI Editor

app.py

@@ -2,14 +2,22 @@ import os
 import base64
 import io
 import dash
-from dash import dcc, html, Input, Output, State
+from dash import dcc, html, Input, Output, State, callback_context
 import dash_bootstrap_components as dbc
 import pandas as pd
 import anthropic
 from threading import Thread
+import logging
+
+# Logging setup
+logging.basicConfig(
+    level=logging.INFO,
+    format='[%(asctime)s] %(levelname)s - %(message)s'
+)
 
 # Initialize Dash app
 app = dash.Dash(__name__, external_stylesheets=[dbc.themes.BOOTSTRAP])
+server = app.server
 
 # Anthropic API setup
 ANTHROPIC_KEY = os.environ.get("ANTHROPIC_API_KEY", "")
@@ -24,125 +32,209 @@ shredded_document = None
 generated_response = None
 
 # Helper functions
+def decode_document(decoded_bytes):
+    try:
+        # Try UTF-8 first
+        content = decoded_bytes.decode('utf-8')
+        logging.info("Document decoded as UTF-8.")
+        return content
+    except UnicodeDecodeError as e_utf8:
+        try:
+            # Try latin-1 fallback
+            content = decoded_bytes.decode('latin-1')
+            logging.warning("Document decoded as Latin-1 due to utf-8 decode error: %s", e_utf8)
+            return content
+        except Exception as e:
+            logging.error("Document decode failed for both utf-8 and latin-1: %s", e)
+            return None
+
 def process_document(content, filename, action):
     global uploaded_document, shredded_document, generated_response
-    
-    content_type, content_string = content.split(',')
-    decoded = base64.b64decode(content_string)
+    logging.info(f"Process document called with action: {action}")
     
     if action == 'upload':
-        uploaded_document = io.StringIO(decoded.decode('utf-8')).read()
+        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."
     
     elif action == 'shred':
-        prompt = f"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}"
-        response = anthropic_client.messages.create(
-            model=CLAUDE3_SONNET_MODEL,
-            max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
-            messages=[{"role": "user", "content": prompt}]
+        if not uploaded_document:
+            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
         )
-        shredded_document = response.content[0].text
+        def thread_shred():
+            try:
+                response = anthropic_client.messages.create(
+                    model=CLAUDE3_SONNET_MODEL,
+                    max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                nonlocal shredded_document
+                shredded_document = response.content[0].text
+                logging.info("Document shredded successfully.")
+            except Exception as e:
+                shredded_document = f"Error during shredding: {e}"
+                logging.error("Error in thread_shred: %s", e)
+        shredded_document = "Shredding in progress..."
+        t = Thread(target=thread_shred)
+        t.start()
+        t.join()
         return shredded_document
     
     elif action == 'generate':
-        prompt = f"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}"
-        response = anthropic_client.messages.create(
-            model=CLAUDE3_SONNET_MODEL,
-            max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
-            messages=[{"role": "user", "content": prompt}]
+        if not shredded_document:
+            logging.warning("No shredded document found when generating response.")
+            return "Shredded document not available."
+        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
         )
-        generated_response = response.content[0].text
+        def thread_generate():
+            try:
+                response = anthropic_client.messages.create(
+                    model=CLAUDE3_SONNET_MODEL,
+                    max_tokens=CLAUDE3_MAX_OUTPUT_TOKENS,
+                    messages=[{"role": "user", "content": prompt}]
+                )
+                nonlocal generated_response
+                generated_response = response.content[0].text
+                logging.info("Proposal response generated successfully.")
+            except Exception as e:
+                generated_response = f"Error during generation: {e}"
+                logging.error("Error in thread_generate: %s", e)
+        generated_response = "Generating response..."
+        t = Thread(target=thread_generate)
+        t.start()
+        t.join()
         return generated_response
     
-    # Implement other actions (check_compliance, recover_document, virtual_board, estimate_loe) similarly
+    return "Action not implemented yet."
 
 # Layout
 app.layout = dbc.Container([
     dbc.Row([
         dbc.Col([
-            html.H3("Navigation"),
-            dbc.Button("Shred RFP/PWS/SOW/RFI", id="shred-btn", className="mb-2 w-100"),
-            dbc.Button("Generate Proposal Response", id="generate-btn", className="mb-2 w-100"),
-            dbc.Button("Check Compliance", id="compliance-btn", className="mb-2 w-100"),
-            dbc.Button("Recover Document", id="recover-btn", className="mb-2 w-100"),
-            dbc.Button("Virtual Board", id="board-btn", className="mb-2 w-100"),
-            dbc.Button("Estimate LOE", id="loe-btn", className="mb-2 w-100"),
-        ], width=3),
+            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"),
+                ])
+            ])
+        ], width=3, style={'minWidth': '260px'}),
         dbc.Col([
-            dbc.Input(id="chat-input", type="text", placeholder="Enter additional instructions..."),
-            dbc.Button("Send", id="chat-send-btn", className="mt-2"),
-            html.Div([
-                dbc.Button("Shred", id="shred-action-btn", className="mr-2"),
-                dbc.Button("Generate", id="generate-action-btn", className="mr-2"),
-                dbc.Button("Check Compliance", id="compliance-action-btn", className="mr-2"),
-                dbc.Button("Recover", id="recover-action-btn", className="mr-2"),
-                dbc.Button("Virtual Board", id="board-action-btn", className="mr-2"),
-                dbc.Button("LOE", id="loe-action-btn"),
-            ], className="mt-3 mb-3"),
-            dcc.Upload(
-                id='upload-document',
-                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-document-upload'),
-            dcc.Loading(
-                id="loading",
-                type="default",
-                children=html.Div(id="output-data-upload")
-            )
+            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"),
+                        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"),
+                        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.Upload(
+                        id='upload-document',
+                        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-document-upload'),
+                    dcc.Loading(
+                        id="loading",
+                        type="default",
+                        children=html.Div(id="output-data-upload"),
+                        style={"textAlign": "center"}
+                    )
+                ])
+            ], style={'backgroundColor': 'white'})
         ], width=9)
-    ])
+    ], style={'marginTop':'20px'})
 ], fluid=True)
 
-# Callbacks
+# Callback for document upload
 @app.callback(
     Output('output-document-upload', 'children'),
     Input('upload-document', 'contents'),
-    State('upload-document', 'filename')
+    State('upload-document', 'filename'),
+    prevent_initial_call=True
 )
-def update_output(content, filename):
+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)
+        return html.Div(result, style={"wordWrap": "break-word"})
+    return ""
 
+# Unified callback for all actions
 @app.callback(
     Output('output-data-upload', 'children'),
-    [Input('shred-action-btn', 'n_clicks'),
-     Input('generate-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')],
-    State('chat-input', 'value')
+    [
+        Input('shred-action-btn', 'n_clicks'),
+        Input('generate-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')
+    ],
+    State('chat-input', 'value'),
+    prevent_initial_call=True
 )
-def update_output(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input):
-    ctx = dash.callback_context
+def handle_actions(shred_clicks, generate_clicks, compliance_clicks, recover_clicks, board_clicks, loe_clicks, chat_input):
+    ctx = callback_context
     if not ctx.triggered:
+        logging.info("No action triggered yet.")
         return "No action taken yet."
-    else:
-        button_id = ctx.triggered[0]['prop_id'].split('.')[0]
-        if button_id == 'shred-action-btn':
-            return process_document(None, None, 'shred')
-        elif button_id == 'generate-action-btn':
-            return process_document(None, None, 'generate')
-        # Implement other actions similarly
-    
+    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')
+    elif button_id == 'generate-action-btn':
+        return process_document(None, None, 'generate')
+    elif button_id == 'compliance-action-btn':
+        return "Compliance checking not implemented yet."
+    elif button_id == 'recover-action-btn':
+        return "Recovery not implemented yet."
+    elif button_id == 'board-action-btn':
+        return "Virtual board not implemented yet."
+    elif button_id == 'loe-action-btn':
+        return "LOE estimation not implemented yet."
     return "Action not implemented yet."
 
 if __name__ == '__main__':
     print("Starting the Dash application...")
-    app.run(debug=True, host='0.0.0.0', port=7860)
+    app.run(debug=True, host='0.0.0.0', port=7860, threaded=True)
     print("Dash application has finished running.")