Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,64 +1,71 @@
 import os
-import shutil
 import tempfile
+import shutil
 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):
     """
-    Synchronous wrapper for Gradio: runs the pipeline, packages the generated app
-    into a ZIP, and returns (status_message, zip_file_path).
+    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.
     """
     try:
-        # 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))
+        # Sanitize and build a unique directory name
+        title = idea.strip().lower().replace(" ", "_") or "app"
+        base_dir = os.path.join(tempfile.gettempdir(), f"robosage_{title}")
 
-        # Get the generated project directory from assets
-        project_dir = assets.get("project_dir")
-        title = assets.get("blueprint", {}).get("title", "<unknown>")
+        # Clean out any old generation
+        shutil.rmtree(base_dir, ignore_errors=True)
+        os.makedirs(base_dir, exist_ok=True)
 
-        if not project_dir or not os.path.isdir(project_dir):
-            return f"❌ No generated directory found for: {idea}", None
+        # 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}}'"
+""")
 
-        # 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)
+        # 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.")
 
-        zip_path = os.path.join(tmp_root, "app.zip")
+        # 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(dst_dir):
+            for root, _, files in os.walk(base_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))
+                    fullpath = os.path.join(root, fname)
+                    arcname = os.path.relpath(fullpath, base_dir)
+                    zf.write(fullpath, arcname)
 
-        return f"✅ Generated app: {title}", zip_path
+        return f"✅ Generated app: {idea}", zip_path
 
     except Exception as e:
-        return f"❌ Failed to generate app for '{idea}': {e}", None
+        return f"❌ Error: {e}", None
+
 
-def robot_behavior(user_input: str) -> str:
+def robot_behavior(cmd: str) -> str:
     """
-    Hook into the VirtualRobot simulator.
+    Simulate a robot response. You can swap this out for your
+    core_creator or simulator_interface logic if desired.
     """
-    return VirtualRobot().perform_action(user_input)
+    # 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()