Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,89 +1,47 @@
-# File: deployer/gradio_generator.py
 import os
+import shutil
+import tempfile
+import zipfile
+
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
-from deployer.revenue_tracker import package_artifacts
 
-def deploy_callback(idea: str):
-    """
-    Generate the robot app from the user's idea and package it as a ZIP.
-    Returns a status message and the path to the ZIP file for download.
+def deploy_and_package(idea: str):
     """
-    # Create app artifacts
-    assets = VoiceToAppCreator(idea).run_pipeline()
-    title = assets.get("blueprint", {}).get("title", "<unknown>")
-    # Package into a ZIP and return
-    zip_path = package_artifacts(assets)
-    return f"✅ Generated app: {title}", zip_path
-
-def robot_behavior(command: str) -> 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)
     """
-    Simulate robot action based on text command.
+    # 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
+
+    # 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)
+
+    # 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)
+
+    status = f"✅ Generated app: **{title}**"
+    return status, zip_path
+
+def robot_behavior(user_input: str) -> str:
     """
-    return VirtualRobot().perform_action(command)
-
-
-# File: app.py
-import os
-import gradio as gr
-from deployer.gradio_generator import deploy_callback, robot_behavior
-
-def main():
-    css = """
-    .gradio-container { max-width: 900px; margin: auto; }
-    .section { padding: 1rem; }
+    Wraps the VirtualRobot simulator for live commands.
     """
-
-    with gr.Blocks(css=css) as demo:
-        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
-
-        with gr.Row():
-            # Section 1: App Generation & Download
-            with gr.Column(scale=1):
-                gr.Markdown("## 1️⃣ Generate & Download App", elem_classes="section")
-                idea_input = gr.Textbox(
-                    label="Your Robot Idea",
-                    placeholder="e.g. A friendly greeting robot.",
-                    lines=2
-                )
-                gen_btn = gr.Button("Generate App")
-                status_out = gr.Textbox(
-                    label="Status",
-                    interactive=False
-                )
-                zip_out = gr.File(
-                    label="Download App ZIP"
-                )
-                gen_btn.click(
-                    fn=deploy_callback,
-                    inputs=[idea_input],
-                    outputs=[status_out, zip_out]
-                )
-
-            # Section 2: Robot Simulator
-            with gr.Column(scale=1):
-                gr.Markdown("## 2️⃣ Robot Simulator", elem_classes="section")
-                cmd_input = gr.Textbox(
-                    label="Command",
-                    placeholder="hello or say You rock!",
-                    lines=1
-                )
-                sim_btn = gr.Button("Send Command")
-                sim_out = gr.Textbox(
-                    label="Robot Response",
-                    interactive=False
-                )
-                sim_btn.click(
-                    fn=robot_behavior,
-                    inputs=[cmd_input],
-                    outputs=[sim_out]
-                )
-
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=int(os.getenv("PORT", 7860)),
-        share=False
-    )
-
-if __name__ == "__main__":
-    main()
+    bot = VirtualRobot()
+    return bot.perform_action(user_input)