Update app.py

app.py

@@ -433,16 +433,11 @@ 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"):
         st.session_state.page = "project_view"
-        # Clear the Gemini output when backing out
-        if "generated_documentation" in st.session_state:
-            del st.session_state["generated_documentation"]
-        st.rerun()
-
+        st.experimental_rerun()
     if st.sidebar.button("Log Out"):
         logout_user()
         st.rerun()
@@ -456,82 +451,76 @@ def generate_documentation_page():
         placeholder="e.g., Explain the function of the file `main.py`",
     )
 
-    # Button to start analyzing functionality
+    # Reset generated files when a new analysis starts
     if st.button("Analyze"):
         if functionality.strip():
-            # Show spinning wheel and message
-            with st.spinner("Analyzing. Please wait..."):
-                time.sleep(1)  # Simulate brief delay
-
-                # 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)
-
-                        # Simulate processing time based on lines of code
-                        total_lines = sum(len(content.splitlines()) for content in file_contents.values())
-                        estimated_time = (total_lines / 2500) * 60  # 1 minute per 2500 lines
-                        estimated_time = max(estimated_time, 2)  # Ensure at least 2 seconds for the spinner
-                        time.sleep(estimated_time)
-
-                        # 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.")
+            st.session_state.generated_documentation = None
+            st.session_state.generated_pdf = None
+            st.session_state.generated_markdown = None
+            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.")
         else:
             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:
-        documentation = st.session_state.generated_documentation
+    # Display the documentation if available
+    if st.session_state.get("generated_documentation"):
+        st.text_area(
+            "Generated Documentation",
+            st.session_state.generated_documentation,
+            height=400,
+        )
 
         # 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)  # Ensure the user folder exists
+        os.makedirs(user_folder, exist_ok=True)
 
         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")
 
-        # Generate PDF Button
-        if not os.path.exists(pdf_path):
+        # 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:
             if st.button("Generate PDF"):
                 try:
-                    generate_pdf(documentation, pdf_path)
-                    st.rerun()
-                    st.success("PDF Available in the 'Saved Documentation' Page.")
+                    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
                 except Exception as e:
                     st.error(f"Failed to generate PDF: {e}")
-        else:
-            st.success("PDF has already been generated! Check the 'Saved Documentation' Page.")
 
-        # Generate Markdown Button
-        if not os.path.exists(markdown_path):
-            if st.button("Generate Markdown File"):
+        # 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:
-                    generate_markdown_file(documentation, markdown_path)
-                    st.rerun()  # Trigger a rerun to show the download button
-                    st.success("Markdown Available in the 'Saved Documentation' Page.")
+                    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
                 except Exception as e:
                     st.error(f"Failed to generate Markdown file: {e}")
-        else:
-            st.success("Markdown has already been generated! Check the 'Saved Documentation' Page.")
+
 
 
 
@@ -592,12 +581,11 @@ 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.rerun()
+        st.experimental_rerun()
     if st.sidebar.button("Log Out"):
         logout_user()
         st.rerun()
@@ -610,31 +598,26 @@ 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
-    pdf_exists = os.path.exists(pdf_path)
-    markdown_exists = os.path.exists(markdown_path)
+    # 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.")
 
-    # 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():
     # Sidebar with logout and return buttons