Update app.py

app.py

@@ -110,17 +110,54 @@ def generate_red_document(document: str, compliance_report: str) -> str:
     {compliance_report}
 
     Generate a revised "Red Team" document that addresses all issues found in the compliance report. 
-    Ensure that the new document:
-    1. Responds to all requirements with sufficient detail
-    2. Explains how MicroHealth will implement solutions
-    3. Incorporates best industry practices
-    4. Includes measurable outcomes for each requirement
+    Follow these guidelines:
+    1. Use Wikipedia style writing with active voice.  Be firm with the approach, no soft words like could be, may be, should, might.  Use definitve language.
+    2. For each requirement, describe in detail how MicroHealth will innovate to address it.
+    3. Explain the industry best practices that will be applied and the workflow to accomplish the steps in the best practice to address the requirement.
+    4. Provide measurable outcomes for the customer.
+    5. Limit the use of bullet points and write predominantly in paragraph format.
+    6. Ensure a logical flow of steps taken by MicroHealth for each requirement.
 
-    Provide the revised document in a clear, structured format.
+    Provide the revised document in a clear, structured format using paragraphs.
+    Limit the use of bullet points and write predominantly in paragraph format.
+    Ensure a logical flow of steps taken by MicroHealth for each requirement.
     """
     response = model.generate_content(prompt)
     return response.text
 
+def generate_loe(document: str) -> Tuple[str, pd.DataFrame]:
+    prompt = f"""
+    Analyze the following document and provide a Level of Effort (LOE) breakdown:
+
+    Document:
+    {document}
+
+    For each section header in the document:
+    1. Identify the tasks to be completed
+    2. Determine the appropriate labor categories for each task
+    3. Estimate the number of hours required for each labor category to complete the task
+
+    Provide a detailed breakdown and then summarize the information in a tabular format with the following columns:
+    - Task Summary
+    - Labor Categories
+    - Hours per Labor Category
+    - Total Hours
+
+    Present the detailed breakdown first, followed by the summary table.
+    """
+    response = model.generate_content(prompt)
+    
+    # Extract the table from the response
+    table_start = response.text.find("| Task Summary |")
+    table_end = response.text.find("\n\n", table_start)
+    table_text = response.text[table_start:table_end]
+    
+    # Convert the table to a pandas DataFrame
+    df = pd.read_csv(io.StringIO(table_text), sep='|', skipinitialspace=True).dropna(axis=1, how='all')
+    df.columns = df.columns.str.strip()
+    
+    return response.text, df
+
 # Layout
 app.layout = dbc.Container([
     html.H1("MicroHealth PWS Analysis and Response Generator", className="my-4"),
@@ -224,6 +261,33 @@ app.layout = dbc.Container([
             dbc.Button("Download R.Review Report", id="download-r-review", className="mt-3"),
             dcc.Download(id="download-r-review-doc")
         ]),
+        dbc.Tab(label="G.Review", tab_id="g-review", children=[
+            dcc.Upload(
+                id='upload-g-review',
+                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
+            ),
+            dbc.Button("Evaluate Compliance", id="evaluate-g-review", className="mt-3"),
+            dbc.Spinner(html.Div(id='g-review-output')),
+            dbc.Button("Download G.Review Report", id="download-g-review", className="mt-3"),
+            dcc.Download(id="download-g-review-doc")
+        ]),
+        dbc.Tab(label="LOE", tab_id="loe", children=[
+            dbc.Button("Generate LOE", id="generate-loe", className="mt-3"),
+            dbc.Spinner(html.Div(id='loe-output')),
+            dbc.Button("Download LOE Report", id="download-loe", className="mt-3"),
+            dcc.Download(id="download-loe-doc")
+        ]),
     ], id="tabs", active_tab="shred"),
 ])
 
@@ -262,20 +326,57 @@ def update_pink_output(n_clicks, shred_output, instructions):
     State('pink-output', 'children'),
     State('shred-output', 'children')
 )
-def update_p_review_output(n_clicks, contents, filename, pink_doc, requirements):
+
+@app.callback(
+    Output('g-review-output', 'children'),
+    Input('evaluate-g-review', 'n_clicks'),
+    State('upload-g-review', 'contents'),
+    State('upload-g-review', 'filename'),
+    State('shred-output', 'children')
+)
+def update_g_review_output(n_clicks, contents, filename, requirements):
     if n_clicks is None:
         return "Click 'Evaluate Compliance' to begin."
     
-    if contents:
-        document = process_document(contents, filename)
-    elif pink_doc:
-        document = pink_doc
-    else:
-        return "Please upload a document or generate a Pink Team document first."
+    if contents is None:
+        return "Please upload a document first."
     
+    document = process_document(contents, filename)
     compliance_report = evaluate_compliance(document, requirements)
     return dcc.Markdown(compliance_report)
 
+@app.callback(
+    Output('loe-output', 'children'),
+    Input('generate-loe', 'n_clicks'),
+    State('shred-output', 'children')
+)
+def update_loe_output(n_clicks, shred_output):
+    if n_clicks is None:
+        return "Click 'Generate LOE' to begin."
+    
+    if shred_output is None:
+        return "Please complete the Shred tab first."
+    
+    loe_text, loe_df = generate_loe(shred_output)
+    
+    return [
+        dcc.Markdown(loe_text),
+        dash_table.DataTable(
+            data=loe_df.to_dict('records'),
+            columns=[{'name': i, 'id': i} for i in loe_df.columns],
+            style_table={'overflowX': 'auto'},
+            style_cell={'textAlign': 'left', 'padding': '5px'},
+            style_header={'backgroundColor': 'rgb(230, 230, 230)', 'fontWeight': 'bold'}
+        )
+    ]
+
+@app.callback(
+    [Output(f"tab-{tab}", "disabled") for tab in ["pink", "p-review", "red", "r-review", "g-review", "loe"]],
+    Input('shred-output', 'children')
+)
+def update_tab_status(shred_output):
+    return [shred_output is None] * 6
+
 @app.callback(
     Output('red-output', 'children'),
     Input('generate-red', 'n_clicks'),
@@ -364,15 +465,27 @@ def download_red(n_clicks, red_output):
     return dict(content=red_output, filename="red_team_document.md")
 
 @app.callback(
-    Output("download-r-review-doc", "data"),
-    Input("download-r-review", "n_clicks"),
-    State('r-review-output', 'children'),
+    Output("download-g-review-doc", "data"),
+    Input("download-g-review", "n_clicks"),
+    State('g-review-output', 'children'),
+    prevent_initial_call=True,
+)
+def download_g_review(n_clicks, g_review_output):
+    if g_review_output is None:
+        return dash.no_update
+    return dict(content=g_review_output, filename="g_review_report.md")
+
+@app.callback(
+    Output("download-loe-doc", "data"),
+    Input("download-loe", "n_clicks"),
+    State('loe-output', 'children'),
     prevent_initial_call=True,
 )
-def download_r_review(n_clicks, r_review_output):
-    if r_review_output is None:
+def download_loe(n_clicks, loe_output):
+    if loe_output is None:
         return dash.no_update
-    return dict(content=r_review_output, filename="r_review_report.md")
+    loe_text = loe_output[0]['props']['children']
+    return dict(content=loe_text, filename="loe_report.md")
 
 if __name__ == '__main__':
     print("Starting the Dash application...")