Update app.py

app.py

@@ -401,19 +401,21 @@ def generate_documentation_page():
 
             if os.path.exists(project_folder):
                 try:
-                    # Call Gemini to identify required functions
-                    gemini_result = identify_required_functions(project_folder, functionality)
+                    # Gather all file paths in the project directory
+                    file_paths = read_project_files(project_folder)
 
-                    # Display the final documentation
-                    st.success("Documentation generated successfully!")
-                    st.text_area("Generated Documentation", gemini_result, height=600)
+                    # Read file contents
+                    file_contents = read_files(file_paths)
+
+                    # Generate documentation using Gemini
+                    documentation = generate_detailed_documentation(file_contents, functionality)
 
-                    # Buttons to generate PDF and Markdown files
-                    if st.button("Generate PDF"):
-                        generate_pdf(gemini_result)
-                    if st.button("Generate Markdown File"):
-                        generate_markdown_file(gemini_result)
+                    # Save the documentation in session state for exporting
+                    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:
@@ -421,6 +423,40 @@ def generate_documentation_page():
         else:
             st.error("Please enter the functionality to analyze.")
 
+    # Add export buttons if documentation is available
+    if "generated_documentation" in st.session_state and st.session_state.generated_documentation:
+        documentation = st.session_state.generated_documentation
+
+        # Button to generate PDF
+        if st.button("Generate PDF"):
+            pdf = fpdf.FPDF()
+            pdf.set_auto_page_break(auto=True, margin=15)
+            pdf.add_page()
+            pdf.set_font("Arial", size=12)
+
+            # Write documentation to PDF with headers
+            for line in documentation.split("\n"):
+                if line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
+                   line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
+                    pdf.set_font("Arial", style="B", size=14)
+                elif line.startswith("* **") or line.startswith("- **"):
+                    pdf.set_font("Arial", style="B", size=12)
+                else:
+                    pdf.set_font("Arial", size=12)
+                pdf.multi_cell(0, 10, line)
+
+            pdf_file_path = os.path.join(user_folder, f"{st.session_state.current_project}_documentation.pdf")
+            pdf.output(pdf_file_path)
+            st.success(f"PDF file generated at: {pdf_file_path}")
+
+        # Button to generate Markdown file
+        if st.button("Generate Markdown File"):
+            markdown_file_path = os.path.join(user_folder, f"{st.session_state.current_project}_documentation.md")
+            with open(markdown_file_path, "w") as md_file:
+                md_file.write(f"# Documentation for {st.session_state.current_project}\n\n")
+                md_file.write(documentation)
+            st.success(f"Markdown file generated at: {markdown_file_path}")
+
 
 # Helper function to generate PDF
 def generate_pdf(documentation):