Add requirements.txt

Author: semjon00

.gitignore

@@ -1 +1,2 @@
-__pycache__/
+__pycache__/
+venv/

install.py

@@ -39,7 +39,12 @@ def ensure(module_name, min_version=None):
 ensure('transforms3d', '0.4.1')
 
 ensure('imageio')  # 2.4.1
-ensure('imageio-ffmpeg')
+try:  # Dirty hack to not reinstall every time
+    importlib_metadata.version('imageio-ffmpeg')
+except:
+    ensure('imageio-ffmpeg')
+
+
 if not launch.is_installed("networkx"):
     launch.run_pip('install install "networkx==2.5"', "networkx requirement for depthmap script")
 if platform.system() == 'Windows':

requirements.txt

@@ -0,0 +1,20 @@
+# Requirements for running in standalone mode
+# First, install the corect version of PyTorch!
+# PyTorch Compute Platform must match the configuration of the hardware.
+
+# pip install -r requirements.txt
+torch
+gradio>=3.32.0,<4.0  # User UI
+timm~=0.6.12  # For midas
+matplotlib
+trimesh  # For creating simple meshes
+numba>=0.57.0  # Speeding up CPU stereoimage generation
+vispy>=0.13.0
+rembg>=2.0.50  # Remove background
+moviepy>=1.0.2,<2.0
+transforms3d>=0.4.1
+imageio>=2.4.1,<3.0
+imageio-ffmpeg
+networkx>=2.5
+pyqt5; sys_platform == 'windows'
+pyqt6; sys_platform != 'windows'

src/misc.py

@@ -4,20 +4,11 @@
 import builtins
 
 def get_commit_hash():
-    def call_git(dir):
+    try:
+        file_path = pathlib.Path(__file__).parent
         return subprocess.check_output(
             [os.environ.get("GIT", "git"), "rev-parse", "HEAD"],
-            cwd=dir, shell=False, stderr=subprocess.DEVNULL, encoding='utf8').strip()[0:8]
-
-    try:
-        file_path = pathlib.Path(__file__)
-        path = file_path.parts
-        while len(path) > 0 and path[-1] != REPOSITORY_NAME:
-            path = path[:-1]
-        if len(path) >= 2 and path[-1] == REPOSITORY_NAME and path[-2] == "extensions":
-            return call_git(str(pathlib.Path(*path)))
-
-        return call_git(pathlib.Path.cwd().joinpath('extensions/stable-diffusion-webui-depthmap-script/'))
+            cwd=file_path, shell=False, stderr=subprocess.DEVNULL, encoding='utf8').strip()[0:8]
     except Exception:
         return "<none>"