Update app.py

app.py

@@ -541,6 +541,11 @@ def generate_pdf(documentation):
 
 # Helper function to generate Markdown file
 def generate_markdown_file(documentation):
+    # Ensure the user folder is defined
+    user_folder = os.path.join("user_projects", st.session_state.username)
+    os.makedirs(user_folder, exist_ok=True)
+
+    # Generate the Markdown file name
     project_name = st.session_state.current_project
     markdown_file_name = f"{project_name} Documentation.md"
 
@@ -562,7 +567,7 @@ def generate_markdown_file(documentation):
     with open(markdown_file.name, "w") as f:
         f.write(formatted_documentation)
 
-    # Download the Markdown file
+    # Provide download button for the Markdown file
     st.download_button(
         label="Download Markdown File",
         data=open(markdown_file.name, "rb").read(),
@@ -571,6 +576,7 @@ def generate_markdown_file(documentation):
     )
     os.unlink(markdown_file.name)
 
+
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------