Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,13 +1,13 @@
-import os
 import asyncio
 import gradio as gr
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 from deployer.revenue_tracker import package_artifacts
 
-async def generate_app_async(idea: str):
+async def _run_pipeline(idea: str):
     """
-    Runs the voice-to-app pipeline, packages the app into a ZIP, and returns status and ZIP path.
+    Run the voice-to-app pipeline, package artifacts into a ZIP,
+    and return (status_message, zip_path).
     """
     creator = VoiceToAppCreator(idea)
     assets = creator.run_pipeline()
@@ -16,30 +16,31 @@ async def generate_app_async(idea: str):
     status = f"✅ Generated app: {title}"
     return status, zip_path
 
-
 def deploy_callback(idea: str):
     """
-    Synchronous wrapper for Gradio: generate app and get ZIP.
+    Gradio callback: generate the app and return status + ZIP path.
     """
-    return asyncio.run(generate_app_async(idea))
-
+    return asyncio.run(_run_pipeline(idea))
 
 def robot_behavior(command: str) -> str:
     """
-    Simulate robot action based on a text command.
+    Gradio callback: simulate robot action from text command.
     """
     return VirtualRobot().perform_action(command)
 
-
-def main():
+def launch_gradio_app():
     """
-    Builds and launches the Gradio UI for RoboSage.
+    Build and return the Gradio Blocks UI:
+      - Generate & Download App (left)
+      - Robot Simulator        (right)
     """
     demo = gr.Blocks()
+
     with demo:
         gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
 
         with gr.Row():
+            # — Generate & Download App —
             with gr.Column():
                 gr.Markdown("## 1️⃣ Generate & Download App")
                 idea_input = gr.Textbox(
@@ -47,9 +48,9 @@ def main():
                     placeholder="e.g. A friendly greeting robot.",
                     lines=2
                 )
-                gen_btn = gr.Button("Generate App")
+                gen_btn   = gr.Button("Generate App")
                 status_out = gr.Textbox(label="Status", interactive=False)
-                zip_out = gr.File(label="Download App ZIP")
+                zip_out    = gr.File(label="Download App ZIP")
 
                 gen_btn.click(
                     fn=deploy_callback,
@@ -57,10 +58,11 @@ def main():
                     outputs=[status_out, zip_out]
                 )
 
+            # — Robot Simulator —
             with gr.Column():
                 gr.Markdown("## 🤖 Robot Simulator")
                 cmd_input = gr.Textbox(
-                    label="Command",
+                    label="Speak or Type Command",
                     placeholder="hello or say You rock!",
                     lines=1
                 )
@@ -73,11 +75,4 @@ def main():
                     outputs=[sim_out]
                 )
 
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=int(os.environ.get("PORT", "7860")),
-        share=False
-    )
-
-if __name__ == "__main__":
-    main()
+    return demo