workspace path setting for videos

Author: Amit0617

interactive_demo_onnx.py

@@ -3,6 +3,7 @@
 import signal
 import sys
 from argparse import ArgumentParser
+from hashlib import sha1
 from pathlib import Path
 
 if "QT_QPA_PLATFORM_PLUGIN_PATH" not in os.environ:
@@ -103,6 +104,39 @@ def resolve_runtime_path(raw_path: str) -> str:
     return raw_path
 
 
+def default_workspace_root() -> Path:
+    return Path.home() / "scutie"
+
+
+def sha1_file(path: Path, chunk_size: int = 1024 * 1024) -> str:
+    digest = sha1()
+    with path.open("rb") as handle:
+        while True:
+            chunk = handle.read(chunk_size)
+            if not chunk:
+                break
+            digest.update(chunk)
+    return digest.hexdigest()
+
+
+def resolve_workspace(args) -> str | None:
+    if args.workspace is not None:
+        return args.workspace
+
+    workspace_root = default_workspace_root()
+    workspace_root.mkdir(parents=True, exist_ok=True)
+
+    if args.video is not None:
+        video_path = Path(args.video).expanduser().resolve()
+        return str(workspace_root / sha1_file(video_path))
+
+    if args.images is not None:
+        images_path = Path(args.images).expanduser().resolve()
+        return str(workspace_root / images_path.name)
+
+    return None
+
+
 if __name__ in "__main__":
     args = get_arguments()
 
@@ -133,6 +167,9 @@ def resolve_runtime_path(raw_path: str) -> str:
 
     args.device = resolve_device(args.device)
     log.info(f"Using ONNX Runtime device: {args.device}")
+    args.video = resolve_runtime_path(args.video) if args.video is not None else None
+    args.images = resolve_runtime_path(args.images) if args.images is not None else None
+    args.workspace = resolve_workspace(args)
 
     args_dict = vars(args)
     for key in ("onnx_encoder", "onnx_memory_write", "onnx_read_decode", "ritm_onnx"):