Update app.py

app.py

@@ -1,44 +1,35 @@
-# app.py
-
 import os
-import asyncio
+import shutil
+from deployer.gradio_generator import deploy_and_package, robot_behavior
 import gradio as gr
-from deployer.gradio_generator import deploy_callback, robot_behavior
-
-async def _deploy_async(idea: str):
-    """
-    Helper to call deploy_callback in an async context.
-    """
-    return deploy_callback(idea)
 
 def generate_app(idea: str):
     """
-    Gradio callback: run the pipeline, package the app, and return (status, zip_path).
+    Runs the pipeline, packages the app into a zip, and returns
+    (status_message, zip_file_path) for Gradio to serve.
     """
-    status, zip_path = asyncio.run(_deploy_async(idea))
+    status, zip_path = deploy_and_package(idea)
+    # Gradio File component needs a Path or file-like
     return status, zip_path
 
 def main():
-    """
-    Build and launch the Gradio UI:
-    - Left column: Generate & Download App
-    - Right column: Robot Simulator
-    """
-    with gr.Blocks() as demo:
-        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
+    with gr.Blocks(css="""
+        .gradio-container { max-width: 900px; margin: auto; }
+        .section { padding: 1rem; }
+    """) as demo:
+        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and download it, then test it live.")
 
         with gr.Row():
-            # ————————— Generate & Download App —————————
-            with gr.Column():
-                gr.Markdown("## 1️⃣ Generate & Download App")
+            with gr.Column(elem_id="generate-section"):
+                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")
+                gen_btn = gr.Button("Generate App & ZIP")
                 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=generate_app,
@@ -46,16 +37,15 @@ def main():
                     outputs=[status_out, zip_out]
                 )
 
-            # ————————— Robot Simulator —————————
-            with gr.Column():
-                gr.Markdown("## 🤖 Robot Simulator")
+            with gr.Column(elem_id="simulate-section"):
+                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,
@@ -63,11 +53,11 @@ def main():
                     outputs=[sim_out]
                 )
 
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=int(os.environ.get("PORT", "7860")),
-        share=False
-    )
+        demo.launch(
+            server_name="0.0.0.0",
+            server_port=int(os.environ.get("PORT", 7860)),
+            share=False
+        )
 
 if __name__ == "__main__":
     main()