Update app.py

app.py

@@ -6,7 +6,8 @@ import time
 from typing import List, Tuple
 import re
 import pandas as pd
-
+from docx import Document
+from io import BytesIO
 import dash
 import dash_bootstrap_components as dbc
 from dash import html, dcc, Input, Output, State, ctx, dash_table
@@ -436,7 +437,12 @@ def update_r_review_output(n_clicks, contents, filename, red_doc, requirements):
 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")
+    content = shred_output['props']['children'] if isinstance(shred_output, dict) else str(shred_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "shred_outline.docx")
 
 @app.callback(
     Output("download-pink-doc", "data"),
@@ -447,7 +453,12 @@ def download_shred(n_clicks, shred_output):
 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")
+    content = pink_output['props']['children'] if isinstance(pink_output, dict) else str(pink_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "pink_team_document.docx")
 
 @app.callback(
     Output("download-p-review-doc", "data"),
@@ -458,7 +469,12 @@ def download_pink(n_clicks, pink_output):
 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")
+    content = p_review_output['props']['children'] if isinstance(p_review_output, dict) else str(p_review_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "p_review_report.docx")
 
 @app.callback(
     Output("download-red-doc", "data"),
@@ -469,7 +485,12 @@ def download_p_review(n_clicks, p_review_output):
 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")
+    content = red_output['props']['children'] if isinstance(red_output, dict) else str(red_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "red_team_document.docx")
 
 @app.callback(
     Output("download-r-review-doc", "data"),
@@ -480,7 +501,12 @@ def download_red(n_clicks, red_output):
 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")
+    content = r_review_output['props']['children'] if isinstance(r_review_output, dict) else str(r_review_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "r_review_report.docx")
 
 @app.callback(
     Output("download-g-review-doc", "data"),
@@ -491,7 +517,12 @@ def download_r_review(n_clicks, r_review_output):
 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")
+    content = g_review_output['props']['children'] if isinstance(g_review_output, dict) else str(g_review_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "g_review_report.docx")
 
 @app.callback(
     Output("download-loe-doc", "data"),
@@ -502,8 +533,15 @@ def download_g_review(n_clicks, g_review_output):
 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 isinstance(loe_output, list) and len(loe_output) > 0:
+        content = loe_output[0]['props']['children'] if isinstance(loe_output[0], dict) else str(loe_output[0])
+    else:
+        content = str(loe_output)
+    doc = create_docx(content)
+    buffer = BytesIO()
+    doc.save(buffer)
+    buffer.seek(0)
+    return dcc.send_bytes(buffer.getvalue(), "loe_report.docx")
 
 if __name__ == '__main__':
     print("Starting the Dash application...")