Create pack_builder/zip_builder.py

pack_builder/zip_builder.py

@@ -0,0 +1,41 @@
+import zipfile
+import uuid
+import os
+from beat_generator.riffusion_cpu import generate_audio_loop
+from content_generator.hooks import generate_hooks
+from content_generator.captions import generate_captions
+from content_generator.ideas import generate_content_ideas
+from content_generator.scripts import generate_short_scripts
+
+def build_creator_pack(niche, style, bpm, tone, goal):
+    folder = f"assets/temp/{uuid.uuid4().hex}"
+    os.makedirs(folder, exist_ok=True)
+
+    # Beats
+    for i in range(3):
+        beat_path = generate_audio_loop(style, bpm)
+        os.system(f"cp {beat_path} {folder}/beat_{i+1}.wav")
+
+    # Text Packs
+    with open(f"{folder}/hooks.txt", "w") as f:
+        f.write("\n".join(generate_hooks(niche, tone)))
+
+    with open(f"{folder}/captions.txt", "w") as f:
+        f.write("\n".join(generate_captions(niche, tone)))
+
+    with open(f"{folder}/ideas.txt", "w") as f:
+        f.write("\n".join(generate_content_ideas(niche, goal)))
+
+    with open(f"{folder}/scripts.txt", "w") as f:
+        f.write("\n".join(generate_short_scripts(niche, tone)))
+
+    # ZIP
+    zip_path = f"{folder}.zip"
+    with zipfile.ZipFile(zip_path, "w") as zipf:
+        for root, _, files in os.walk(folder):
+            for file in files:
+                p = os.path.join(root, file)
+                arc = os.path.relpath(p, folder)
+                zipf.write(p, arc)
+
+    return zip_path