Update app.py

app.py

@@ -638,27 +638,6 @@ def saved_documentation_page():
                     mime="text/markdown",
                 )
 
-
-
-
-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
-        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 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}")
-
-
 def project_view_page():
     # Sidebar with logout and return buttons
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
@@ -680,26 +659,16 @@ 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)
 
-    # 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")
+    # Count number of files
+    num_files = sum([len(files) for _, _, files in os.walk(project_folder)])
 
-    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 in `origin`
+    # Count total lines of code in all text files
     num_lines = 0
-    for root, _, files in os.walk(origin_dir):
+    for root, _, files in os.walk(project_folder):
         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
@@ -725,18 +694,28 @@ def project_view_page():
         if st.button("Show File Structure"):
             st.session_state.show_file_structure = not st.session_state.show_file_structure
 
-    # Display tree structure for `origin`
+    # Display file structure if toggled
     if "show_file_structure" not in st.session_state:
         st.session_state.show_file_structure = False
 
     if st.session_state.show_file_structure:
+        def generate_tree(folder_path, prefix=""):
+            """Generates a tree-like structure for the given folder."""
+            items = sorted(os.listdir(folder_path))
+            tree = ""
+            for index, item in enumerate(items):
+                item_path = os.path.join(folder_path, item)
+                connector = "└── " if index == len(items) - 1 else "├── "
+                tree += f"{prefix}{connector}{item}\n"
+                if os.path.isdir(item_path):
+                    subtree_prefix = prefix + ("    " if index == len(items) - 1 else "│   ")
+                    tree += generate_tree(item_path, subtree_prefix)
+            return tree
+
+        # Display the tree structure
         st.write("File structure:")
-        st.write(".")  # Root representation
-        display_tree_structure(origin_dir)
-
-
-
-
+        tree_structure = f".\n{generate_tree(project_folder)}"
+        st.text(tree_structure)
 
 if __name__ == "__main__":
     main()