Infer model backend from selected model path

Replace the hardcoded 'pytorch' model_type with a value inferred from the selected model file. main_window.py now reads the model_path into a local variable and uses DLCLiveProcessor.get_model_backend(model_path) when building DLCProcessorSettings. dlc_processor.py imports Engine from dlclive and adds a static get_model_backend method that returns Engine.from_model_path(model_path).value. Preserves existing config fields and error handling.

Author: C-Achard

dlclivegui/gui/main_window.py

@@ -875,14 +875,15 @@ def _parse_json(self, value: str) -> dict:
         return json.loads(text)
 
     def _dlc_settings_from_ui(self) -> DLCProcessorSettings:
+        model_path = self.model_path_edit.text().strip()
         return DLCProcessorSettings(
-            model_path=self.model_path_edit.text().strip(),
+            model_path=model_path,
             model_directory=self._config.dlc.model_directory,  # Preserve from config
             device=self._config.dlc.device,  # Preserve from config
             dynamic=self._config.dlc.dynamic,  # Preserve from config
             resize=self._config.dlc.resize,  # Preserve from config
             precision=self._config.dlc.precision,  # Preserve from config
-            model_type="pytorch",  # FIXME @C-Achard hardcoded for now, we should allow tf models too
+            model_type=DLCLiveProcessor.get_model_backend(model_path),
             # additional_options=self._parse_json(self.additional_options_edit.toPlainText()),
         )
 

dlclivegui/services/dlc_processor.py

@@ -26,7 +26,10 @@
 ENABLE_PROFILING = True
 
 try:  # pragma: no cover - optional dependency
-    from dlclive import DLCLive  # type: ignore
+    from dlclive import (
+        DLCLive,  # type: ignore
+        Engine,  # type: ignore
+    )
 except Exception as e:  # pragma: no cover - handled gracefully
     logger.error(f"dlclive package could not be imported: {e}")
     DLCLive = None  # type: ignore[assignment]
@@ -96,6 +99,10 @@ def __init__(self) -> None:
         self._gpu_inference_times: deque[float] = deque(maxlen=60)
         self._processor_overhead_times: deque[float] = deque(maxlen=60)
 
+    @staticmethod
+    def get_model_backend(model_path: str) -> str:
+        return Engine.from_model_path(model_path).value
+
     def configure(self, settings: DLCProcessorSettings, processor: Any | None = None) -> None:
         self._settings = settings
         self._processor = processor