Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,47 +1,47 @@
 import os
-import shutil
 import tempfile
 import zipfile
-
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 
-def deploy_and_package(idea: str):
+def deploy_callback(idea: str):
     """
-    1) Runs the VoiceToAppCreator pipeline to generate the app files.
-    2) Packages the generated app directory into a zip under /tmp.
-    Returns:
-        status_message (str), zip_file_path (str)
+    1) Generate the app assets via VoiceToAppCreator
+    2) Package them into a ZIP under /tmp (tempdir)
+    Returns (status_message, zip_file_path_or_None)
     """
-    # 1) Generate assets
-    creator = VoiceToAppCreator(idea)
-    assets = creator.run_pipeline()
-    title = assets.get("blueprint", {}).get("title", "robot_app")
-    generated_dir = assets.get("output_dir")
-    if not generated_dir or not os.path.isdir(generated_dir):
-        return f"❌ Failed to generate app for: {idea}", None
+    # --- 1) Run pipeline ---
+    try:
+        creator = VoiceToAppCreator(idea)
+        assets = creator.run_pipeline()  # should return {"blueprint": {...}, "files": {filename:content_str}}
+        title = assets.get("blueprint", {}).get("title", "<unknown>")
+    except Exception as e:
+        return f"❌ Failed to generate app: {e}", None
+
+    # --- 2) Package into a ZIP ---
+    try:
+        # create a fresh temp directory
+        base_dir = tempfile.mkdtemp(prefix="robosage_")
+        zip_path = os.path.join(base_dir, "app.zip")
 
-    # 2) Create a temporary zip under /tmp
-    zip_name = f"{title.replace(' ', '_')}.zip"
-    zip_path = os.path.join(tempfile.gettempdir(), zip_name)
-    # Clean up any old zip
-    if os.path.exists(zip_path):
-        os.remove(zip_path)
+        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+            for fname, content in assets.get("files", {}).items():
+                # ensure subdirs exist
+                full_path = os.path.join(base_dir, fname)
+                os.makedirs(os.path.dirname(full_path), exist_ok=True)
+                # write the generated file
+                with open(full_path, "w", encoding="utf-8") as f:
+                    f.write(content)
+                # add to zip under its relative name
+                zf.write(full_path, arcname=fname)
 
-    # Walk the generated_dir and add to zip
-    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-        for root, _, files in os.walk(generated_dir):
-            for fname in files:
-                full = os.path.join(root, fname)
-                rel = os.path.relpath(full, generated_dir)
-                zf.write(full, arcname=rel)
+        return f"✅ Generated app: {title}", zip_path
 
-    status = f"✅ Generated app: **{title}**"
-    return status, zip_path
+    except Exception as e:
+        return f"❌ Failed to package app: {e}", None
 
 def robot_behavior(user_input: str) -> str:
     """
-    Wraps the VirtualRobot simulator for live commands.
+    Hook into your simulator.
     """
-    bot = VirtualRobot()
-    return bot.perform_action(user_input)
+    return VirtualRobot().perform_action(user_input)