Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,42 +1,64 @@
 import os
 import shutil
 import tempfile
+import zipfile
+import asyncio
+
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 
+async def _run_pipeline(idea: str) -> dict:
+    """
+    Asynchronously run the voice-to-app pipeline.
+    Returns the assets dict from VoiceToAppCreator.run_pipeline().
+    """
+    creator = VoiceToAppCreator(idea)
+    return creator.run_pipeline()
+
 def deploy_callback(idea: str):
     """
-    Generates the robot app for the given idea, zips it, and returns
-    (status_message, path_to_zip_file or None).
+    Synchronous wrapper for Gradio: runs the pipeline, packages the generated app
+    into a ZIP, and returns (status_message, zip_file_path).
     """
     try:
-        # Run your pipeline
-        creator = VoiceToAppCreator(idea)
-        assets = creator.run_pipeline()
+        # Run pipeline in the existing event loop or new one
+        try:
+            loop = asyncio.get_running_loop()
+        except RuntimeError:
+            loop = None
+
+        if loop and loop.is_running():
+            assets = asyncio.run_coroutine_threadsafe(_run_pipeline(idea), loop).result()
+        else:
+            assets = asyncio.new_event_loop().run_until_complete(_run_pipeline(idea))
 
-        # Extract output directory & title
-        app_dir = assets.get("app_dir")
+        # Get the generated project directory from assets
+        project_dir = assets.get("project_dir")
         title = assets.get("blueprint", {}).get("title", "<unknown>")
 
-        if not app_dir or not os.path.isdir(app_dir):
-            return f"❌ Failed to generate app directory for: {idea}", None
+        if not project_dir or not os.path.isdir(project_dir):
+            return f"❌ No generated directory found for: {idea}", None
 
-        # Create a temp ZIP file (will persist until process exits)
-        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
-        tmp_path = tmp.name
-        tmp.close()
+        # Copy into a temp dir to zip
+        tmp_root = tempfile.mkdtemp(prefix="robosage_")
+        dst_dir = os.path.join(tmp_root, "app")
+        shutil.copytree(project_dir, dst_dir)
 
-        # Make archive: shutil wants base name without extension
-        base = tmp_path[:-4]
-        shutil.make_archive(base, 'zip', root_dir=app_dir)
+        zip_path = os.path.join(tmp_root, "app.zip")
+        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+            for root, _, files in os.walk(dst_dir):
+                for fname in files:
+                    full_path = os.path.join(root, fname)
+                    rel_path = os.path.relpath(full_path, dst_dir)
+                    zf.write(full_path, arcname=os.path.join(title, rel_path))
 
-        return f"✅ Generated app: {title}", tmp_path
+        return f"✅ Generated app: {title}", zip_path
 
     except Exception as e:
-        return f"❌ Error generating app: {e}", None
+        return f"❌ Failed to generate app for '{idea}': {e}", None
 
 def robot_behavior(user_input: str) -> str:
     """
-    Simulate robot behavior.
+    Hook into the VirtualRobot simulator.
     """
     return VirtualRobot().perform_action(user_input)