Update app.py

app.py

@@ -102,16 +102,18 @@ def workspace_page():
 
     # Main content area
     st.subheader("Workspace")
-    st.write("You can create a new project by uploading files or cloning a GitHub repository.")
+    st.write("You can create a new project by uploading files or folders, or by cloning a GitHub repository.")
 
     # User action selection
-    action = st.radio("Choose an action", ["Upload Files", "Clone GitHub Repository"], horizontal=True)
+    action = st.radio("Choose an action", ["Upload Files or Folders", "Clone GitHub Repository"], horizontal=True)
 
     project_name = st.text_input("Enter a project name")
 
-    if action == "Upload Files":
-        st.subheader("Upload Files")
-        uploaded_files = st.file_uploader("Upload one or more files for your project", accept_multiple_files=True)
+    if action == "Upload Files or Folders":
+        st.subheader("Upload Files or Folders")
+        uploaded_files = st.file_uploader(
+            "Upload one or more files or a .zip archive for folders", accept_multiple_files=True
+        )
 
         if uploaded_files and project_name:
             if st.button("Upload Project"):
@@ -119,10 +121,23 @@ def workspace_page():
                 os.makedirs(user_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)
+
                     with open(file_path, "wb") as f:
                         f.write(uploaded_file.getbuffer())
-                st.success(f"Project '{project_name}' uploaded successfully!")
+
+                    # If a .zip file is uploaded, extract its contents
+                    if uploaded_file.name.endswith(".zip"):
+                        try:
+                            with zipfile.ZipFile(file_path, "r") as zip_ref:
+                                zip_ref.extractall(user_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!")
 
     elif action == "Clone GitHub Repository":
         st.subheader("Clone GitHub Repository")