Update app.py

app.py

@@ -425,6 +425,10 @@ def generate_documentation_page():
     st.subheader(f"Generate Documentation for {st.session_state.current_project}")
     st.write("Enter the functionality or parts of the project for which you'd like to generate documentation.")
 
+    # Get the path of the current project
+    user_folder = os.path.join("user_projects", st.session_state.username)
+    project_folder = os.path.join(user_folder, st.session_state.current_project)
+
     # Prompt user for functionality description
     functionality = st.text_area(
         "Describe the functionality",
@@ -436,10 +440,6 @@ def generate_documentation_page():
         if functionality.strip():
             st.write("Analyzing project files... Please wait.")
 
-            # Get the path of the current project
-            user_folder = os.path.join("user_projects", st.session_state.username)
-            project_folder = os.path.join(user_folder, st.session_state.current_project)
-
             if os.path.exists(project_folder):
                 try:
                     # Gather all file paths in the project directory
@@ -492,18 +492,26 @@ def generate_documentation_page():
             st.download_button(
                 label="Download PDF",
                 data=open(pdf_file.name, "rb").read(),
-                file_name="documentation.pdf",
+                file_name=f"{st.session_state.current_project} Documentation.pdf",
                 mime="application/pdf",
             )
             os.unlink(pdf_file.name)
 
         # Button to generate Markdown file
         if st.button("Generate Markdown File"):
-            markdown_file_path = os.path.join(user_folder, f"{st.session_state.current_project}_documentation.md")
+            markdown_file_path = os.path.join(user_folder, f"{st.session_state.current_project} Documentation.md")
             with open(markdown_file_path, "w") as md_file:
                 md_file.write(f"# Documentation for {st.session_state.current_project}\n\n")
                 md_file.write(documentation)
-            st.success(f"Markdown file generated at: {markdown_file_path}")
+
+            # Save and allow download of the Markdown file
+            with open(markdown_file_path, "rb") as md_file:
+                st.download_button(
+                    label="Download Markdown File",
+                    data=md_file.read(),
+                    file_name=f"{st.session_state.current_project} Documentation.md",
+                    mime="text/markdown",
+                )
 
 # Helper function to generate PDF
 def generate_pdf(documentation):