Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,75 +1,42 @@
-### deployer/gradio_generator.py
 import os
 import shutil
-tempfile
+import tempfile
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 
 def deploy_callback(idea: str):
     """
-    Generate robot app from idea and package into a ZIP.
-    Returns (status_message, zip_path)
+    Generates the robot app for the given idea, zips it, and returns
+    (status_message, path_to_zip_file or None).
     """
     try:
+        # Run your pipeline
         creator = VoiceToAppCreator(idea)
         assets = creator.run_pipeline()
-        title = assets.get("blueprint", {}).get("title", "<unknown>")
+
+        # Extract output directory & title
         app_dir = assets.get("app_dir")
+        title = assets.get("blueprint", {}).get("title", "<unknown>")
+
         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"❌ Error: {str(e)}", None
+            return f"❌ Failed to generate app directory for: {idea}", None
 
+        # Create a temp ZIP file (will persist until process exits)
+        tmp = tempfile.NamedTemporaryFile(delete=False, suffix=".zip")
+        tmp_path = tmp.name
+        tmp.close()
 
-def robot_behavior(command: str) -> str:
-    """
-    Simulate robot behavior based on input command.
-    """
-    try:
-        return VirtualRobot().perform_action(command)
-    except Exception as e:
-        return f"❌ Simulator error: {str(e)}"
+        # Make archive: shutil wants base name without extension
+        base = tmp_path[:-4]
+        shutil.make_archive(base, 'zip', root_dir=app_dir)
 
+        return f"✅ Generated app: {title}", tmp_path
 
-### app.py
-import os
-import gradio as gr
-from deployer.gradio_generator import deploy_callback, robot_behavior
+    except Exception as e:
+        return f"❌ Error generating app: {e}", None
 
-def generate_app(idea: str):
+def robot_behavior(user_input: str) -> str:
     """
-    Gradio interface callback to generate app and return status and ZIP file.
+    Simulate robot behavior.
     """
-    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)
+    return VirtualRobot().perform_action(user_input)