Update app.py

app.py

@@ -447,67 +447,48 @@ def generate_documentation_page():
     # Prompt user for functionality description
     functionality = st.text_area(
         "Describe the functionality",
-        placeholder="e.g., Displaying output to the screen",
+        placeholder="e.g., Explain the function of the file `main.py`",
     )
 
     # Button to start analyzing functionality
     if st.button("Analyze"):
         if functionality.strip():
-            # 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)
-
-                    # 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 = max(estimated_time, 1)  # Minimum 1 second for short tasks
-
-                    # Show progress bar and tracker
-                    progress_bar = st.progress(0)
-                    progress_message = st.empty()
-
-                    # 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(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)
-
-                    # Save the documentation in session state for exporting and viewing
-                    st.session_state.generated_documentation = documentation
-
-                    # 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
-
-                    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.")
+            # 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.")
         else:
             st.error("Please enter the functionality to analyze.")
 
@@ -558,6 +539,7 @@ def generate_documentation_page():
 
 
 
+
 # Helper function to generate PDF
 def generate_pdf(documentation, pdf_path):
     pdf = FPDF()