Update app.py

app.py

@@ -640,6 +640,24 @@ 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."""
+    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:
+            item_path = os.path.join(folder_path, 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}")
+    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)
@@ -700,27 +718,9 @@ def project_view_page():
     if "show_file_structure" not in st.session_state:
         st.session_state.show_file_structure = False
 
-        def display_folder_contents(folder_path, prefix=""):
-            """Recursively lists the contents of a folder in a bash 'ls -R'-like style."""
-            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:
-                    item_path = os.path.join(folder_path, 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}")
-            except Exception as e:
-                st.error(f"Error displaying folder contents: {e}")
-
-        # Example usage in your `project_view_page` function:
-        if st.session_state.show_file_structure:
-            st.write("File structure:")
-            display_folder_contents(project_folder)
+    if st.session_state.show_file_structure:
+        st.write("File structure:")
+        display_folder_contents(project_folder)