Update app.py

app.py

@@ -144,7 +144,7 @@ def parse_roboflow_url(s: str):
             version = None
             if len(p) >= 3:
                 v = p[2]
-                if v.lower().startswith('v') and v[1:].isdigit():
+                if v.lower().startsWith('v') and v[1:].isdigit():
                     version = int(v[1:])
                 elif v.isdigit():
                     version = int(v)
@@ -459,14 +459,16 @@ def _install_supervisely_logger_shim():
     return str(root)
 
 # ---- NEW: kwargs-aware sitecustomize shim (safe, non-invasive) ---------------
-def _install_workspace_shim_v3(cwd_for_train: str, module_default: str = "rtdetrv2_pytorch.src"):
+def _install_workspace_shim_v3(dest_dir: str, module_default: str = "rtdetrv2_pytorch.src"):
     """
     Writes a sitecustomize.py that monkeypatches
     rtdetrv2_pytorch.src.core.workspace.create so it works with the real signature:
         def create(name, **kwargs):
     The shim ensures kwargs['cfg'] is a dict, then guarantees cfg['_pymodule'] is a *module object*.
+    `dest_dir` MUST be on sys.path at interpreter startup (we'll prepend it to PYTHONPATH).
     """
-    sc_path = os.path.join(cwd_for_train, "sitecustomize.py")
+    os.makedirs(dest_dir, exist_ok=True)
+    sc_path = os.path.join(dest_dir, "sitecustomize.py")
 
     # NOTE: Not an f-string. Escape literal braces with {{ }} and only format {module_default}.
     code = textwrap.dedent("""\
@@ -505,7 +507,7 @@ def _install_workspace_shim_v3(cwd_for_train: str, module_default: str = "rtdetr
             def create(name, **kwargs):
                 cfg = kwargs.get("cfg")
                 if not isinstance(cfg, dict):
-                    cfg = {{}} if cfg is None else dict(cfg)
+                    cfg = {} if cfg is None else dict(cfg)
                     kwargs["cfg"] = cfg
                 _ensure_pymodule_object(cfg)
                 return _orig_create(name, **kwargs)
@@ -910,20 +912,33 @@ def training_handler(dataset_path, model_key, run_name, epochs, batch, imgsz, lr
         try:
             train_cwd = os.path.dirname(train_script)
 
-            # Install kwargs-aware sitecustomize shim (safe, non-invasive)
-            _install_workspace_shim_v3(train_cwd, module_default="rtdetrv2_pytorch.src")
+            # --- NEW: create a temp dir for sitecustomize and put it FIRST on PYTHONPATH
+            shim_dir = tempfile.mkdtemp(prefix="rtdetr_site_")
+            _install_workspace_shim_v3(shim_dir, module_default="rtdetrv2_pytorch.src")
 
             env = os.environ.copy()
-            # Make sure repo code can be imported
+
+            # Supervisely logger shim (can be later in path)
+            sly_shim_root = _install_supervisely_logger_shim()
+
+            # Build PYTHONPATH — order matters!
+            #  1) shim_dir (so sitecustomize auto-imports)
+            #  2) train_cwd (belt & suspenders; makes local imports easy)
+            #  3) PY_IMPL_DIR + REPO_DIR (RT-DETRv2 code)
+            #  4) sly_shim_root (optional)
+            #  5) existing PYTHONPATH
             env["PYTHONPATH"] = os.pathsep.join(filter(None, [
-                PY_IMPL_DIR, REPO_DIR, env.get("PYTHONPATH", "")
+                shim_dir,
+                train_cwd,
+                PY_IMPL_DIR,
+                REPO_DIR,
+                sly_shim_root,
+                env.get("PYTHONPATH", "")
             ]))
-            # Put our shim first so supervisely import never breaks
-            shim_root = _install_supervisely_logger_shim()
-            env["PYTHONPATH"] = os.pathsep.join([shim_root, env["PYTHONPATH"]])
+
             env.setdefault("WANDB_DISABLED", "true")
-            # Provide a secondary hint for some config loaders
             env.setdefault("RTDETR_PYMODULE", "rtdetrv2_pytorch.src")
+            env.setdefault("PYTHONUNBUFFERED", "1")  # nicer real-time logs
 
             proc = subprocess.Popen(cmd, cwd=train_cwd,
                                     stdout=subprocess.PIPE, stderr=subprocess.STDOUT,