Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,71 +1,51 @@
 import os
-import tempfile
 import shutil
+import tempfile
 import zipfile
+import asyncio
+
+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):
     """
-    Generates a minimal robot app based on the user's idea, packages it
-    into a ZIP in the system temp directory, and returns a status message
-    along with the path to the ZIP file.
+    Called by Gradio. Runs pipeline, packages into a ZIP,
+    and returns a status message + path to the ZIP.
     """
     try:
-        # Sanitize and build a unique directory name
-        title = idea.strip().lower().replace(" ", "_") or "app"
-        base_dir = os.path.join(tempfile.gettempdir(), f"robosage_{title}")
-
-        # Clean out any old generation
-        shutil.rmtree(base_dir, ignore_errors=True)
-        os.makedirs(base_dir, exist_ok=True)
-
-        # 1) Create main.py with a simple on_command stub
-        main_py = os.path.join(base_dir, "main.py")
-        with open(main_py, "w") as f:
-            f.write(f"""def on_command(cmd):
-    cmd_lower = cmd.strip().lower()
-    if "hello" in cmd_lower:
-        return "🤖 Hello there!"
-    else:
-        return f"🤔 I don't understand: '{{cmd}}'"
-""")
-
-        # 2) Create a README
-        readme = os.path.join(base_dir, "README.md")
-        with open(readme, "w") as f:
-            f.write(f"# RoboSage App\nGenerated for idea: {idea}\n\nRun `main.on_command(...)` to simulate.")
-
-        # 3) Zip the directory
-        zip_name = f"robosage_{title}.zip"
-        zip_path = os.path.join(tempfile.gettempdir(), zip_name)
-        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-            for root, _, files in os.walk(base_dir):
+        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
+
+        # 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:
-                    fullpath = os.path.join(root, fname)
-                    arcname = os.path.relpath(fullpath, base_dir)
-                    zf.write(fullpath, arcname)
+                    abs_path = os.path.join(root, fname)
+                    rel_path = os.path.relpath(abs_path, gen_dir)
+                    zf.write(abs_path, arcname=rel_path)
 
-        return f"✅ Generated app: {idea}", zip_path
+        return f"✅ Generated app: {title}", tmp_path
 
     except Exception as e:
-        return f"❌ Error: {e}", None
+        return f"❌ Failed to generate app for: {idea} ({e})", None
 
-
-def robot_behavior(cmd: str) -> str:
+def robot_behavior(user_input: str) -> str:
     """
-    Simulate a robot response. You can swap this out for your
-    core_creator or simulator_interface logic if desired.
+    Simulate the robot action via VirtualRobot.
     """
-    # Very basic echo-like behavior
-    lower = cmd.strip().lower()
-    if "hello" in lower:
-        return "🤖 *waves* Hello there!"
-    elif lower:
-        return f"🤔 I don't know how to '{cmd}'."
-    else:
-        return "❓ Please say something."
-
-
-# Optional entrypoint if you want to import a launcher function
-def launch_gradio_app():
-    from app import main
-    main()
+    return VirtualRobot().perform_action(user_input)