Update app.py

app.py

@@ -470,35 +470,49 @@ def generate_documentation_page():
 
         # Define paths for PDF and Markdown files
         user_folder = os.path.join("user_projects", st.session_state.username)
+        os.makedirs(user_folder, exist_ok=True)  # Ensure the user folder exists
+
         pdf_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.pdf")
         markdown_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.md")
 
         # Button to generate PDF
         if st.button("Generate PDF"):
-            generate_pdf(documentation, pdf_path)
-            st.success(f"PDF file generated successfully: {pdf_path}")
+            try:
+                generate_pdf(documentation, pdf_path)
+                st.success(f"PDF file generated successfully: {pdf_path}")
+            except Exception as e:
+                st.error(f"Failed to generate PDF: {e}")
 
         # Button to generate Markdown file
         if st.button("Generate Markdown File"):
-            generate_markdown_file(documentation, markdown_path)
-            st.success(f"Markdown file generated successfully: {markdown_path}")
+            try:
+                generate_markdown_file(documentation, markdown_path)
+                st.success(f"Markdown file generated successfully: {markdown_path}")
+            except Exception as e:
+                st.error(f"Failed to generate Markdown file: {e}")
 
         # Provide download buttons
-        with open(pdf_path, "rb") as pdf_file:
-            st.download_button(
-                label="Download PDF",
-                data=pdf_file.read(),
-                file_name=f"{st.session_state.current_project}_Documentation.pdf",
-                mime="application/pdf",
-            )
-
-        with open(markdown_path, "rb") as markdown_file:
-            st.download_button(
-                label="Download Markdown File",
-                data=markdown_file.read(),
-                file_name=f"{st.session_state.current_project}_Documentation.md",
-                mime="text/markdown",
-            )
+        try:
+            with open(pdf_path, "rb") as pdf_file:
+                st.download_button(
+                    label="Download PDF",
+                    data=pdf_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.pdf",
+                    mime="application/pdf",
+                )
+        except FileNotFoundError:
+            st.error("PDF file not found. Please regenerate it.")
+
+        try:
+            with open(markdown_path, "rb") as markdown_file:
+                st.download_button(
+                    label="Download Markdown File",
+                    data=markdown_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.md",
+                    mime="text/markdown",
+                )
+        except FileNotFoundError:
+            st.error("Markdown file not found. Please regenerate it.")
 
 
 # Helper function to generate PDF