Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -2,38 +2,80 @@ import gradio as gr
 from .simulator_interface import VirtualRobot
 import logging
 
-class RoboSageApp:
+class RoboSageInterface:
     def __init__(self, title="RoboSage", description="Your virtual assistant"):
         self.robot = VirtualRobot()
         self.title = title
         self.description = description
         self.logger = logging.getLogger(__name__)
-        self.interface = self._build_interface()
+        self._setup_logging()
+        
+        # Initialize components
+        self.input_txt = None
+        self.output_txt = None
+        self.btn = None
+        
+        self.interface = self._create_interface()
+
+    def _setup_logging(self):
+        logging.basicConfig(
+            level=logging.INFO,
+            format='%(asctime)s - %(levelname)s - %(message)s'
+        )
 
-    def _build_interface(self) -> gr.Blocks:
+    def _create_interface(self) -> gr.Blocks:
         with gr.Blocks(title=self.title) as interface:
-            gr.Markdown(f"# 🤖 {self.title}\n\n{self.description}")
-            
-            with gr.Row():
-                input_txt = gr.Textbox(label="Command", placeholder="Try 'wave' or 'say hello'")
-                btn = gr.Button("Submit", variant="primary")
-            
-            output_txt = gr.Textbox(label="Response", interactive=False)
-            
-            def process_command(cmd: str) -> str:
-                self.logger.info(f"Processing: {cmd}")
-                return self.robot.perform_action(cmd)
-            
-            btn.click(process_command, inputs=input_txt, outputs=output_txt)
-            input_txt.submit(process_command, inputs=input_txt, outputs=output_txt)
+            # Store references to components as instance attributes
+            self._create_components()
+            self._setup_event_handlers()
         
         return interface
 
-def launch_gradio_app(title="RoboSage", description="Your virtual assistant") -> gr.Blocks:
-    """Now accepts title and description parameters"""
-    return RoboSageApp(title=title, description=description).interface
+    def _create_components(self):
+        """Create and store all interface components"""
+        gr.Markdown(f"# 🤖 {self.title}\n\n{self.description}")
+        
+        with gr.Row():
+            self.input_txt = gr.Textbox(
+                label="Command",
+                placeholder="Try 'wave' or 'say hello'",
+                elem_id="command_input"
+            )
+            self.btn = gr.Button(
+                "Submit",
+                variant="primary",
+                elem_id="submit_btn"
+            )
+        
+        self.output_txt = gr.Textbox(
+            label="Response",
+            interactive=False,
+            elem_id="response_output"
+        )
 
-if __name__ == "__main__":
-    logging.basicConfig(level=logging.INFO)
-    app = launch_gradio_app()
-    app.launch()
+    def _setup_event_handlers(self):
+        """Configure all event handlers"""
+        def wrapped_process_command(cmd: str) -> str:
+            self.logger.info(f"Processing command: {cmd}")
+            return self.robot.perform_action(cmd)
+        
+        # Reference components through instance attributes
+        self.btn.click(
+            fn=wrapped_process_command,
+            inputs=self.input_txt,
+            outputs=self.output_txt
+        )
+        
+        self.input_txt.submit(
+            fn=wrapped_process_command,
+            inputs=self.input_txt,
+            outputs=self.output_txt
+        )
+
+def launch_gradio_app(title="RoboSage", description="Your virtual assistant") -> gr.Blocks:
+    """Public interface for launching the app"""
+    try:
+        return RoboSageInterface(title, description).interface
+    except Exception as e:
+        logging.critical(f"Failed to create interface: {str(e)}")
+        raise