Update app.py

app.py

@@ -2,57 +2,58 @@ import os
 import gradio as gr
 from deployer.gradio_generator import deploy_callback, robot_behavior
 
-def main():
-    # CSS to constrain width and add spacing
-    custom_css = """
-    .gradio-container { max-width: 900px; margin: auto; }
-    .section { padding: 1rem 0; }
+def generate_app(idea: str):
+    """
+    Gradio callback: returns (status_message, path_to_zip).
+    Gradio's File component will serve the zip.
     """
+    return deploy_callback(idea)
 
-    with gr.Blocks(css=custom_css) as demo:
-        gr.Markdown("# 🚀 RoboSage")
-        gr.Markdown("Generate your custom robot app, download it, then test it live.")
-
-        # Section 1️⃣ Generate & Download
-        with gr.Group(elem_id="section_generate"):
-            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=deploy_callback,
-                inputs=idea_input,
-                outputs=[status_out, zip_out]
-            )
-
-        # Section 2️⃣ Simulator
-        with gr.Group(elem_id="section_simulator"):
-            gr.Markdown("## 2️⃣ Robot Simulator")
-            cmd_input = gr.Textbox(
-                label="Command",
-                placeholder="say 'hello' or 'You rock!'",
-                lines=1
-            )
-            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
-            )
-
-        # Launch settings
+def main():
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🚀 RoboSage\nGenerate & download a simple robot app, then test it live.")
+
+        with gr.Row():
+            # ▶️ Generate & Download App
+            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],
+                )
+
+            # ▶️ Robot Simulator
+            with gr.Column():
+                gr.Markdown("## 2️⃣ Robot Simulator")
+                cmd_input = gr.Textbox(
+                    label="Command",
+                    placeholder="say 'hello' or 'You rock!'",
+                    lines=1,
+                )
+                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],
+                )
+
+        # Launch on Hugging Face Spaces or locally
         demo.launch(
             server_name="0.0.0.0",
-            server_port=int(os.getenv("PORT", 7860)),
-            share=False
+            server_port=int(os.environ.get("PORT", 7860)),
+            share=False,
         )
 
 if __name__ == "__main__":