Update app.py

app.py

@@ -212,7 +212,9 @@ def read_project_files(project_path):
     file_paths = []
     for root, _, files in os.walk(project_path):
         for file in files:
-            file_paths.append(os.path.join(root, file))
+            # Skip .git files or folders from GitHub clones
+            if ".git" not in root:
+                file_paths.append(os.path.join(root, file))
     return file_paths
 
 def read_files(file_paths):
@@ -261,13 +263,13 @@ def identify_required_functions(project_path, functionality_description):
 
 def generate_documentation_page():
     st.subheader(f"Generate Documentation for {st.session_state.current_project}")
-    st.write("Enter the functionality or parts of the project for which you'd like to generate documentation.")
+    st.write("Enter the functionality or parts of the project for which you'd like to identify relevant functions.")
 
     # Prompt user for functionality description
-    functionality = st.text_area("Describe the functionality for documentation generation", placeholder="e.g., Explain the function of the file `main.py`")
+    functionality = st.text_area("Describe the functionality", placeholder="e.g., Explain the function of the file `main.py`")
 
-    # Button to start generating documentation
-    if st.button("Generate"):
+    # Button to start analyzing functionality
+    if st.button("Analyze"):
         if functionality.strip():
             st.write("Analyzing project files... Please wait.")
             
@@ -275,15 +277,24 @@ def generate_documentation_page():
             user_folder = os.path.join("user_projects", st.session_state.username)
             project_folder = os.path.join(user_folder, st.session_state.current_project)
             
-            # Call the function to identify required functions
-            try:
-                result = identify_required_functions(project_folder, functionality)
-                st.success("Documentation generated successfully!")
-                st.text_area("Generated Documentation", result, height=400)
-            except Exception as e:
-                st.error(f"An error occurred: {e}")
+            # Ensure compatibility with GitHub repositories
+            if os.path.exists(project_folder):
+                try:
+                    # Call the function to identify relevant functions
+                    result = identify_required_functions(project_folder, functionality)
+
+                    # Check if Gemini returned an error message
+                    if "No relevant functions found" in result:
+                        st.error("No relevant functions were found for the specified functionality. Please refine your input.")
+                    else:
+                        st.success("Relevant functions identified successfully!")
+                        st.text_area("Relevant Functions", result, height=400)
+                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 generate documentation.")
+            st.error("Please enter the functionality to analyze.")
 
     # Button to navigate back to the project page
     if st.button("Back to Project"):
@@ -291,6 +302,7 @@ def generate_documentation_page():
         st.rerun()
 
 
+
 #------------------------------------------------------------------------------------------------------------------------------------------------------------------------