Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,51 +1,43 @@
 import os
+import re
 import shutil
-import tempfile
 import zipfile
 import asyncio
+import tempfile
+from core_creator.app_blueprint import build_app_code
+from core_creator.voice_to_app import robot_action
 
-from core_creator.voice_to_app import VoiceToAppCreator
-from deployer.simulator_interface import VirtualRobot
-
-def run_pipeline(idea: str) -> dict:
-    """
-    Synchronous wrapper to run the voice-to-app pipeline and return
-    the assets dict, which must include a 'blueprint' with a 'title',
-    and the path to the generated project directory.
-    """
-    creator = VoiceToAppCreator(idea)
-    assets = creator.run_pipeline()
-    return assets
-
-def deploy_callback(idea: str):
+async def deploy_callback(idea: str):
     """
-    Called by Gradio. Runs pipeline, packages into a ZIP,
-    and returns a status message + path to the ZIP.
+    Generate the app directory, package into zip, return status and path.
     """
-    try:
-        assets = run_pipeline(idea)
-        title = assets.get("blueprint", {}).get("title", None) or "unknown"
-        gen_dir = assets.get("project_dir", None)
-        if not gen_dir or not os.path.isdir(gen_dir):
-            return f"❌ No generated directory found for: {idea}", None
+    # Normalize name
+    safe_name = re.sub(r"[^a-z0-9]+", "_", idea.lower()).strip("_") or "app"
+    with tempfile.TemporaryDirectory() as tmpdir:
+        app_dir = os.path.join(tmpdir, f"robosage_{safe_name}")
+        try:
+            # Build code files
+            os.makedirs(app_dir, exist_ok=False)
+            build_app_code(idea, app_dir)
+        except Exception as e:
+            return f"❌ Failed to generate app directory for: {idea}", None
 
-        # Create ZIP in temporary file
-        tmp_fd, tmp_path = tempfile.mkstemp(suffix=".zip")
-        os.close(tmp_fd)
-        with zipfile.ZipFile(tmp_path, "w", zipfile.ZIP_DEFLATED) as zf:
-            for root, _, files in os.walk(gen_dir):
-                for fname in files:
-                    abs_path = os.path.join(root, fname)
-                    rel_path = os.path.relpath(abs_path, gen_dir)
-                    zf.write(abs_path, arcname=rel_path)
+        # Create zip
+        zip_path = os.path.join(tmpdir, f"robosage_{safe_name}.zip")
+        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+            for root, _, files in os.walk(app_dir):
+                for file in files:
+                    full = os.path.join(root, file)
+                    arc = os.path.relpath(full, app_dir)
+                    zf.write(full, arc)
 
-        return f"✅ Generated app: {title}", tmp_path
+        return f"✅ Generated app: {idea}", zip_path
 
-    except Exception as e:
-        return f"❌ Failed to generate app for: {idea} ({e})", None
-
-def robot_behavior(user_input: str) -> str:
+def robot_behavior(command: str):
     """
-    Simulate the robot action via VirtualRobot.
+    Simulate robot behavior: map command to action.
     """
-    return VirtualRobot().perform_action(user_input)
+    try:
+        return robot_action(command)
+    except Exception:
+        return f"❓ Unknown action: {command}"