Update app.py

app.py

@@ -24,6 +24,10 @@ def health_check():
 
 
 
+from datetime import datetime
+import tempfile
+import uuid
+
 @app.post("/upload-zip")
 async def upload_zip(file: UploadFile = File(...)):
     if not file.filename.endswith(".zip"):
@@ -34,32 +38,42 @@ async def upload_zip(file: UploadFile = File(...)):
     with open(temp_zip_path, "wb") as f:
         f.write(await file.read())
 
+    # Create a unique subfolder name inside 'demo/'
+    timestamp = datetime.utcnow().strftime("%Y%m%d_%H%M%S")
+    unique_id = uuid.uuid4().hex[:6]
+    folder_name = f"upload_{timestamp}_{unique_id}"
+    hf_folder_prefix = f"demo/{folder_name}"
+
     try:
         with tempfile.TemporaryDirectory() as extract_dir:
-            # Extract all contents of the zip file
+            # Extract zip
             with zipfile.ZipFile(temp_zip_path, 'r') as zip_ref:
                 zip_ref.extractall(extract_dir)
 
             uploaded_files = []
 
-            # Walk through the extracted files and upload
+            # Upload all extracted files
             for root_dir, _, files in os.walk(extract_dir):
                 for name in files:
                     file_path = os.path.join(root_dir, name)
                     relative_path = os.path.relpath(file_path, extract_dir)
-                    repo_path = f"demo/{relative_path}".replace("\\", "/")  # handle Windows paths
+                    repo_path = f"{hf_folder_prefix}/{relative_path}".replace("\\", "/")
 
                     upload_file(
                         path_or_fileobj=file_path,
                         path_in_repo=repo_path,
                         repo_id="rahul7star/ohamlab",
                         repo_type="model",
-                        commit_message=f"Upload {relative_path} from zip",
+                        commit_message=f"Upload {relative_path} to {folder_name}",
                         token=True,
                     )
                     uploaded_files.append(repo_path)
 
-        return {"message": f"✅ Uploaded {len(uploaded_files)} files", "files": uploaded_files}
+        return {
+            "message": f"✅ Uploaded {len(uploaded_files)} files",
+            "folder": folder_name,
+            "files": uploaded_files,
+        }
 
     except Exception as e:
         return {"error": f"❌ Failed to process zip: {str(e)}"}