Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -6,39 +6,40 @@ from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 from deployer.revenue_tracker import package_artifacts
 
-# Asynchronous pipeline runner
+# --- Async pipeline runner ---
 async def _run_pipeline_async(idea: str) -> str:
     creator = VoiceToAppCreator(idea)
     assets = creator.run_pipeline()
+    # blueprint title fallback
     return assets.get("blueprint", {}).get("title", "<unknown>")
 
-# Synchronous wrapper for Gradio
+# --- Sync wrappers for Gradio ---
 def generate_app(idea: str) -> str:
     title = asyncio.run(_run_pipeline_async(idea))
     return f"✅ Generated app: {title}"
 
-# Package the last-generated artifacts into a ZIP and return its path
 def get_app_zip() -> str:
     zip_path = package_artifacts()
     if not os.path.isfile(zip_path):
-        raise FileNotFoundError(f"ZIP not found at {zip_path}")
+        raise FileNotFoundError(f"Could not find ZIP at {zip_path}")
     return zip_path
 
-# Simulated robot behavior endpoint
 def robot_behavior(command: str) -> str:
     return VirtualRobot().perform_action(command)
 
-# Build and launch the Gradio interface
+# --- Build & launch Gradio UI ---
 def launch_gradio_app():
     css = r"""
     .gradio-container { max-width: 800px; margin: auto; }
-    .section { margin-bottom: 2em; }
+    .gradio-box { padding: 1em; margin-bottom: 2em; border: 1px solid #e2e2e2; border-radius: 8px; }
     """
 
     with gr.Blocks(css=css) as demo:
-        gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
+        gr.Markdown("# 🚀 RoboSage")
+        gr.Markdown("Generate your custom robot app and test it live.")
 
-        with gr.Box(elem_id="generate-app", variant="panel"):
+        # Generate App section
+        with gr.Box(elem_id="generate-app", css_class="gradio-box"):
             gr.Markdown("## 1️⃣ Generate App")
             idea_input = gr.Textbox(label="Your Robot Idea", placeholder="e.g. A friendly greeting robot", lines=1)
             gen_btn     = gr.Button("Generate App")
@@ -46,14 +47,16 @@ def launch_gradio_app():
 
             gen_btn.click(fn=generate_app, inputs=[idea_input], outputs=[status_out])
 
-        with gr.Box(elem_id="download-app", variant="panel"):
+        # Download ZIP section
+        with gr.Box(elem_id="download-app", css_class="gradio-box"):
             gr.Markdown("## 📦 Download Packaged App")
             dl_btn  = gr.Button("Download App ZIP")
             dl_file = gr.File(label="App ZIP File")
 
             dl_btn.click(fn=get_app_zip, inputs=None, outputs=[dl_file])
 
-        with gr.Box(elem_id="robot-sim", variant="panel"):
+        # Simulator section
+        with gr.Box(elem_id="robot-sim", css_class="gradio-box"):
             gr.Markdown("## 2️⃣ Robot Simulator")
             cmd_input  = gr.Textbox(label="Command", placeholder="hello or say You rock!", lines=1)
             sim_btn    = gr.Button("Send")
@@ -69,6 +72,6 @@ def launch_gradio_app():
 
     return demo
 
-# Expose for import
-robot_behavior   = robot_behavior
-launch_gradio_app = launch_gradio_app
+# Expose for external import
+robot_behavior     = robot_behavior
+launch_gradio_app  = launch_gradio_app