Update app.py

app.py

@@ -74,7 +74,6 @@ except Exception:
     logging.exception("Bootstrap failed, UI will still load so you can see errors")
 
 # === model choices (restricted to Supervisely RT-DETRv2) ======================
-# Exact mapping to configs and reference COCO checkpoints you provided
 MODEL_CHOICES = [
     ("rtdetrv2_s",   "S (r18vd, 120e) — default"),
     ("rtdetrv2_m",   "M (r34vd, 120e)"),
@@ -115,7 +114,7 @@ _ROBO_URL_RX = re.compile(r"""
    |
     (?P<ws2>[A-Za-z0-9\-_]+)/(?P<proj2>[A-Za-z0-9\-_]+)(?:/(?:v)?(?P<ver2>\d+))?
  )$
-""", re.IGNORECASE | re.VERBOSE)
+""", re.VERBOSE | re.IGNORECASE)
 
 def parse_roboflow_url(s: str):
     s = s.strip()
@@ -441,10 +440,7 @@ def _set_first_existing_key_deep(cfg: dict, keys: list, value):
 
 def _install_supervisely_logger_shim():
     """
-    Create a *package* shim that shadows site-packages:
-      supervisely/nn/training/__init__.py  -> exposes train_logger
-    Ensures 'from supervisely.nn.training import train_logger' works even if the
-    installed 'supervisely' no longer provides it.
+    Create a package shim so 'from supervisely.nn.training import train_logger' works.
     """
     root = pathlib.Path(tempfile.gettempdir()) / "sly_shim_pkg"
     pkg_training = root / "supervisely" / "nn" / "training"
@@ -456,7 +452,7 @@ def _install_supervisely_logger_shim():
         if not init_file.exists():
             init_file.write_text("")
 
-    # Our shim __init__.py provides train_logger
+    # Expose train_logger from the package's __init__
     (pkg_training / "__init__.py").write_text(textwrap.dedent("""
         # Minimal shim for backward-compat with older RT-DETRv2 training code.
         class _TrainLogger:
@@ -467,14 +463,9 @@ def _install_supervisely_logger_shim():
             def log_image(self, *a, **k): pass
         train_logger = _TrainLogger()
     """))
-    # Return path to prepend to PYTHONPATH
     return str(root)
 
 def _ensure_checkpoint(model_key: str, out_dir: str) -> str | None:
-    """
-    Download the reference COCO checkpoint for the selected model if not present.
-    Returns local path (or None if not available).
-    """
     url = CKPT_URLS.get(model_key)
     if not url:
         return None
@@ -500,6 +491,29 @@ def _ensure_checkpoint(model_key: str, out_dir: str) -> str | None:
             pass
         return None
 
+def _absify_base_includes(cfg: dict, base_cfg_path: str):
+    """
+    If the upstream template uses relative 'base' includes (e.g. '../dataset/coco_detection.yml'),
+    rewrite them to absolute paths so loading your generated config outside the repo tree works.
+    """
+    if not isinstance(cfg, dict):
+        return
+    repo_base_dir = os.path.dirname(base_cfg_path)
+    for key in ("base", "_base_", "BASE", "BASE_YAML"):
+        if key in cfg:
+            val = cfg[key]
+            if isinstance(val, str):
+                if not os.path.isabs(val):
+                    cfg[key] = os.path.abspath(os.path.join(repo_base_dir, val))
+            elif isinstance(val, list):
+                new_list = []
+                for item in val:
+                    if isinstance(item, str) and not os.path.isabs(item):
+                        new_list.append(os.path.abspath(os.path.join(repo_base_dir, item)))
+                    else:
+                        new_list.append(item)
+                cfg[key] = new_list
+
 def patch_base_config(base_cfg_path, merged_dir, class_count, run_name,
                       epochs, batch, imgsz, lr, optimizer, pretrained_path: str | None):
     if not base_cfg_path or not os.path.exists(base_cfg_path):
@@ -508,6 +522,9 @@ def patch_base_config(base_cfg_path, merged_dir, class_count, run_name,
     with open(base_cfg_path, "r") as f:
         cfg = yaml.safe_load(f)
 
+    # <<< Fix: ensure 'base' includes are absolute, not relative to generated_configs >>>
+    _absify_base_includes(cfg, base_cfg_path)
+
     ann_dir = os.path.join(merged_dir, "annotations")
     paths = {
         "train_json": os.path.abspath(os.path.join(ann_dir, "instances_train.json")),
@@ -970,7 +987,6 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="sky")) as app:
 
 if __name__ == "__main__":
     os.environ.setdefault("YOLO_CONFIG_DIR", "/tmp/Ultralytics")  # silence stray warnings from other libs
-    # Log training script resolution at startup for quick troubleshooting
     try:
         ts = find_training_script(REPO_DIR)
         logging.info(f"Startup check — training script at: {ts}")