Update app.py

app.py

@@ -640,20 +640,21 @@ def saved_documentation_page():
 
 
 
-def display_folder_contents(folder_path, prefix=""):
-    """Recursively lists the contents of a folder in a bash 'ls -R'-like style."""
+
+def display_tree_structure(folder_path, prefix=""):
+    """Recursively display the folder structure in a tree-like format."""
     try:
         items = sorted(os.listdir(folder_path))  # Sort items for consistent display
-        st.write(f"{prefix}{os.path.basename(folder_path)}/")  # Show the folder name
-
-        for item in items:
+        for i, item in enumerate(items):
             item_path = os.path.join(folder_path, item)
+            is_last_item = i == len(items) - 1
+            connector = "└── " if is_last_item else "├── "
+            st.write(f"{prefix}{connector}{item}")
+
             if os.path.isdir(item_path):
-                # Recursively display sub-folders
-                display_folder_contents(item_path, prefix=prefix + "    ")
-            else:
-                # Display files with an indented structure
-                st.write(f"{prefix}    {item}")
+                # Recursively display sub-folders with additional indentation
+                new_prefix = prefix + ("    " if is_last_item else "│   ")
+                display_tree_structure(item_path, new_prefix)
     except Exception as e:
         st.error(f"Error displaying folder contents: {e}")
 
@@ -679,16 +680,26 @@ def project_view_page():
     user_folder = os.path.join("user_projects", st.session_state.username)
     project_folder = os.path.join(user_folder, st.session_state.current_project)
 
-    # Count number of files
-    num_files = sum([len(files) for _, _, files in os.walk(project_folder)])
+    # Check for `.git` directory and focus on `origin` folder
+    git_dir = os.path.join(project_folder, ".git")
+    origin_dir = os.path.join(git_dir, "refs", "remotes", "origin")
+
+    if not os.path.exists(origin_dir):
+        st.error("The 'origin' folder does not exist in this project.")
+        return
+
+    # Count number of files in `origin`
+    num_files = sum(
+        [len(files) for _, _, files in os.walk(origin_dir)]
+    )
 
-    # Count total lines of code in all text files
+    # Count total lines of code in all text files in `origin`
     num_lines = 0
-    for root, _, files in os.walk(project_folder):
+    for root, _, files in os.walk(origin_dir):
         for file in files:
             file_path = os.path.join(root, file)
             try:
-                with open(file_path, 'r', encoding='utf-8') as f:
+                with open(file_path, "r", encoding="utf-8") as f:
                     num_lines += sum(1 for _ in f)
             except (UnicodeDecodeError, IsADirectoryError):
                 continue
@@ -714,13 +725,14 @@ def project_view_page():
         if st.button("Show File Structure"):
             st.session_state.show_file_structure = not st.session_state.show_file_structure
 
-    # Display file structure if toggled
+    # Display tree structure for `origin`
     if "show_file_structure" not in st.session_state:
         st.session_state.show_file_structure = False
 
     if st.session_state.show_file_structure:
         st.write("File structure:")
-        display_folder_contents(project_folder)
+        st.write(".")  # Root representation
+        display_tree_structure(origin_dir)