Update app.py

app.py

@@ -93,6 +93,7 @@ def login_page():
             else:
                 st.error("Please fill in all fields.")
 
+
 def workspace_page():
     # Sidebar with logout button
     st.sidebar.title(f"Hello, {st.session_state.username}!")
@@ -101,6 +102,16 @@ def workspace_page():
         st.session_state.username = None
         st.session_state.page = "login"
 
+    # Display "Projects" dropdown
+    user_folder = os.path.join("user_projects", st.session_state.username)
+    os.makedirs(user_folder, exist_ok=True)
+    projects = [d for d in os.listdir(user_folder) if os.path.isdir(os.path.join(user_folder, d))]
+    selected_project = st.sidebar.selectbox("Projects", ["Select a project"] + projects)
+
+    if selected_project != "Select a project":
+        st.session_state.current_project = selected_project
+        st.session_state.page = "project_view"
+
     # Main content area
     st.subheader("Workspace")
     st.write("You can create a new project by uploading files or folders, or by cloning a GitHub repository.")
@@ -118,12 +129,12 @@ def workspace_page():
 
         if uploaded_files and project_name:
             if st.button("Upload Project"):
-                user_folder = os.path.join("user_projects", st.session_state.username, project_name)
-                os.makedirs(user_folder, exist_ok=True)
+                project_folder = os.path.join(user_folder, project_name)
+                os.makedirs(project_folder, exist_ok=True)
 
                 for uploaded_file in uploaded_files:
                     # Save uploaded .zip files or regular files
-                    file_path = os.path.join(user_folder, uploaded_file.name)
+                    file_path = os.path.join(project_folder, uploaded_file.name)
 
                     with open(file_path, "wb") as f:
                         f.write(uploaded_file.getbuffer())
@@ -132,13 +143,14 @@ def workspace_page():
                     if uploaded_file.name.endswith(".zip"):
                         try:
                             with zipfile.ZipFile(file_path, "r") as zip_ref:
-                                zip_ref.extractall(user_folder)
+                                zip_ref.extractall(project_folder)
                             os.remove(file_path)  # Remove the .zip file after extraction
                             st.success(f"Folder from {uploaded_file.name} extracted successfully!")
                         except zipfile.BadZipFile:
                             st.error(f"File {uploaded_file.name} is not a valid .zip file.")
                     else:
                         st.success(f"File {uploaded_file.name} saved successfully!")
+                st.success(f"Project '{project_name}' uploaded successfully!")
 
     elif action == "Clone GitHub Repository":
         st.subheader("Clone GitHub Repository")
@@ -146,15 +158,40 @@ def workspace_page():
 
         if repo_url and project_name:
             if st.button("Upload Project"):
-                user_folder = os.path.join("user_projects", st.session_state.username, project_name)
-                os.makedirs(user_folder, exist_ok=True)
+                project_folder = os.path.join(user_folder, project_name)
+                os.makedirs(project_folder, exist_ok=True)
 
                 try:
-                    Repo.clone_from(repo_url, user_folder)
+                    Repo.clone_from(repo_url, project_folder)
                     st.success(f"Project '{project_name}' cloned successfully!")
                 except Exception as e:
                     st.error(f"Failed to clone repository: {e}")
 
+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"
+
+    # Display project file structure
+    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:")
+    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}")
+
+
 if __name__ == "__main__":
     main()