Update app.py

app.py

@@ -3,45 +3,56 @@ import gradio as gr
 from deployer.gradio_generator import deploy_callback, robot_behavior
 
 def main():
-    css = """
+    # CSS to constrain width and add spacing
+    custom_css = """
     .gradio-container { max-width: 900px; margin: auto; }
-    .section { padding: 1rem; }
+    .section { padding: 1rem 0; }
     """
-    with gr.Blocks(css=css) as demo:
-        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app, download it, then test it live.")
 
-        with gr.Column(elem_id="generate-section"):
+    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)
+            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],
+                inputs=idea_input,
+                outputs=[status_out, zip_out]
             )
 
-        with gr.Column(elem_id="simulate-section"):
+        # 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)
+            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],
+                inputs=cmd_input,
+                outputs=sim_out
             )
 
-        # Launch in Hugging Face or locally
+        # Launch settings
         demo.launch(
             server_name="0.0.0.0",
-            server_port=int(os.environ.get("PORT", 7860)),
-            share=False,
+            server_port=int(os.getenv("PORT", 7860)),
+            share=False
         )
 
 if __name__ == "__main__":