Update app.py

app.py

@@ -426,6 +426,7 @@ def process_gemini_output(output):
 
 def generate_documentation_page():
     import time
+
     # 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}")
@@ -476,11 +477,13 @@ def generate_documentation_page():
                     # Show progress bar
                     progress_bar = st.progress(0)
                     progress_message = st.empty()
-                    for i in range(101):
-                        time.sleep(estimated_time / 100)  # Simulate the process
+
+                    for i in range(90):  # Simulate the process up to 90% completion
+                        time.sleep(estimated_time / 100)
                         progress_bar.progress(i)
-                        progress_message.write(f"Estimated time remaining: {int(estimated_time * (100 - i) / 100)} seconds")
-                    progress_message.empty()
+                        progress_message.write(
+                            f"Estimated time remaining: {int(estimated_time * (100 - i) / 100)} seconds"
+                        )
 
                     # Generate documentation using Gemini
                     documentation = generate_detailed_documentation(file_contents, functionality)
@@ -489,12 +492,15 @@ def generate_documentation_page():
                     st.session_state.generated_documentation = documentation
 
                     # Display the final documentation
+                    progress_bar.progress(100)  # Fill the progress bar to 100%
+                    progress_message.empty()  # Clear the message
+
+                    time.sleep(1)  # Pause briefly for visual effect
+                    progress_bar.empty()  # Remove the progress bar completely
+
                     st.success("Documentation generated successfully!")
                     st.text_area("Generated Documentation", documentation, height=600)
 
-                    # Remove the progress bar once the output is ready
-                    progress_bar.empty()
-
                 except Exception as e:
                     st.error(f"An error occurred: {e}")
             else:
@@ -548,6 +554,31 @@ def generate_documentation_page():
 
 
 
+# Helper function to generate PDF
+def generate_pdf(documentation, pdf_path):
+    pdf = FPDF()
+    pdf.set_auto_page_break(auto=True, margin=15)
+    pdf.add_page()
+    pdf.set_font("Courier", size=12)  # Maintain the IDE-like font style
+
+    # Add headers and content
+    for line in documentation.splitlines():
+        try:
+            if line.startswith("- '") or line.startswith("File:"):
+                pdf.set_font("Courier", style="B", size=12)  # Bold for specific lines
+                pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+            elif line.startswith("Project Summary:") or line.startswith("Functionality Summary:") or \
+               line.startswith("Functionality Flow:") or line.startswith("Function Documentation:"):
+                pdf.set_font("Courier", style="B", size=14)  # Bold larger headers
+                pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+            else:
+                pdf.set_font("Courier", size=12)  # Regular for other lines
+                pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+        except UnicodeEncodeError as e:
+            pdf.multi_cell(0, 10, line.encode('latin-1', 'replace').decode('latin-1'))
+
+    # Save the PDF
+    pdf.output(pdf_path)
 
 
 # Helper function to generate Markdown file