added PDF format too

app.py

@@ -192,6 +192,23 @@ def initialize_state():
     st.session_state.previous_major_cycle = None # Track the previous cycle to detect changes
     logger.info("Streamlit session state initialized/reset.")
 
+    # --- ADDED Initialization for MD/PDF Paths ---
+    st.session_state.refined_prompt_path = None
+    st.session_state.refined_prompt_pdf_path = None
+    st.session_state.final_user_story_path = None
+    st.session_state.final_user_story_pdf_path = None
+    st.session_state.final_product_review_path = None
+    st.session_state.final_product_review_pdf_path = None
+    st.session_state.final_design_document_path = None
+    st.session_state.final_design_document_pdf_path = None
+    st.session_state.final_quality_analysis_path = None
+    st.session_state.final_quality_analysis_pdf_path = None
+    st.session_state.final_deployment_path = None
+    st.session_state.final_deployment_pdf_path = None
+    # --- END ADDED ---
+
+    logger.info("Streamlit session state initialized/reset including PDF paths.")
+
 def update_display(new_content: str):
     """Updates the main display area content in the session state."""
     st.session_state.display_content = new_content
@@ -410,14 +427,45 @@ with st.sidebar:
     st.header("Downloads")
     st.caption("Access generated artifacts and code snapshots.")
 
-    # Document Downloads
+    # --- MODIFIED: Document Downloads (MD and PDF) ---
+    st.markdown("---")
+    st.subheader("Documents") # Combined header
+
+    # Refined Prompt (Appears after Q&A cycle is complete)
+    st.markdown("**Requirements Cycle:**") # Header for this artifact
+    create_download_button(st.session_state.workflow_state.get("refined_prompt_path"), "Refined Prompt (MD)", "text/markdown", "refined_prompt_md", help_text="The final prompt generated after Q&A.")
+    create_download_button(st.session_state.workflow_state.get("refined_prompt_pdf_path"), "Refined Prompt (PDF)", "application/pdf", "refined_prompt_pdf", help_text="PDF version of the refined prompt.")
+    st.markdown("---") # Separator
+
+    # User Story
+    st.markdown("**User Story Cycle:**")
+    create_download_button(st.session_state.workflow_state.get("final_user_story_path"), "User Story (MD)", "text/markdown", "final_us_md")
+    create_download_button(st.session_state.workflow_state.get("final_user_story_pdf_path"), "User Story (PDF)", "application/pdf", "final_us_pdf")
     st.markdown("---")
-    st.subheader("Documents")
-    create_download_button(st.session_state.workflow_state.get("final_user_story_path"), "User Story", "text/markdown", "final_us")
-    create_download_button(st.session_state.workflow_state.get("final_product_review_path"), "Product Review", "text/markdown", "final_pr")
-    create_download_button(st.session_state.workflow_state.get("final_design_document_path"), "Design Document", "text/markdown", "final_dd")
-    create_download_button(st.session_state.workflow_state.get("final_quality_analysis_path"), "QA Report", "text/markdown", "final_qa")
-    create_download_button(st.session_state.workflow_state.get("final_deployment_path"), "Deployment Plan", "text/markdown", "final_deploy")
+
+    # Product Review
+    st.markdown("**Product Review Cycle:**")
+    create_download_button(st.session_state.workflow_state.get("final_product_review_path"), "Product Review (MD)", "text/markdown", "final_pr_md")
+    create_download_button(st.session_state.workflow_state.get("final_product_review_pdf_path"), "Product Review (PDF)", "application/pdf", "final_pr_pdf")
+    st.markdown("---")
+
+    # Design Document
+    st.markdown("**Design Cycle:**")
+    create_download_button(st.session_state.workflow_state.get("final_design_document_path"), "Design Document (MD)", "text/markdown", "final_dd_md")
+    create_download_button(st.session_state.workflow_state.get("final_design_document_pdf_path"), "Design Document (PDF)", "application/pdf", "final_dd_pdf")
+    st.markdown("---")
+
+    # QA Report
+    st.markdown("**Quality Analysis Cycle:**")
+    create_download_button(st.session_state.workflow_state.get("final_quality_analysis_path"), "QA Report (MD)", "text/markdown", "final_qa_md")
+    create_download_button(st.session_state.workflow_state.get("final_quality_analysis_pdf_path"), "QA Report (PDF)", "application/pdf", "final_qa_pdf")
+    st.markdown("---")
+
+    # Deployment Plan
+    st.markdown("**Deployment Cycle:**")
+    create_download_button(st.session_state.workflow_state.get("final_deployment_path"), "Deployment Plan (MD)", "text/markdown", "final_deploy_md")
+    create_download_button(st.session_state.workflow_state.get("final_deployment_pdf_path"), "Deployment Plan (PDF)", "application/pdf", "final_deploy_pdf")
+    # --- END MODIFIED Document Downloads ---
 
     # UML Diagram Downloads
     st.markdown("---")
@@ -784,13 +832,27 @@ with main_col:
                                     "final_uml_codes": [], "final_code_files": [], "final_code_review": "",
                                     "final_security_issues": "", "final_test_code_files": [], "final_quality_analysis": "",
                                     "final_deployment_process": "",
-                                    # File paths (initialize as None)
-                                    "final_user_story_path": None, "final_product_review_path": None, "final_design_document_path": None,
-                                    "final_uml_diagram_folder": None, "final_uml_png_paths": [],
-                                    "final_review_security_folder": None, "review_code_snapshot_folder": None,
-                                    "final_testing_folder": None, "testing_passed_code_folder": None,
-                                    "final_quality_analysis_path": None, "final_code_folder": None,
-                                    "final_deployment_path": None,
+                                    # --- ADDED/MODIFIED: Initialize MD/PDF paths ---
+                                    "refined_prompt_path": None,
+                                    "refined_prompt_pdf_path": None,
+                                    "final_user_story_path": None,
+                                    "final_user_story_pdf_path": None,
+                                    "final_product_review_path": None,
+                                    "final_product_review_pdf_path": None,
+                                    "final_design_document_path": None,
+                                    "final_design_document_pdf_path": None,
+                                    "final_uml_diagram_folder": None,
+                                    "final_uml_png_paths": [],
+                                    "final_review_security_folder": None,
+                                    "review_code_snapshot_folder": None,
+                                    "final_testing_folder": None,
+                                    "testing_passed_code_folder": None,
+                                    "final_quality_analysis_path": None, # MD path
+                                    "final_quality_analysis_pdf_path": None, # PDF path
+                                    "final_code_folder": None,
+                                    "final_deployment_path": None, # MD path
+                                    "final_deployment_pdf_path": None, # PDF path
+                                    # --- END ADDED/MODIFIED ---
                                 }
 
                                 # Update the main session state variables

requirements.txt

@@ -32,4 +32,7 @@ typing-extensions
 httpx
 requests
 
-nest_asyncio
+nest_asyncio
+
+markdown-it-py 
+weasyprint