Update app.py

app.py

@@ -99,6 +99,7 @@ def login_page():
                 st.session_state.authenticated = True
                 st.session_state.username = username
                 st.session_state.page = "project_staging"  # Updated from "workspace"
+                st.rerun()
             else:
                 st.error("Invalid username or password. Please try again.")
 
@@ -504,11 +505,10 @@ def generate_documentation_page():
         markdown_path = os.path.join(user_folder, f"{st.session_state.current_project}_Documentation.md")
 
         # Generate PDF Button
-        if not os.path.exists(pdf_path) and not pdf_generated:
+        if not os.path.exists(pdf_path):
             if st.button("Generate PDF"):
                 try:
                     generate_pdf(documentation, pdf_path)
-                    st.session_state.pdf_generated = True
                     st.rerun()
                     st.success("PDF Available in the 'Saved Documentation' Page.")
                 except Exception as e:
@@ -517,11 +517,10 @@ def generate_documentation_page():
             st.success("PDF has already been generated! Check the 'Saved Documentation' Page.")
 
         # Generate Markdown Button
-        if not os.path.exists(markdown_path) and not markdown_generated:
+        if not os.path.exists(markdown_path):
             if st.button("Generate Markdown File"):
                 try:
                     generate_markdown_file(documentation, markdown_path)
-                    st.session_state.markdown_generated = True  # Set a flag for Markdown generation
                     st.rerun()  # Trigger a rerun to show the download button
                     st.success("Markdown Available in the 'Saved Documentation' Page.")
                 except Exception as e: