Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -7,7 +7,7 @@ from deployer.revenue_tracker import package_artifacts
 
 async def generate_app_async(idea: str):
     """
-    Runs the voice-to-app pipeline and packages the app, returning status and ZIP path.
+    Run the voice-to-app pipeline asynchronously and return a tuple of (status_message, zip_path).
     """
     creator = VoiceToAppCreator(idea)
     assets = creator.run_pipeline()
@@ -19,27 +19,28 @@ async def generate_app_async(idea: str):
 
 def deploy_callback(idea: str):
     """
-    Synchronous wrapper for Gradio.
+    Synchronous wrapper for Gradio: generates the app and packages it into a ZIP.
+    Returns (status_message, zip_path).
     """
     return asyncio.run(generate_app_async(idea))
 
 
 def robot_behavior(command: str) -> str:
     """
-    Simulate robot action based on a text command.
+    Simulate robot behavior based on a text command.
     """
     return VirtualRobot().perform_action(command)
 
 
-def launch_gradio_app():
+def main():
     """
-    Build and return the Gradio interface.
+    Build and launch the Gradio UI for RoboSage.
     """
     with gr.Blocks() as demo:
         gr.Markdown("# 🚀 RoboSage\nGenerate your custom robot app and test it live.")
 
         with gr.Row():
-            # Generate & Download
+            # Generate & Download App Section
             with gr.Column():
                 gr.Markdown("## 1️⃣ Generate & Download App")
                 idea_input = gr.Textbox(
@@ -57,7 +58,7 @@ def launch_gradio_app():
                     outputs=[status_out, zip_out]
                 )
 
-            # Simulator
+            # Robot Simulator Section
             with gr.Column():
                 gr.Markdown("## 🤖 Robot Simulator")
                 cmd_input = gr.Textbox(
@@ -74,13 +75,11 @@ def launch_gradio_app():
                     outputs=[sim_out]
                 )
 
-    return demo
-
-
-if __name__ == "__main__":
-    demo = launch_gradio_app()
     demo.launch(
         server_name="0.0.0.0",
         server_port=int(os.environ.get("PORT", 7860)),
         share=False
     )
+
+if __name__ == "__main__":
+    main()