Update main.py

main.py

@@ -5,6 +5,7 @@ from pathlib import Path
 import shutil
 import zipfile
 import logging
+import tempfile
 from typing import Set
 
 # Configure logging
@@ -41,10 +42,19 @@ class SiteManager:
     async def deploy_site(self, unique_id: str, zip_file: UploadFile) -> dict:
         """Deploy a new site from a ZIP file"""
         site_path = self.sites_dir / unique_id
+        temp_file = None
         
         try:
-            # Save and extract ZIP
-            with zipfile.ZipFile(zip_file.file) as zip_ref:
+            # Create a temporary file
+            temp_file = tempfile.NamedTemporaryFile(delete=False)
+            # Read the uploaded file content
+            content = await zip_file.read()
+            # Write to the temporary file
+            temp_file.write(content)
+            temp_file.close()
+            
+            # Extract ZIP contents
+            with zipfile.ZipFile(temp_file.name) as zip_ref:
                 # Clear existing site if present
                 if site_path.exists():
                     shutil.rmtree(site_path)
@@ -67,6 +77,14 @@ class SiteManager:
             if site_path.exists():
                 shutil.rmtree(site_path)
             raise HTTPException(status_code=500, detail=str(e))
+        
+        finally:
+            # Clean up the temporary file
+            if temp_file:
+                try:
+                    Path(temp_file.name).unlink()
+                except:
+                    pass
 
     def remove_site(self, unique_id: str) -> bool:
         """Remove a deployed site"""