Update app.py

app.py

@@ -425,8 +425,6 @@ 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}")
@@ -449,14 +447,12 @@ def generate_documentation_page():
     # Prompt user for functionality description
     functionality = st.text_area(
         "Describe the functionality",
-        placeholder="e.g., Explain the function of the file `main.py`",
+        placeholder="e.g., Displaying output to the screen",
     )
 
     # Button to start analyzing functionality
     if st.button("Analyze"):
         if functionality.strip():
-            st.write("Analyzing project files... Please wait.")
-
             # 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)
@@ -471,19 +467,27 @@ def generate_documentation_page():
 
                     # Calculate estimated 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 = (total_lines / 2500) * 60  # 1 minute per 2500 lines
                     estimated_time = max(estimated_time, 1)  # Minimum 1 second for short tasks
 
-                    # Show progress bar
+                    # Show progress bar and tracker
                     progress_bar = st.progress(0)
                     progress_message = st.empty()
 
-                    for i in range(90):  # Simulate the process up to 90% completion
-                        time.sleep(estimated_time / 100)
-                        progress_bar.progress(i)
+                    # Progress simulation
+                    start_time = time.time()
+                    for i in range(101):
+                        elapsed_time = time.time() - start_time
+                        progress = min(i, int((elapsed_time / estimated_time) * 100))
+                        progress_bar.progress(progress)
+                        time_remaining = max(estimated_time - elapsed_time, 0)
                         progress_message.write(
-                            f"Estimated time remaining: {int(estimated_time * (100 - i) / 100)} seconds"
+                            f"Estimated time remaining: {int(time_remaining)} seconds"
                         )
+                        time.sleep(estimated_time / 100)
+
+                        if elapsed_time >= estimated_time:
+                            break
 
                     # Generate documentation using Gemini
                     documentation = generate_detailed_documentation(file_contents, functionality)
@@ -493,11 +497,10 @@ def generate_documentation_page():
 
                     # Display the final documentation
                     progress_bar.progress(100)  # Fill the progress bar to 100%
+                    time.sleep(1)  # Brief pause
+                    progress_bar.empty()  # Remove the progress bar
                     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)
 
@@ -554,6 +557,7 @@ def generate_documentation_page():
 
 
 
+
 # Helper function to generate PDF
 def generate_pdf(documentation, pdf_path):
     pdf = FPDF()