Update app.py

app.py

@@ -194,36 +194,38 @@ 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"):
         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
+    # Main content for project page
     st.subheader(f"Project: {st.session_state.current_project}")
-    user_folder = os.path.join("user_projects", st.session_state.username)
-    project_folder = os.path.join(user_folder, st.session_state.current_project)
-
-    st.write("File structure:")
+    st.write("Manage your project and explore its files.")
 
-    for root, dirs, files in os.walk(project_folder):
-        level = root.replace(project_folder, "").count(os.sep)
-        indent = " " * 4 * level
+    # Button to display file structure
+    if st.button("Show File Structure"):
+        user_folder = os.path.join("user_projects", st.session_state.username)
+        project_folder = os.path.join(user_folder, st.session_state.current_project)
 
-        # 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)
+        st.write("File structure:")
 
-                # List files in the folder
-                for file in files:
-                    st.write(f"{sub_indent}📄 {file}")
+        for root, dirs, files in os.walk(project_folder):
+            level = root.replace(project_folder, "").count(os.sep)
+            indent = " " * 4 * level
 
+            # 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__":
     main()