Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,52 +1,58 @@
 import gradio as gr
 from deployer.simulator_interface import VirtualRobot
 
-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!")
+class RoboSageApp:
+    def __init__(self):
+        self.bot = VirtualRobot()
+        self.interface = self._create_interface()
 
-def create_interface():
-    """Create the Gradio interface components"""
-    with gr.Blocks() as demo:
-        gr.Markdown("# 🤖 RoboSage App\n\nYour robot, your voice.")
+    def _robot_behavior(self, user_input: str):
+        """Handle robot interaction logic"""
+        text = user_input.lower().strip()
         
-        with gr.Row():
-            input_text = gr.Textbox(label="Speak or Type", lines=1)
-            send_button = gr.Button("Send", variant="primary")
-        
-        output_text = gr.Textbox(label="Robot Response", interactive=False)
-        
-        # Event handlers
-        send_button.click(
-            fn=robot_behavior,
-            inputs=input_text,
-            outputs=output_text
-        )
+        if any(g in text for g in ["hello", "hi", "hey", "welcome"]):
+            return self.bot.perform_action("wave") + "\n" + self.bot.perform_action("say Hello there!")
+        if text.startswith("say"):
+            return self.bot.perform_action(text)
+        return self.bot.perform_action("say I didn't understand that. Try again!")
+
+    def _create_interface(self):
+        """Create the Gradio interface with proper component binding"""
+        with gr.Blocks() as demo:
+            gr.Markdown("# 🤖 RoboSage App\n\nYour robot, your voice.")
+            
+            with gr.Row():
+                self.input_text = gr.Textbox(label="Speak or Type", lines=1)
+                self.send_button = gr.Button("Send", variant="primary")
+            
+            self.output_text = gr.Textbox(label="Robot Response", interactive=False)
+            
+            # Event handlers using bound methods
+            self.send_button.click(
+                fn=self._robot_behavior,
+                inputs=self.input_text,
+                outputs=self.output_text
+            )
+            
+            self.input_text.submit(
+                fn=self._robot_behavior,
+                inputs=self.input_text,
+                outputs=self.output_text
+            )
         
-        input_text.submit(
-            fn=robot_behavior,
-            inputs=input_text,
-            outputs=output_text
-        )
-    
-    return demo
+        return demo
 
-def launch_gradio_app(title="RoboSage App", description="Your robot, your voice."):
-    """Launch the Gradio application"""
-    interface = create_interface()
-    return interface
+def launch_gradio_app():
+    """Launch the application with proper component lifecycle management"""
+    app = RoboSageApp()
+    return app.interface
 
 if __name__ == "__main__":
-    app = launch_gradio_app()
-    app.launch(
+    # For local testing with explicit configuration
+    interface = launch_gradio_app()
+    interface.launch(
         server_name="0.0.0.0",
         server_port=7860,
-        share=False
+        share=False,
+        ssr_mode=False  # Explicitly disable SSR if not needed
     )