Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -1,116 +1,39 @@
 import gradio as gr
-from .simulator_interface import VirtualRobot  # Relative import
+from .simulator_interface import VirtualRobot
 import logging
 
 class RoboSageApp:
-    """Main application class for Gradio interface."""
-    
     def __init__(self):
-        """Initialize application components."""
         self.robot = VirtualRobot()
-        self._setup_logging()
-        self.interface = self._build_interface()
-    
-    def _setup_logging(self):
-        """Configure application logging."""
-        logging.basicConfig(
-            level=logging.INFO,
-            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
-        )
         self.logger = logging.getLogger(__name__)
-    
+        self.interface = self._build_interface()
+
     def _build_interface(self) -> gr.Blocks:
-        """Construct the Gradio interface."""
-        with gr.Blocks(
-            title="RoboSage Assistant",
-            css="""
-                .gradio-container {max-width: 600px !important}
-                .dark {background: #f9f9f9}
-            """
-        ) as interface:
+        with gr.Blocks(title="RoboSage") as interface:
+            gr.Markdown("# 🤖 RoboSage Assistant")
             
-            # Header section
-            gr.Markdown("""
-            # 🤖 RoboSage
-            ### Your Personal Virtual Assistant
-            """)
-            
-            # Interaction panel
-            with gr.Row():
-                self.input_txt = gr.Textbox(
-                    label="Your Command",
-                    placeholder="Try 'wave' or 'say hello'...",
-                    max_lines=3,
-                    interactive=True
-                )
-                
             with gr.Row():
-                self.send_btn = gr.Button(
-                    "Submit",
-                    variant="primary",
-                    size="lg"
-                )
-                self.clear_btn = gr.Button(
-                    "Clear",
-                    variant="secondary"
-                )
-            
-            # Output section
-            self.output_txt = gr.Textbox(
-                label="Robot Response",
-                interactive=False,
-                lines=5,
-                show_copy_button=True
-            )
+                input_txt = gr.Textbox(label="Command", placeholder="Try 'wave' or 'say hello'")
+                btn = gr.Button("Submit", variant="primary")
             
-            # Event handlers
-            self.send_btn.click(
-                fn=self._process_command,
-                inputs=self.input_txt,
-                outputs=self.output_txt
-            )
+            output_txt = gr.Textbox(label="Response", interactive=False)
             
-            self.input_txt.submit(
-                fn=self._process_command,
-                inputs=self.input_txt,
-                outputs=self.output_txt
-            )
+            def process_command(cmd: str) -> str:
+                self.logger.info(f"Processing: {cmd}")
+                return self.robot.perform_action(cmd)
             
-            self.clear_btn.click(
-                fn=lambda: ("", ""),
-                inputs=None,
-                outputs=[self.input_txt, self.output_txt]
-            )
+            btn.click(process_command, inputs=input_txt, outputs=output_txt)
+            input_txt.submit(process_command, inputs=input_txt, outputs=output_txt)
         
         return interface
-    
-    def _process_command(self, command: str) -> str:
-        """Handle user commands with logging."""
-        self.logger.info("Processing command: %s", command)
-        try:
-            response = self.robot.perform_action(command)
-            self.logger.info("Generated response: %s", response)
-            return response
-        except Exception as e:
-            self.logger.error("Command processing failed: %s", str(e))
-            return f"⚠️ System error: {str(e)}"
 
-def launch_app() -> gr.Blocks:
-    """Initialize and return the application interface."""
-    try:
-        app = RoboSageApp()
-        return app.interface
-    except Exception as e:
-        logging.critical("Application failed to initialize: %s", str(e))
-        raise
+# The exported function name must match what's imported
+def launch_gradio_app() -> gr.Blocks:
+    """The function that app.py imports must have this exact name"""
+    return RoboSageApp().interface
 
+# For local testing
 if __name__ == "__main__":
-    # Local development configuration
-    interface = launch_app()
-    interface.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        favicon_path=None,
-        debug=False
-    )
+    logging.basicConfig(level=logging.INFO)
+    app = launch_gradio_app()
+    app.launch()