From c071566cc8a8ddf6e1e5e37a68ffe2e266e9059d Mon Sep 17 00:00:00 2001
From: AlUlkesh <99896447+AlUlkesh@users.noreply.github.com>
Date: Wed, 7 Jun 2023 23:31:43 +0200
Subject: [PATCH] install.py fix for dist_name <> pkg_name

---
 install.py | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/install.py b/install.py
index 041470a..3748cc7 100644
--- a/install.py
+++ b/install.py
@@ -3,6 +3,16 @@ import os
 import pkg_resources
 
 req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requirements.txt")
+
+def dist2package(dist):
+    if dist == "python-dotenv":
+        package = "dotenv"
+    elif dist == "Pillow":
+        package = "PIL"
+    else:
+        package = dist
+    return package
+
 # copy from controlnet, thanks
 with open(req_file) as file:
     for package in file:
@@ -13,8 +23,8 @@ with open(req_file) as file:
                 installed_version = pkg_resources.get_distribution(package_name).version
                 if installed_version != package_version:
                     launch.run_pip(f"install {package}", f"sd-webui-infinite-image-browsing requirement: changing {package_name} version from {installed_version} to {package_version}")
-            elif not launch.is_installed(package):
+            elif not launch.is_installed(dist2package(package)):
                 launch.run_pip(f"install {package}", f"sd-webui-infinite-image-browsing requirement: {package}")
         except Exception as e:
             print(e)
-            print(f'Warning: Failed to install {package}, some preprocessors may not work.')
\ No newline at end of file
+            print(f'Warning: Failed to install {package}, something may not work.')