Update app.py

app.py

@@ -11,6 +11,7 @@ import requests
 from fpdf import FPDF
 import markdown
 import tempfile
+import time
 
 
 # Database setup
@@ -422,6 +423,7 @@ def process_gemini_output(output):
 
     return "\n".join(processed_lines)
 
+
 def generate_documentation_page():
     # Sidebar with "Log Out" and "Back to Project" buttons
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
@@ -461,6 +463,20 @@ def generate_documentation_page():
                     # Read file contents
                     file_contents = read_files(file_paths)
 
+                    # Calculate the total lines of code
+                    total_lines = sum(content.count("\n") for content in file_contents.values())
+                    
+                    # Estimate time remaining
+                    lines_per_minute = 2500
+                    estimated_time = total_lines / lines_per_minute
+                    st.info(f"Estimated time remaining: {int(estimated_time)} minute(s)")
+
+                    # Simulate real-time progress (optional for UX)
+                    progress_bar = st.progress(0)
+                    for i in range(int(estimated_time * 10)):  # Simulate 10 steps per minute
+                        time.sleep(6)  # Update every 6 seconds
+                        progress_bar.progress((i + 1) / (estimated_time * 10))
+
                     # Generate documentation using Gemini
                     documentation = generate_detailed_documentation(file_contents, functionality)
 
@@ -534,32 +550,6 @@ def generate_documentation_page():
                 st.write("Click 'Generate Markdown File' to be able to download and store the documentation as a markdown file")
 
 
-# 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)
-