Update app.py

app.py

@@ -514,7 +514,7 @@ def update_red_output(n_clicks, contents, filename, p_review_output, instruction
         return "Please upload a document or complete the P.Review first."
     
     red_doc = generate_red_document(document, p_review_output, instructions or "")
-    return dcc.Markdown(red_doc)
+    return dcc.Markdown(red_doc)  # Wrap the output in dcc.Markdown)
 
 @app.callback(
     Output('r-review-output', 'children'),
@@ -634,7 +634,14 @@ 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
-    doc = create_docx(red_output)
+    
+    # Extract the content from the Markdown component
+    if isinstance(red_output, dict) and 'props' in red_output and 'children' in red_output['props']:
+        content = red_output['props']['children']
+    else:
+        content = str(red_output)
+    
+    doc = create_docx(content)
     buffer = BytesIO()
     doc.save(buffer)
     return dcc.send_bytes(buffer.getvalue(), "red_team_document.docx")