From 86cdf485e53ba56ecb65cf4580491c199a47a3ea Mon Sep 17 00:00:00 2001
From: Bingsu <ks2515@naver.com>
Date: Wed, 26 Apr 2023 13:42:11 +0900
Subject: [PATCH] feat: install.py

---
 install.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)
 create mode 100644 install.py

diff --git a/install.py b/install.py
new file mode 100644
index 0000000..477998a
--- /dev/null
+++ b/install.py
@@ -0,0 +1,52 @@
+from __future__ import annotations
+
+import importlib.util
+import subprocess
+import sys
+from importlib.metadata import version  # python >= 3.8
+
+from packaging.version import parse
+
+
+def is_installed(package: str, min_version: str | None = None):
+    try:
+        spec = importlib.util.find_spec(package)
+    except ModuleNotFoundError:
+        return False
+
+    if spec is None:
+        return False
+
+    if not min_version:
+        return True
+
+    try:
+        return parse(version(package)) >= parse(min_version)
+    except Exception:
+        return False
+
+
+def run_pip(*args):
+    subprocess.run([sys.executable, "-m", "pip", "install", *args])
+
+
+def install():
+    deps = [
+        ("ultralytics", "8.0.87"),
+        ("mediapipe", "0.9.3.0"),
+        ("huggingface_hub", None),
+        ("pydantic", None),  # pydantic >= 2.0.0 compatible
+    ]
+
+    for name, ver in deps:
+        if not is_installed(name, ver):
+            run_pip("-U", name, "--prefer-binary")
+
+
+try:
+    from launch import skip_install
+except ImportError:
+    skip_install = False
+
+if not skip_install:
+    install()