Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,54 +1,75 @@
+### deployer/gradio_generator.py
 import os
-import tempfile
-import zipfile
+import shutil
+tempfile
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 
-def _make_app_zip(src_dir: str, title: str) -> str:
-    """
-    Zips up the directory at src_dir into a file named <title>.zip
-    in the system temp directory, and returns the full path.
-    """
-    safe_title = "".join(c if c.isalnum() or c in "-_." else "_" for c in title)
-    zip_filename = f"{safe_title}.zip"
-    tmp_dir = tempfile.gettempdir()
-    zip_path = os.path.join(tmp_dir, zip_filename)
-    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-        for root, _, files in os.walk(src_dir):
-            for fname in files:
-                full_path = os.path.join(root, fname)
-                # store relative path within the zip
-                arcname = os.path.relpath(full_path, src_dir)
-                zf.write(full_path, arcname)
-    return zip_path
-
 def deploy_callback(idea: str):
     """
-    Generates an app from the user idea and returns:
-      (status_message: str, path_to_zip: Optional[str])
+    Generate robot app from idea and package into a ZIP.
+    Returns (status_message, zip_path)
     """
     try:
         creator = VoiceToAppCreator(idea)
         assets = creator.run_pipeline()
-        # Title for status
         title = assets.get("blueprint", {}).get("title", "<unknown>")
-        # Directory where the app was generated
-        project_dir = assets.get("project_dir") or assets.get("app_dir")
-        if project_dir is None or not os.path.isdir(project_dir):
-            # No directory to zip up
-            return f"❌ No generated directory found for '{title}'", None
-
-        zip_path = _make_app_zip(project_dir, title)
+        app_dir = assets.get("app_dir")
+        if not app_dir or not os.path.isdir(app_dir):
+            return f"❌ Failed to generate app: {idea}", None
+        # Create temporary directory for zip
+        tmpdir = tempfile.mkdtemp()
+        zip_base = os.path.join(tmpdir, "app")
+        zip_path = shutil.make_archive(zip_base, 'zip', app_dir)
         return f"✅ Generated app: {title}", zip_path
-
     except Exception as e:
-        return f"❌ Failed to generate app: {e}", None
+        return f"❌ Error: {str(e)}", None
+
 
 def robot_behavior(command: str) -> str:
     """
-    Feeds the user command into the VirtualRobot simulator.
+    Simulate robot behavior based on input command.
     """
     try:
         return VirtualRobot().perform_action(command)
     except Exception as e:
-        return f"❌ Simulator error: {e}"
+        return f"❌ Simulator error: {str(e)}"
+
+
+### app.py
+import os
+import gradio as gr
+from deployer.gradio_generator import deploy_callback, robot_behavior
+
+def generate_app(idea: str):
+    """
+    Gradio interface callback to generate app and return status and ZIP file.
+    """
+    status, zip_path = deploy_callback(idea)
+    return status, zip_path
+
+# Build Gradio UI
+with gr.Blocks(css="""
+.gradio-container { max-width: 900px; margin: auto; }
+.section { padding: 1rem; }
+""") as demo:
+    gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app, download it, then test it live.")
+
+    with gr.Column():
+        gr.Markdown("## 1️⃣ Generate & Download App")
+        idea_input = gr.Textbox(label="Your Robot Idea", placeholder="e.g. A friendly greeting robot.", lines=2)
+        gen_btn = gr.Button("Generate App & ZIP")
+        status_out = gr.Textbox(label="Status", interactive=False)
+        zip_out = gr.File(label="Download App ZIP")
+        gen_btn.click(fn=generate_app, inputs=[idea_input], outputs=[status_out, zip_out])
+
+    with gr.Column():
+        gr.Markdown("## 2️⃣ Robot Simulator")
+        cmd_input = gr.Textbox(label="Command", placeholder="say 'hello' or 'You rock!'")
+        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])
+
+
+if __name__ == "__main__":
+    demo.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), share=False)