diff --git a/scripts/debug_tool/tabs/edge_net_tab.py b/scripts/debug_tool/tabs/edge_net_tab.py index b626af893633a2d6634ed6fc183bd442d4b95802..dc888bed9a226847193e7eecfc4546fc2c1e37dc 100644 --- a/scripts/debug_tool/tabs/edge_net_tab.py +++ b/scripts/debug_tool/tabs/edge_net_tab.py @@ -270,7 +270,9 @@ class EdgeNetTab(QWidget): def on_include_top_changed(self, state): """Handle include top checkbox change""" - self.include_top = state == Qt.CheckState.Checked + # In PyQt6, CheckState.Checked has value 2 + self.include_top = (state == 2) # or state == Qt.CheckState.Checked + self.update_status(f"Include top changed to: {self.include_top}") def verify_inputs(self): """Verify input files exist""" @@ -319,10 +321,22 @@ class EdgeNetTab(QWidget): def _build_infer_command(self): """Build infer360.py command""" - include_top = "--include_top y" if self.include_top else "" - return (f'wsl bash -c "source {self.config_reader.config["wslAnacondaDir"]}/activate' - f' {self.config_reader.config["edgeNetEnv"]} && ' - f'python infer360.py Input enhanced_depth_e.png material.png rgb.png Input {include_top}"') + # Debug print to verify include_top state + self.update_status(f"Current include_top state: {self.include_top}") + + base_cmd = (f'wsl bash -c "source {self.config_reader.config["wslAnacondaDir"]}/activate' + f' {self.config_reader.config["edgeNetEnv"]} && ' + f'python infer360.py Input enhanced_depth_e.png material.png rgb.png Input') + + if self.include_top: + command = base_cmd + ' --include_top y"' + else: + command = base_cmd + '"' + + # Log final command + self.update_status(f"Final command: {command}") + + return command def on_edge_net_complete(self, success, error_message): """Handle EdgeNet completion"""