Update app.py

app.py

@@ -194,12 +194,12 @@ def workspace_page():
 def project_view_page():
     # Sidebar with logout and return buttons
     st.sidebar.title(f"Project: {st.session_state.current_project}")
-    if st.sidebar.button("Back to Workspace"):
-        st.session_state.page = "workspace"
-    if st.sidebar.button("Log Out"):
+        if st.sidebar.button("Log Out"):
         st.session_state.authenticated = False
         st.session_state.username = None
         st.session_state.page = "login"
+    if st.sidebar.button("Back to Workspace"):
+        st.session_state.page = "workspace"
 
     # Display project file structure
     st.subheader(f"Project: {st.session_state.current_project}")
@@ -207,13 +207,22 @@ def project_view_page():
     project_folder = os.path.join(user_folder, st.session_state.current_project)
 
     st.write("File structure:")
+
     for root, dirs, files in os.walk(project_folder):
         level = root.replace(project_folder, "").count(os.sep)
         indent = " " * 4 * level
-        st.text(f"{indent}{os.path.basename(root)}/")
-        sub_indent = " " * 4 * (level + 1)
-        for file in files:
-            st.text(f"{sub_indent}{file}")
+
+        # Display the folder name
+        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)
+
+                # List files in the folder
+                for file in files:
+                    st.write(f"{sub_indent}📄 {file}")
+
 
 
 if __name__ == "__main__":