Update app.py

app.py

@@ -446,13 +446,6 @@ 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.")
 
-    # Clear previously generated documentation files when starting a new analysis
-    if st.session_state.get("new_analysis_started", False):
-        st.session_state.pop("pdf_generated", None)
-        st.session_state.pop("markdown_generated", None)
-        st.session_state.pop("generated_documentation", None)
-        st.session_state.new_analysis_started = False
-
     # Prompt user for functionality description
     functionality = st.text_area(
         "Describe the functionality",
@@ -462,36 +455,35 @@ def generate_documentation_page():
     # Button to start analyzing functionality
     if st.button("Analyze"):
         if functionality.strip():
-            st.session_state.new_analysis_started = True
-            st.write("Analyzing project files... Please wait.")
+            with st.spinner("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)
+                # 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
-                    file_paths = read_project_files(project_folder)
+                if os.path.exists(project_folder):
+                    try:
+                        # Gather all file paths in the project directory
+                        file_paths = read_project_files(project_folder)
 
-                    # Read file contents
-                    file_contents = read_files(file_paths)
+                        # Read file contents
+                        file_contents = read_files(file_paths)
 
-                    # Generate documentation using Gemini
-                    documentation = generate_detailed_documentation(file_contents, functionality)
+                        # Generate documentation using Gemini
+                        documentation = generate_detailed_documentation(file_contents, functionality)
 
-                    # Save the documentation in session state for exporting and viewing
-                    st.session_state.generated_documentation = documentation
+                        # Save the documentation in session state for exporting and viewing
+                        st.session_state.generated_documentation = documentation
 
-                    # Display the final documentation
-                    st.success("Documentation generated successfully!")
-                    st.text_area("Generated Documentation", documentation, height=600)
-                except Exception as e:
-                    st.error(f"An error occurred: {e}")
+                        # Display the final documentation
+                        st.success("Documentation generated successfully!")
+                        st.text_area("Generated Documentation", documentation, height=600)
+                    except Exception as e:
+                        st.error(f"An error occurred: {e}")
+                else:
+                    st.error("Project folder not found. Ensure the GitHub repository was cloned successfully.")
             else:
-                st.error("Project folder not found. Ensure the GitHub repository was cloned successfully.")
-        else:
-            st.error("Please enter the functionality to analyze.")
+                st.error("Please enter the functionality to analyze.")
 
     # Add export/download buttons if documentation is available
     if "generated_documentation" in st.session_state and st.session_state.generated_documentation:
@@ -504,53 +496,50 @@ def generate_documentation_page():
         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")
 
-        col1, col2 = st.columns(2)
+        # Show appropriate buttons based on file generation status
+        pdf_generated = os.path.exists(pdf_path)
+        markdown_generated = os.path.exists(markdown_path)
 
-        # PDF generation and download logic
-        with col1:
-            if not st.session_state.get("pdf_generated", False):
-                if st.button("Generate PDF"):
-                    try:
-                        generate_pdf(documentation, pdf_path)
-                        st.session_state.pdf_generated = True
-                        st.success("PDF Available in the 'Saved Documentation' Page")
-                        st.rerun()
-                    except Exception as e:
-                        st.error(f"Failed to generate PDF: {e}")
-            else:
+        # Generate and replace buttons dynamically
+        if not pdf_generated:
+            if st.button("Generate PDF"):
                 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. Try generating it again.")
-
-        # Markdown generation and download logic
-        with col2:
-            if not st.session_state.get("markdown_generated", False):
-                if st.button("Generate Markdown File"):
-                    try:
-                        generate_markdown_file(documentation, markdown_path)
-                        st.session_state.markdown_generated = True
-                        st.success("Markdown Available in the 'Saved Documentation' Page")
-                        st.rerun()
-                    except Exception as e:
-                        st.error(f"Failed to generate Markdown file: {e}")
-            else:
+                    generate_pdf(documentation, pdf_path)
+                    st.session_state.pdf_generated = True  # Set a flag for PDF generation
+                    st.rerun()  # Trigger a rerun to show the download button
+                except Exception as e:
+                    st.error(f"Failed to generate PDF: {e}")
+        else:
+            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.write("Click 'Generate PDF' to be able to download and store the documentation as a PDF")
+
+        if not markdown_generated:
+            if st.button("Generate Markdown File"):
                 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. Try generating it again.")
+                    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
+                except Exception as e:
+                    st.error(f"Failed to generate Markdown file: {e}")
+        else:
+            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.write("Click 'Generate Markdown File' to be able to download and store the documentation as a markdown file")
 
 
 
@@ -613,11 +602,12 @@ def generate_markdown_file(documentation, output_path):
 
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 def saved_documentation_page():
+    # Sidebar with "Back to Project" and "Log Out" buttons
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
     st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Project"):
         st.session_state.page = "project_view"
-        st.experimental_rerun()
+        st.rerun()
     if st.sidebar.button("Log Out"):
         logout_user()
         st.rerun()
@@ -630,25 +620,31 @@ def saved_documentation_page():
     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")
 
-    # Check if files exist and display download options
-    if os.path.exists(pdf_path):
-        with open(pdf_path, "rb") as pdf_file:
-            st.download_button(
-                label="Download PDF Documentation",
-                data=pdf_file.read(),
-                file_name=f"{st.session_state.current_project}_Documentation.pdf",
-                mime="application/pdf",
-            )
-    if os.path.exists(markdown_path):
-        with open(markdown_path, "rb") as markdown_file:
-            st.download_button(
-                label="Download Markdown Documentation",
-                data=markdown_file.read(),
-                file_name=f"{st.session_state.current_project}_Documentation.md",
-                mime="text/markdown",
-            )
-    if not os.path.exists(pdf_path) and not os.path.exists(markdown_path):
-        st.info("No documentation has been generated yet.")
+    # Check if files exist
+    pdf_exists = os.path.exists(pdf_path)
+    markdown_exists = os.path.exists(markdown_path)
+
+    # Display available documentation
+    if not pdf_exists and not markdown_exists:
+        st.info("No documentation has been generated yet. Please go to the 'Generate Documentation' page.")
+    else:
+        if pdf_exists:
+            with open(pdf_path, "rb") as pdf_file:
+                st.download_button(
+                    label="Download PDF Documentation",
+                    data=pdf_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.pdf",
+                    mime="application/pdf",
+                )
+
+        if markdown_exists:
+            with open(markdown_path, "rb") as markdown_file:
+                st.download_button(
+                    label="Download Markdown Documentation",
+                    data=markdown_file.read(),
+                    file_name=f"{st.session_state.current_project}_Documentation.md",
+                    mime="text/markdown",
+                )
 
 
 def project_view_page():