Update app.py

app.py

@@ -3,54 +3,45 @@ import gradio as gr
 from deployer.gradio_generator import deploy_callback, robot_behavior
 
 def main():
-    with gr.Blocks() as demo:
+    css = """
+    .gradio-container { max-width: 900px; margin: auto; }
+    .section { padding: 1rem; }
+    """
+    with gr.Blocks(css=css) as demo:
         gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app, download it, then test it live.")
 
-        with gr.Row():
-            # ────────────────────────────────
-            #  1️⃣ Generate & Download App
-            # ────────────────────────────────
-            with gr.Column():
-                gr.Markdown("## 1️⃣ Generate & Download App")
-                idea_input = gr.Textbox(
-                    lines=2,
-                    label="Your Robot Idea",
-                    placeholder="e.g. A friendly greeting robot."
-                )
-                gen_button = gr.Button("Generate App & ZIP")
-                status_output = gr.Textbox(label="Status", interactive=False)
-                zip_output = gr.File(label="Download App ZIP")
+        with gr.Column(elem_id="generate-section"):
+            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_button.click(
-                    fn=deploy_callback,
-                    inputs=[idea_input],
-                    outputs=[status_output, zip_output],
-                )
+            gen_btn.click(
+                fn=deploy_callback,
+                inputs=[idea_input],
+                outputs=[status_out, zip_out],
+            )
 
-            # ────────────────────────────────
-            #  2️⃣ Robot Simulator
-            # ────────────────────────────────
-            with gr.Column():
-                gr.Markdown("## 2️⃣ Robot Simulator")
-                cmd_input = gr.Textbox(
-                    lines=1,
-                    label="Command",
-                    placeholder="say 'hello' or 'You rock!'"
-                )
-                sim_button = gr.Button("Send Command")
-                sim_output = gr.Textbox(label="Robot Response", interactive=False)
+        with gr.Column(elem_id="simulate-section"):
+            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_button.click(
-                    fn=robot_behavior,
-                    inputs=[cmd_input],
-                    outputs=[sim_output],
-                )
+            sim_btn.click(
+                fn=robot_behavior,
+                inputs=[cmd_input],
+                outputs=[sim_out],
+            )
 
-        # Launch on 0.0.0.0 for Spaces, default port or override via PORT env
+        # Launch in Hugging Face or locally
         demo.launch(
             server_name="0.0.0.0",
             server_port=int(os.environ.get("PORT", 7860)),
-            share=False
+            share=False,
         )
 
 if __name__ == "__main__":