Update app.py

app.py

@@ -369,4 +369,143 @@ def update_loe_output(n_clicks, shred_output):
         return "Click 'Generate LOE' to begin."
     
     if shred_output is None:
-        return "Please complete the Sh
+        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('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):
+    if n_clicks is None:
+        return "Click 'Evaluate Compliance' to begin."
+    
+    if contents:
+        document = process_document(contents, filename)
+    elif red_doc:
+        document = red_doc
+    else:
+        return "Please upload a document or generate a Red 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")
+
+@app.callback(
+    Output("download-r-review-doc", "data"),
+    Input("download-r-review", "n_clicks"),
+    State('r-review-output', 'children'),
+    prevent_initial_call=True,
+)
+def download_r_review(n_clicks, r_review_output):
+    if r_review_output is None:
+        return dash.no_update
+    return dict(content=r_review_output, filename="r_review_report.md")
+
+@app.callback(
+    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_loe(n_clicks, loe_output):
+    if loe_output is None:
+        return dash.no_update
+    loe_text = loe_output[0]['props']['children']
+    return dict(content=loe_text, filename="loe_report.md")
+
+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.")