Update helpers.py

helpers.py

@@ -1,9 +1,10 @@
-import io, zipfile
+import zipfile
+import io
 
-def save_as_zip(bundle):
-    mem = io.BytesIO()
-    z = zipfile.ZipFile(mem, "w")
-    z.writestr("app_bundle.txt", bundle["full_code"])
-    z.close()
-    mem.seek(0)
-    return mem.getvalue()
+def save_as_zip(bundle: dict) -> bytes:
+    zip_buffer = io.BytesIO()
+    with zipfile.ZipFile(zip_buffer, "w") as zipf:
+        for filename, content in bundle.get("files", {}).items():
+            zipf.writestr(filename, content)
+    zip_buffer.seek(0)
+    return zip_buffer.read()