Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,32 +1,50 @@
 import gradio as gr
 from deployer.simulator_interface import VirtualRobot
 
-# Robot behavior logic
 def robot_behavior(user_input: str):
+    """Handle robot interaction logic"""
     bot = VirtualRobot()
     text = user_input.lower().strip()
+    
     if any(g in text for g in ["hello", "hi", "hey", "welcome"]):
         return bot.perform_action("wave") + "\n" + bot.perform_action("say Hello there!")
     if text.startswith("say"):
         return bot.perform_action(text)
     return bot.perform_action("say I didn't understand that. Try again!")
 
-# Build Gradio app with correct Blocks context
 def launch_gradio_app(title="RoboSage App", description="Your robot, your voice."):
+    """Create and configure the Gradio interface"""
     with gr.Blocks() as demo:
         gr.Markdown(f"# 🤖 {title}\n\n{description}")
-
-        # Define components without trailing commas
-        inp = gr.Textbox(label="Speak or Type", lines=1)
-        out = gr.Textbox(label="Robot Response")
-        btn = gr.Button("Send", key="send-btn")
-
-        # Wire the click event inside the same Blocks context
-        btn.click(fn=robot_behavior, inputs=[inp], outputs=[out])
+        
+        # Input/Output components
+        with gr.Row():
+            inp = gr.Textbox(label="Speak or Type", lines=1)
+            btn = gr.Button("Send", variant="primary")
+        
+        out = gr.Textbox(label="Robot Response", interactive=False)
+        
+        # Event handling
+        btn.click(
+            fn=robot_behavior,
+            inputs=inp,
+            outputs=out,
+            api_name="process_command"
+        )
+        
+        # Additional UX improvements
+        inp.submit(
+            fn=robot_behavior,
+            inputs=inp,
+            outputs=out
+        )
 
     return demo
 
-# For local testing
 if __name__ == "__main__":
     app = launch_gradio_app()
-    app.launch()
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=False
+    )