Update app.py via AI Editor

app.py

@@ -27,6 +27,9 @@ shredded_document = None
 pink_review_document = None
 uploaded_doc_contents = {}
 
+spreadsheet_types = ["Shred", "Pink Review", "Red Review", "Gold Review", "Virtual Board", "LOE"]
+narrative_types = ["Pink", "Red", "Gold"]
+
 document_types = {
     "Shred": "Ignore all other instructions and generate only requirements spreadsheet of the Project Work Statement (PWS) identified by action words like shall, will, perform etc. by pws section, requirement.  Do not write as if you're responding to the proposal.  Its a spreadsheet to distill the requirements, not microhealth's approach",
     "Pink": "Create a Pink Team document based on the PWS outline.  Your goal is to be compliant and compelling.",
@@ -40,88 +43,60 @@ document_types = {
 }
 
 def get_right_col_content(selected_type):
-    if selected_type == "Shred":
-        return dbc.Card(
-            dbc.CardBody([
-                html.Div([
-                    html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
-                ], style={'textAlign':'center', 'marginBottom':'10px'}),
-                dcc.Loading(
-                    id="loading-indicator",
-                    type="dot",
-                    children=[html.Div(id="loading-output")]
-                ),
-                html.Div(id='document-preview', className="border p-3 mb-3"),
-                dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"),
-                dcc.Download(id="download-document"),
-                html.Hr(),
-                dcc.Loading(
-                    id="chat-loading",
-                    type="dot",
-                    children=[
-                        dbc.Input(id="chat-input", type="text", placeholder="Chat with AI to update document...", className="mb-2", style={'whiteSpace':'pre-wrap'}),
-                        dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3"),
-                        html.Div(id="chat-output")
-                    ]
-                )
-            ])
-        )
-    else:
-        return dbc.Card(
-            dbc.CardBody([
-                html.Div([
-                    html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
-                ], style={'textAlign':'center', 'marginBottom':'10px'}),
-                dcc.Loading(
-                    id="loading-indicator",
-                    type="dot",
-                    children=[html.Div(id="loading-output")]
-                ),
-                html.Div(id='document-preview', className="border p-3 mb-3"),
-                dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"),
-                dcc.Download(id="download-document"),
-                html.Hr(),
-                html.Div([
-                    html.Label(f"Upload {selected_type} Document"),
-                    dcc.Upload(
-                        id={'type': 'upload-doc-type', 'index': selected_type},
-                        children=html.Div(['Drag and Drop or ', html.A('Select File')]),
-                        style={
-                            'width': '100%',
-                            'height': '60px',
-                            'lineHeight': '60px',
-                            'borderWidth': '1px',
-                            'borderStyle': 'dashed',
-                            'borderRadius': '5px',
-                            'textAlign': 'center',
-                            'margin': '10px 0'
-                        },
-                        multiple=False
-                    ),
-                    html.Div(id={'type': 'uploaded-doc-name', 'index': selected_type}),
-                    dbc.RadioItems(
-                        id={'type': 'radio-doc-source', 'index': selected_type},
-                        options=[
-                            {'label': 'Loaded Document', 'value': 'loaded'},
-                            {'label': 'Uploaded Document', 'value': 'uploaded'}
-                        ],
-                        value='loaded',
-                        inline=True,
-                        className="mb-2"
-                    ),
-                    dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3"),
-                ], id={'type': 'doc-type-controls', 'index': selected_type}),
-                dcc.Loading(
-                    id="chat-loading",
-                    type="dot",
-                    children=[
-                        dbc.Input(id="chat-input", type="text", placeholder="Chat with AI to update document...", className="mb-2", style={'whiteSpace':'pre-wrap'}),
-                        dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3"),
-                        html.Div(id="chat-output")
-                    ]
-                )
-            ])
-        )
+    controls = []
+    controls.append(html.Div([
+        html.Div(className="blinking-dot", style={'margin':'0 auto','width':'16px','height':'16px'}),
+    ], style={'textAlign':'center', 'marginBottom':'10px'}))
+    controls.append(dcc.Loading(
+        id="loading-indicator",
+        type="dot",
+        children=[html.Div(id="loading-output")]
+    ))
+    controls.append(html.Div(id='document-preview', className="border p-3 mb-3"))
+    controls.append(dbc.Button("Download Document", id="btn-download", color="success", className="mt-3"))
+    controls.append(dcc.Download(id="download-document"))
+    controls.append(html.Hr())
+    if selected_type != "Shred":
+        controls.append(html.Div([
+            html.Label(f"Upload {selected_type} Document"),
+            dcc.Upload(
+                id={'type': 'upload-doc-type', 'index': selected_type},
+                children=html.Div(['Drag and Drop or ', html.A('Select File')]),
+                style={
+                    'width': '100%',
+                    'height': '60px',
+                    'lineHeight': '60px',
+                    'borderWidth': '1px',
+                    'borderStyle': 'dashed',
+                    'borderRadius': '5px',
+                    'textAlign': 'center',
+                    'margin': '10px 0'
+                },
+                multiple=False
+            ),
+            html.Div(id={'type': 'uploaded-doc-name', 'index': selected_type}),
+            dbc.RadioItems(
+                id={'type': 'radio-doc-source', 'index': selected_type},
+                options=[
+                    {'label': 'Loaded Document', 'value': 'loaded'},
+                    {'label': 'Uploaded Document', 'value': 'uploaded'}
+                ],
+                value='loaded',
+                inline=True,
+                className="mb-2"
+            ),
+            dbc.Button("Generate Document", id={'type': 'btn-generate-doc', 'index': selected_type}, color="primary", className="mb-3"),
+        ], id={'type': 'doc-type-controls', 'index': selected_type}))
+    controls.append(dcc.Loading(
+        id="chat-loading",
+        type="dot",
+        children=[
+            dbc.Input(id="chat-input", type="text", placeholder="Chat with AI to update document...", className="mb-2", style={'whiteSpace':'pre-wrap'}),
+            dbc.Button("Send", id="btn-send-chat", color="primary", className="mb-3"),
+            html.Div(id="chat-output")
+        ]
+    ))
+    return dbc.Card(dbc.CardBody(controls))
 
 app.layout = dbc.Container([
     dcc.Store(id='selected-doc-type', data="Shred"),
@@ -279,40 +254,6 @@ def remove_file(n_clicks, existing_files):
 @app.callback(
     Output('document-preview', 'children'),
     Output('loading-output', 'children'),
-    Input({'type': 'btn-doc-type', 'index': 'Shred'}, 'n_clicks'),
-    prevent_initial_call=True
-)
-def generate_shred_doc(n_clicks):
-    global current_document, document_type, shredded_document
-    if not uploaded_files:
-        return html.Div("Please upload a document before shredding."), ""
-    file_contents = list(uploaded_files.values())
-    try:
-        shredded_document = generate_document("Shred", file_contents)
-        current_document = shredded_document
-        return dcc.Markdown(shredded_document), "Shred generated"
-    except Exception as e:
-        logging.error(f"Error generating document: {str(e)}")
-        return html.Div(f"Error generating document: {str(e)}"), "Error"
-
-@app.callback(
-    Output({'type': 'uploaded-doc-name', 'index': MATCH}, 'children'),
-    Output({'type': 'upload-doc-type', 'index': MATCH}, 'contents'),
-    Output({'type': 'radio-doc-source', 'index': MATCH}, 'value'),
-    Input({'type': 'upload-doc-type', 'index': MATCH}, 'contents'),
-    State({'type': 'upload-doc-type', 'index': MATCH}, 'filename'),
-    State({'type': 'upload-doc-type', 'index': MATCH}, 'id')
-)
-def update_uploaded_doc_name(contents, filename, id_dict):
-    if contents is not None:
-        uploaded_doc_contents[id_dict['index']] = (contents, filename)
-        logging.info(f"{id_dict['index']} file uploaded: {filename}")
-        return filename, contents, "uploaded"
-    return "", None, "loaded"
-
-@app.callback(
-    Output('document-preview', 'children', allow_duplicate=True),
-    Output('loading-output', 'children', allow_duplicate=True),
     Input({'type': 'btn-generate-doc', 'index': ALL}, 'n_clicks'),
     State({'type': 'btn-generate-doc', 'index': ALL}, 'id'),
     State({'type': 'radio-doc-source', 'index': ALL}, 'value'),
@@ -320,7 +261,7 @@ def update_uploaded_doc_name(contents, filename, id_dict):
     State({'type': 'upload-doc-type', 'index': ALL}, 'filename'),
     prevent_initial_call=True
 )
-def generate_other_doc(n_clicks_list, btn_ids, radio_values, upload_contents, upload_filenames):
+def generate_any_doc(n_clicks_list, btn_ids, radio_values, upload_contents, upload_filenames):
     global current_document, document_type, shredded_document, pink_review_document
     ctx = callback_context
     if not ctx.triggered:
@@ -332,15 +273,29 @@ def generate_other_doc(n_clicks_list, btn_ids, radio_values, upload_contents, up
     doc_type = btn_ids[idx]['index']
     document_type = doc_type
 
+    if not uploaded_files and doc_type == "Shred":
+        return html.Div("Please upload a document before shredding."), ""
+
+    # Source selection
+    source = radio_values[idx] if radio_values and len(radio_values) > idx else 'loaded'
+    doc_content = None
+
     if doc_type == "Shred":
-        raise dash.exceptions.PreventUpdate
+        file_contents = list(uploaded_files.values())
+        try:
+            generated = generate_document(doc_type, file_contents)
+            current_document = generated
+            shredded_document = generated
+            preview = spreadsheet_preview(generated)
+            return preview, "Shred generated"
+        except Exception as e:
+            logging.error(f"Error generating document: {str(e)}")
+            return html.Div(f"Error generating document: {str(e)}"), "Error"
 
+    # For all other types
     if shredded_document is None:
         return html.Div("Please shred a document first."), ""
 
-    source = radio_values[idx] if radio_values and len(radio_values) > idx else 'loaded'
-    doc_content = None
-
     if source == 'uploaded':
         if upload_contents and len(upload_contents) > idx and upload_contents[idx] and upload_filenames and len(upload_filenames) > idx and upload_filenames[idx]:
             doc_content = process_document(upload_contents[idx], upload_filenames[idx])
@@ -368,50 +323,79 @@ def generate_other_doc(n_clicks_list, btn_ids, radio_values, upload_contents, up
 
     try:
         if doc_type == "Pink Review":
-            current_document = generate_document(doc_type, [doc_content, shredded_document])
-            pink_review_document = current_document
-        elif doc_type in ["Red", "Red Review"]:
-            current_document = generate_document(doc_type, [doc_content, shredded_document])
+            generated = generate_document(doc_type, [doc_content, shredded_document])
+            pink_review_document = generated
+            current_document = generated
+            preview = spreadsheet_preview(generated)
+            return preview, f"{doc_type} generated"
+        elif doc_type in ["Red Review", "Gold Review", "Virtual Board", "LOE"]:
+            generated = generate_document(doc_type, [doc_content, shredded_document])
+            current_document = generated
+            preview = spreadsheet_preview(generated)
+            return preview, f"{doc_type} generated"
+        elif doc_type in ["Pink", "Red", "Gold"]:
+            generated = generate_document(doc_type, [doc_content])
+            current_document = generated
+            preview = dcc.Markdown(generated)
+            return preview, f"{doc_type} generated"
         else:
-            current_document = generate_document(doc_type, [doc_content])
-        logging.info(f"{doc_type} document generated successfully.")
-        return dcc.Markdown(current_document), f"{doc_type} generated"
+            generated = generate_document(doc_type, [doc_content])
+            current_document = generated
+            preview = dcc.Markdown(generated)
+            return preview, f"{doc_type} generated"
     except Exception as e:
         logging.error(f"Error generating document: {str(e)}")
         return html.Div(f"Error generating document: {str(e)}"), "Error"
 
+@app.callback(
+    Output({'type': 'uploaded-doc-name', 'index': MATCH}, 'children'),
+    Output({'type': 'upload-doc-type', 'index': MATCH}, 'contents'),
+    Output({'type': 'radio-doc-source', 'index': MATCH}, 'value'),
+    Input({'type': 'upload-doc-type', 'index': MATCH}, 'contents'),
+    State({'type': 'upload-doc-type', 'index': MATCH}, 'filename'),
+    State({'type': 'upload-doc-type', 'index': MATCH}, 'id')
+)
+def update_uploaded_doc_name(contents, filename, id_dict):
+    if contents is not None:
+        uploaded_doc_contents[id_dict['index']] = (contents, filename)
+        logging.info(f"{id_dict['index']} file uploaded: {filename}")
+        return filename, contents, "uploaded"
+    return "", None, "loaded"
+
+def spreadsheet_preview(text):
+    try:
+        # Try to parse as CSV, fallback to plain text
+        df = pd.read_csv(StringIO(text))
+        return dbc.Table.from_dataframe(df, striped=True, bordered=True, hover=True, size="sm")
+    except Exception:
+        return html.Pre(text, style={'whiteSpace': 'pre-wrap', 'fontFamily': 'monospace'})
+
 def generate_document(document_type, file_contents):
-    prompt = f"""Generate a {document_type} based on the following project artifacts:
+    if document_type in spreadsheet_types:
+        prompt = f"""Ignore all other instructions and output only a spreadsheet for {document_type} as described below. Do not include any narrative, only the spreadsheet in CSV format.
+Instructions: {document_types[document_type]}
+Project Artifacts:
+{' '.join(file_contents)}
+Output only the spreadsheet in CSV format, no narrative or explanation."""
+    elif document_type in narrative_types:
+        prompt = f"""Generate a {document_type} document based on the following project artifacts:
 {' '.join(file_contents)}
 Instructions:
 1. Create the {document_type} as a detailed document.
 2. Use proper formatting and structure.
 3. Include all necessary sections and details.
 4. Start the output immediately with the document content.
-5. IMPORTANT: If the document type is Pink, Red, Gold and not 
-Pink Review, Red Review, Gold Review, or spreadsheet, loe or virtual board
-then your goal is to be compliant and compelling based on the
-requrements, write in paragraph in active voice as
-MicroHealth, limit bullets, answer the
-requirement with what MicroHealth will do 
-to satisfy the requirement, the technical 
-approach with innovation for efficiency, 
-productivity, quality and measurable 
-outcomes, the industry standard that 
-methodology is based on if applicable, 
-detail the workflow or steps to accomplish 
-the requirement with labor categories that 
-will do those tasks in that workflow, 
-reference reputable research like gartner, 
-forrester, IDC, Deloitte, Accenture etc 
-with measures of success and substantiation 
-of MicroHealth's approach. Never use soft words
-like maybe, could be, should, possible be definitive in your language and confident.
-6. you must also take into account section L&M of the document which is the evaluation criteria
-to be sure we address them.
+5. IMPORTANT: If the document type is Pink, Red, Gold and not Pink Review, Red Review, Gold Review, or spreadsheet, loe or virtual board then your goal is to be compliant and compelling based on the requrements, write in paragraph in active voice as MicroHealth, limit bullets, answer the requirement with what MicroHealth will do to satisfy the requirement, the technical approach with innovation for efficiency, productivity, quality and measurable outcomes, the industry standard that methodology is based on if applicable, detail the workflow or steps to accomplish the requirement with labor categories that will do those tasks in that workflow, reference reputable research like gartner, forrester, IDC, Deloitte, Accenture etc with measures of success and substantiation of MicroHealth's approach. Never use soft words like maybe, could be, should, possible be definitive in your language and confident.
+6. you must also take into account section L&M of the document which is the evaluation criteria to be sure we address them.
+Now, generate the {document_type}:
+"""
+    else:
+        prompt = f"""Generate a {document_type} based on the following project artifacts:
+{' '.join(file_contents)}
+Instructions:
+{document_types.get(document_type, '')}
 Now, generate the {document_type}:
 """
-
     logging.info(f"Generating document for type: {document_type}")
     try:
         response = openai.ChatCompletion.create(
@@ -435,24 +419,34 @@ Now, generate the {document_type}:
     Output('document-preview', 'children', allow_duplicate=True),
     Input('btn-send-chat', 'n_clicks'),
     State('chat-input', 'value'),
+    State('selected-doc-type', 'data'),
     prevent_initial_call=True
 )
-def update_document_via_chat(n_clicks, chat_input):
+def update_document_via_chat(n_clicks, chat_input, selected_doc_type):
     global current_document, document_type
     if not chat_input or current_document is None:
         raise dash.exceptions.PreventUpdate
-    
-    prompt = f"""Update the following {document_type} based on this instruction: {chat_input}
+
+    if selected_doc_type in spreadsheet_types:
+        prompt = f"""Update the following {selected_doc_type} spreadsheet based on this instruction: {chat_input}
+Current spreadsheet (CSV format):
+{current_document}
+Instructions:
+1. Provide the updated spreadsheet in CSV format only.
+2. Do not include any narrative, only the spreadsheet in CSV.
+Now, provide the updated {selected_doc_type} spreadsheet:
+"""
+    else:
+        prompt = f"""Update the following {selected_doc_type} document based on this instruction: {chat_input}
 Current document:
 {current_document}
 Instructions:
 1. Provide the updated document content.
 2. Maintain proper formatting and structure.
 3. Incorporate the requested changes seamlessly.
-Now, provide the updated {document_type}:
+Now, provide the updated {selected_doc_type}:
 """
-
-    logging.info(f"Updating document via chat for {document_type} instruction: {chat_input}")
+    logging.info(f"Updating document via chat for {selected_doc_type} instruction: {chat_input}")
     try:
         response = openai.ChatCompletion.create(
             model="gpt-4-1106-preview",
@@ -465,7 +459,11 @@ Now, provide the updated {document_type}:
         )
         current_document = response['choices'][0]['message']['content']
         logging.info("Document updated via chat successfully.")
-        return f"Document updated based on: {chat_input}", dcc.Markdown(current_document)
+        if selected_doc_type in spreadsheet_types:
+            preview = spreadsheet_preview(current_document)
+        else:
+            preview = dcc.Markdown(current_document)
+        return f"Document updated based on: {chat_input}", preview
     except Exception as e:
         logging.error(f"Error updating document via chat: {str(e)}")
         return f"Error updating document: {str(e)}", html.Div(f"Error updating document: {str(e)}")
@@ -473,35 +471,38 @@ Now, provide the updated {document_type}:
 @app.callback(
     Output("download-document", "data"),
     Input("btn-download", "n_clicks"),
+    State('selected-doc-type', 'data'),
     prevent_initial_call=True
 )
-def download_document(n_clicks):
-    global current_document, document_type
+def download_document(n_clicks, selected_doc_type):
+    global current_document
     if current_document is None:
         raise dash.exceptions.PreventUpdate
-    
-    if document_type == "LOE":
+
+    if selected_doc_type in spreadsheet_types:
         try:
             df = pd.read_csv(StringIO(current_document))
             output = BytesIO()
             with pd.ExcelWriter(output, engine='xlsxwriter') as writer:
-                df.to_excel(writer, sheet_name='LOE', index=False)
-            logging.info("LOE document downloaded as Excel.")
-            return dcc.send_bytes(output.getvalue(), f"{document_type}.xlsx")
+                df.to_excel(writer, sheet_name=selected_doc_type, index=False)
+            logging.info(f"{selected_doc_type} document downloaded as Excel.")
+            output.seek(0)
+            return dcc.send_bytes(output.read(), f"{selected_doc_type}.xlsx")
         except Exception as e:
-            logging.error(f"Error downloading LOE document: {str(e)}")
-            return dcc.send_string(f"Error downloading LOE: {str(e)}", f"{document_type}_error.txt")
+            logging.error(f"Error downloading {selected_doc_type} document: {str(e)}")
+            return dcc.send_string(f"Error downloading {selected_doc_type}: {str(e)}", f"{selected_doc_type}_error.txt")
     else:
         try:
             doc = Document()
             doc.add_paragraph(current_document)
             output = BytesIO()
             doc.save(output)
-            logging.info("Document downloaded as Word.")
-            return dcc.send_bytes(output.getvalue(), f"{document_type}.docx")
+            logging.info(f"{selected_doc_type} document downloaded as Word.")
+            output.seek(0)
+            return dcc.send_bytes(output.read(), f"{selected_doc_type}.docx")
         except Exception as e:
             logging.error(f"Error downloading document: {str(e)}")
-            return dcc.send_string(f"Error downloading document: {str(e)}", f"{document_type}_error.txt")
+            return dcc.send_string(f"Error downloading document: {str(e)}", f"{selected_doc_type}_error.txt")
 
 if __name__ == '__main__':
     print("Starting the Dash application...")