Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,72 +1,89 @@
-import gradio as gr
+# File: deployer/gradio_generator.py
+import os
 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 app from the user idea, package artifacts into a ZIP,
-    and return a (status_message, zip_path) tuple.
+    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.
     """
-    # 1. Run the core pipeline
-    creator = VoiceToAppCreator(idea)
-    assets = creator.run_pipeline()
-
-    # 2. Extract a title for status display
+    # Create app artifacts
+    assets = VoiceToAppCreator(idea).run_pipeline()
     title = assets.get("blueprint", {}).get("title", "<unknown>")
-
-    # 3. Package everything into app_package.zip
+    # Package into a ZIP and return
     zip_path = package_artifacts(assets)
-
-    # 4. Return status message and ZIP filepath
-    status = f"✅ Generated app: {title}"
-    return status, zip_path
+    return f"✅ Generated app: {title}", zip_path
 
 def robot_behavior(command: str) -> str:
     """
-    Simulate robot action based on a text command.
+    Simulate robot action based on text command.
     """
     return VirtualRobot().perform_action(command)
 
-def launch_gradio_app():
-    """
-    Build and return the Gradio Blocks UI:
-      - Left column: generate & download
-      - Right column: robot simulator
+
+# 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; }
     """
-    demo = gr.Blocks()
 
-    with demo:
+    with gr.Blocks(css=css) as 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_in   = 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")
-
-                # Wire up generation → status + ZIP
-                gen_btn.click(fn=deploy_callback,
-                              inputs=[idea_in],
-                              outputs=[status_out, zip_out])
+            # 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]
+                )
 
-            # — Robot Simulator —
-            with gr.Column():
-                gr.Markdown("## 🤖 Robot Simulator")
-                cmd_in  = gr.Textbox(label="Command",
-                                     placeholder="hello or say You rock!",
-                                     lines=1)
+            # 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", lines=4, interactive=False)
+                sim_out = gr.Textbox(
+                    label="Robot Response",
+                    interactive=False
+                )
+                sim_btn.click(
+                    fn=robot_behavior,
+                    inputs=[cmd_input],
+                    outputs=[sim_out]
+                )
 
-                # Wire up simulator
-                sim_btn.click(fn=robot_behavior,
-                              inputs=[cmd_in],
-                              outputs=[sim_out])
+    demo.launch(
+        server_name="0.0.0.0",
+        server_port=int(os.getenv("PORT", 7860)),
+        share=False
+    )
 
-    return demo
+if __name__ == "__main__":
+    main()