Update app.py

app.py

@@ -5,10 +5,11 @@ import threading
 import time
 from typing import List, Tuple
 import re
-import pandas as pd 
+import pandas as pd
+
 import dash
 import dash_bootstrap_components as dbc
-from dash import html, dcc, Input, Output, State, ctx
+from dash import html, dcc, Input, Output, State, ctx, dash_table
 import google.generativeai as genai
 from docx import Document
 from PyPDF2 import PdfReader
@@ -326,145 +327,20 @@ def update_pink_output(n_clicks, shred_output, instructions):
     State('pink-output', 'children'),
     State('shred-output', 'children')
 )
-
-@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 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'),
-    State('upload-red', 'contents'),
-    State('upload-red', 'filename'),
-    State('p-review-output', 'children')
-)
-def update_red_output(n_clicks, contents, filename, p_review_output):
-    if n_clicks is None:
-        return "Click 'Generate Red Team Document' to begin."
-    
-    if contents:
-        document = process_document(contents, filename)
-    elif p_review_output:
-        document = p_review_output
-    else:
-        return "Please upload a document or complete the P.Review first."
-    
-    red_doc = generate_red_document(document, p_review_output)
-    return dcc.Markdown(red_doc)
-
-@app.callback(
-    Output('r-review-output', 'children'),
-    Input('evaluate-r-review', 'n_clicks'),
-    State('upload-r-review', 'contents'),
-    State('upload-r-review', 'filename'),
-    State('red-output', 'children'),
-    State('shred-output', 'children')
-)
-def update_r_review_output(n_clicks, contents, filename, red_doc, requirements):
+def update_p_review_output(n_clicks, contents, filename, pink_doc, requirements):
     if n_clicks is None:
         return "Click 'Evaluate Compliance' to begin."
     
     if contents:
         document = process_document(contents, filename)
-    elif red_doc:
-        document = red_doc
+    elif pink_doc:
+        document = pink_doc
     else:
-        return "Please upload a document or generate a Red Team document first."
+        return "Please upload a document or generate a Pink Team document first."
     
     compliance_report = evaluate_compliance(document, requirements)
     return dcc.Markdown(compliance_report)
 
-@app.callback(
-    Output("download-shred-doc", "data"),
-    Input("download-shred", "n_clicks"),
-    State('shred-output', 'children'),
-    prevent_initial_call=True,
-)
-def download_shred(n_clicks, shred_output):
-    if shred_output is None:
-        return dash.no_update
-    return dict(content=shred_output, filename="shred_outline.md")
-
-@app.callback(
-    Output("download-pink-doc", "data"),
-    Input("download-pink", "n_clicks"),
-    State('pink-output', 'children'),
-    prevent_initial_call=True,
-)
-def download_pink(n_clicks, pink_output):
-    if pink_output is None:
-        return dash.no_update
-    return dict(content=pink_output, filename="pink_team_document.md")
-
-@app.callback(
-    Output("download-p-review-doc", "data"),
-    Input("download-p-review", "n_clicks"),
-    State('p-review-output', 'children'),
-    prevent_initial_call=True,
-)
-def download_p_review(n_clicks, p_review_output):
-    if p_review_output is None:
-        return dash.no_update
-    return dict(content=p_review_output, filename="p_review_report.md")
-
-@app.callback(
-    Output("download-red-doc", "data"),
-    Input("download-red", "n_clicks"),
-    State('red-output', 'children'),
-    prevent_initial_call=True,
-)
-def download_red(n_clicks, red_output):
-    if red_output is None:
-        return dash.no_update
-    return dict(content=red_output, filename="red_team_document.md")
-
-# Update the g-review callback
 @app.callback(
     Output('g-review-output', 'children'),
     Input('evaluate-g-review', 'n_clicks'),
@@ -483,86 +359,14 @@ def update_g_review_output(n_clicks, contents, filename, requirements):
     compliance_report = evaluate_compliance(document, requirements)
     return dcc.Markdown(compliance_report)
 
-# Update the tab disabling callback
 @app.callback(
-    [Output(f"tabs", "active_tab")] +
-    [Output(f"tabs", "children") for _ in range(6)],
-    Input('shred-output', 'children')
+    Output('loe-output', 'children'),
+    Input('generate-loe', 'n_clicks'),
+    State('shred-output', 'children')
 )
-def update_tab_status(shred_output):
-    tabs = [
-        dbc.Tab(label="Shred", tab_id="shred"),
-        dbc.Tab(label="Pink", tab_id="pink"),
-        dbc.Tab(label="P.Review", tab_id="p-review"),
-        dbc.Tab(label="Red", tab_id="red"),
-        dbc.Tab(label="R.Review", tab_id="r-review"),
-        dbc.Tab(label="G.Review", tab_id="g-review"),
-        dbc.Tab(label="LOE", tab_id="loe")
-    ]
+def update_loe_output(n_clicks, shred_output):
+    if n_clicks is None:
+        return "Click 'Generate LOE' to begin."
     
     if shred_output is None:
-        for tab in tabs[1:]:
-            tab.disabled = True
-    else:
-        for tab in tabs[1:]:
-            tab.disabled = False
-    
-    return ["shred"] + tabs
-
-# Update the app layout
-app.layout = dbc.Container([
-    html.H1("MicroHealth PWS Analysis and Response Generator", className="my-4"),
-    dbc.Tabs(id="tabs", active_tab="shred"),
-    html.Div(id='tab-content')
-])
-
-# Add a callback to update tab content
-@app.callback(
-    Output('tab-content', 'children'),
-    Input('tabs', 'active_tab')
-)
-def render_tab_content(active_tab):
-    if active_tab == "shred":
-        return [
-            dbc.Textarea(
-                id='shred-instructions',
-                placeholder="Enter any additional instructions for shredding the document...",
-                style={'height': '100px', 'marginBottom': '10px'}
-            ),
-            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
-            ),
-            dbc.Spinner(html.Div(id='shred-output')),
-            dbc.Button("Download Outline", id="download-shred", className="mt-3"),
-            dcc.Download(id="download-shred-doc")
-        ]
-    elif active_tab == "pink":
-        return [
-            dbc.Textarea(
-                id='pink-instructions',
-                placeholder="Enter any additional instructions for generating the Pink Team document...",
-                style={'height': '100px', 'marginBottom': '10px'}
-            ),
-            dbc.Button("Generate Pink Team Document", id="generate-pink", className="mt-3"),
-            dbc.Spinner(html.Div(id='pink-output')),
-            dbc.Button("Download Pink Team Document", id="download-pink", className="mt-3"),
-            dcc.Download(id="download-pink-doc")
-        ]
-    # Add similar content for other tabs...
-
-if __name__ == '__main__':
-    print("Starting the Dash application...")
-    app.run(debug=True, host='0.0.0.0', port=7860)
-    print("Dash application has finished running.")
+        return "Please complete the Sh