Update app.py

app.py

@@ -433,6 +433,7 @@ def process_gemini_output(output):
 
 
 def generate_documentation_page():
+    # Sidebar with "Log Out" and "Back to Project" 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"):
@@ -445,81 +446,107 @@ 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",
         placeholder="e.g., Explain the function of the file `main.py`",
     )
 
-    # Reset generated files when a new analysis starts
+    # Button to start analyzing functionality
     if st.button("Analyze"):
         if functionality.strip():
-            st.session_state.generated_documentation = None
-            st.session_state.generated_pdf = None
-            st.session_state.generated_markdown = None
+            st.session_state.new_analysis_started = True
             st.write("Analyzing project files... Please wait.")
-            # Simulate analysis
-            import time
-            time.sleep(3)  # Replace this with actual analysis logic
 
-            # Example analysis result
-            st.session_state.generated_documentation = f"Documentation for functionality: {functionality}"
-            st.success("Analysis complete! Documentation is ready.")
+            # 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)
+
+                    # Read file contents
+                    file_contents = read_files(file_paths)
+
+                    # 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
+
+                    # 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("Please enter the functionality to analyze.")
 
-    # Display the documentation if available
-    if st.session_state.get("generated_documentation"):
-        st.text_area(
-            "Generated Documentation",
-            st.session_state.generated_documentation,
-            height=400,
-        )
+    # Add export/download buttons if documentation is available
+    if "generated_documentation" in st.session_state and st.session_state.generated_documentation:
+        documentation = st.session_state.generated_documentation
 
         # 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)
+        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")
 
-        # Handle PDF generation and download
-        if st.session_state.get("generated_pdf"):
-            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",
-                )
-        else:
+        # PDF generation and download logic
+        if not st.session_state.get("pdf_generated", False):
             if st.button("Generate PDF"):
                 try:
-                    generate_pdf(st.session_state.generated_documentation, pdf_path)
-                    st.session_state.generated_pdf = True
-                    st.info("PDF Available in the 'Saved Documentation' Page")
-                    st.experimental_rerun()  # Trigger rerun to show the download button
+                    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}")
-
-        # Handle Markdown generation and download
-        if st.session_state.get("generated_markdown"):
-            with open(markdown_path, "rb") as markdown_file:
-                st.download_button(
-                    label="Download Markdown",
-                    data=markdown_file.read(),
-                    file_name=f"{st.session_state.current_project}_Documentation.md",
-                    mime="text/markdown",
-                )
         else:
-            if st.button("Generate 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. Try generating it again.")
+
+        # Markdown generation and download logic
+        if not st.session_state.get("markdown_generated", False):
+            if st.button("Generate Markdown File"):
                 try:
-                    generate_markdown_file(st.session_state.generated_documentation, markdown_path)
-                    st.session_state.generated_markdown = True
-                    st.info("Markdown Available in the 'Saved Documentation' Page")
-                    st.experimental_rerun()  # Trigger rerun to show the download button
+                    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:
+            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.")