Update app.py

app.py

@@ -534,11 +534,9 @@ def create_docx(content):
 def download_shred(n_clicks, shred_output):
     if shred_output is None:
         return dash.no_update
-    content = shred_output['props']['children'] if isinstance(shred_output, dict) else str(shred_output)
-    doc = create_docx(content)
+    doc = create_docx(shred_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "shred_outline.docx")
 
 @app.callback(
@@ -550,11 +548,9 @@ def download_shred(n_clicks, shred_output):
 def download_pink(n_clicks, pink_output):
     if pink_output is None:
         return dash.no_update
-    content = pink_output['props']['children'] if isinstance(pink_output, dict) else str(pink_output)
-    doc = create_docx(content)
+    doc = create_docx(pink_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "pink_team_document.docx")
 
 @app.callback(
@@ -566,11 +562,9 @@ 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
-    content = p_review_output['props']['children'] if isinstance(p_review_output, dict) else str(p_review_output)
-    doc = create_docx(content)
+    doc = create_docx(p_review_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "p_review_report.docx")
 
 @app.callback(
@@ -582,11 +576,9 @@ 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
-    content = red_output['props']['children'] if isinstance(red_output, dict) else str(red_output)
-    doc = create_docx(content)
+    doc = create_docx(red_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "red_team_document.docx")
 
 @app.callback(
@@ -598,11 +590,9 @@ 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
-    content = r_review_output['props']['children'] if isinstance(r_review_output, dict) else str(r_review_output)
-    doc = create_docx(content)
+    doc = create_docx(r_review_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "r_review_report.docx")
 
 @app.callback(
@@ -614,11 +604,9 @@ 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
-    content = g_review_output['props']['children'] if isinstance(g_review_output, dict) else str(g_review_output)
-    doc = create_docx(content)
+    doc = create_docx(g_review_output)
     buffer = BytesIO()
     doc.save(buffer)
-    buffer.seek(0)
     return dcc.send_bytes(buffer.getvalue(), "g_review_report.docx")
 
 @app.callback(
@@ -627,6 +615,14 @@ def download_g_review(n_clicks, g_review_output):
     State('loe-output', 'children'),
     prevent_initial_call=True,
 )
+def download_loe(n_clicks, loe_output):
+    if loe_output is None or isinstance(loe_output, str):
+        return dash.no_update
+    loe_text = loe_output[0]['props']['children']
+    doc = create_docx(loe_text)
+    buffer = BytesIO()
+    doc.save(buffer)
+    return dcc.send_bytes(buffer.getvalue(), "loe_report.docx")
 
 @app.callback(
     Output('loe-output', 'children'),