Update app.py

app.py

@@ -645,7 +645,7 @@ def project_view_page():
     st.sidebar.image("SimplifAI Logo Long.jpeg", use_container_width=True)
     st.sidebar.title(f"Project: {st.session_state.current_project}")
     if st.sidebar.button("Back to Project Staging"):
-        st.session_state.page = "project_staging"  # Corrected navigation
+        st.session_state.page = "project_staging"
         st.rerun()
     if st.sidebar.button("Log Out"):
         st.session_state.authenticated = False
@@ -673,7 +673,7 @@ def project_view_page():
                 with open(file_path, 'r', encoding='utf-8') as f:
                     num_lines += sum(1 for _ in f)
             except (UnicodeDecodeError, IsADirectoryError):
-                continue  # Skip binary or non-text files
+                continue
 
     # Display metrics side by side
     col1, col2 = st.columns(2)
@@ -702,20 +702,30 @@ def project_view_page():
 
     if st.session_state.show_file_structure:
         st.write("File structure:")
-        for root, dirs, files in os.walk(project_folder):
-            level = root.replace(project_folder, "").count(os.sep)
-            indent = " " * 4 * level
 
-            # Display folders and files recursively
-            if level == 0:
-                st.write(f"📂 {os.path.basename(root)}")
-            else:
-                with st.expander(f"{indent}📂 {os.path.basename(root)}"):
-                    sub_indent = " " * 4 * (level + 1)
-                    for folder in dirs:
-                        st.write(f"{sub_indent}📂 {folder}")
-                    for file in files:
-                        st.write(f"{sub_indent}📄 {file}")
+        def display_folder_contents(path, level=0):
+            """
+            Recursively display folder contents using Streamlit expanders.
+
+            Args:
+                path (str): The folder path.
+                level (int): The current depth of the folder in the hierarchy.
+            """
+            # Get subfolders and files
+            items = sorted(os.listdir(path))
+            for item in items:
+                item_path = os.path.join(path, item)
+                if os.path.isdir(item_path):
+                    # Display folder as an expander
+                    with st.expander(f"{' ' * 4 * level}📂 {item}", expanded=False):
+                        display_folder_contents(item_path, level + 1)
+                else:
+                    # Display file
+                    st.write(f"{' ' * 4 * (level + 1)}📄 {item}")
+
+        # Start displaying from the project folder root
+        display_folder_contents(project_folder)
+