feat:conditional pkg install

app.py

@@ -2,27 +2,17 @@
 import subprocess
 import os
 import sys
-import pkg_resources
-'''
-env_name = os.path.basename(sys.prefix)
-if env_name == 'fossil':
-    subprocess.call('pip install tensorflow==2.9'.split())
-    subprocess.call('pip install keras==2.9'.split())
-    subprocess.call('pip install git+https://github.com/facebookresearch/segment-anything.git'.split()) 
-    subprocess.call('pip install opencv-python-headless==4.5.5.64'.split())
-    subprocess.call('pip install git+https://github.com/cocodataset/panopticapi.git'.split())
-    subprocess.call('pip install python-dotenv'.split())
-    subprocess.call('pip install torch torchvision '.split())
-    subprocess.call('pip install xplique'.split())
-    subprocess.call('pip install gradio'.split())
-else:
-    print("Please use venv named 'fossil' in case pkgs installed to other unexpected envs or locations.")
+import importlib.metadata
 
 def is_pkg_installed(pkg_name):
-    installed_pkgs = {pkg.key for pkg in pkg_resources.working_set}
-    return pkg_name.lower() in installed_pkgs
-
+    try:
+        importlib.metadata.distribution(pkg_name)
+        return True
+    except importlib.metadata.PackageNotFoundError:
+        return False
 packages_to_install = [
+    ('gradio',None),
+    ('numpy',None),
     ('tensorflow', '2.9'),
     ('keras', '2.9'),
     ('opencv-python-headless', '4.5.5.64'),
@@ -32,17 +22,20 @@ packages_to_install = [
     ('xplique', None),
 ]
 
-for package, version in packages_to_install:
-    package_spec = f"{package}=={version}" if version else package
-    if not is_pkg_installed(package):
-        print(f"Installing {package_spec}...")
-        subprocess.call(f"pip install {package_spec}".split())
-    else:
-        print(f"{package_spec} is already installed.")
+env_name = os.path.basename(sys.prefix)
+if env_name == 'fossil': # in case pkgs installed to unexpected env during local dev
+    for package, version in packages_to_install:
+        package_spec = f"{package}=={version}" if version else package
+        if not is_pkg_installed(package):
+            subprocess.call(f"pip install {package_spec}".split())
+        else:
+            print(f"{package_spec} is already installed.")
+else:
+    print("Please use venv named 'fossil' in case pkgs installed to other unexpected envs or locations.")
 
 
 #print(os.getenv('SYSTEM'))
-'''
+
 '''
 if os.getenv('SYSTEM') == 'spaces':
     subprocess.call('pip install tensorflow==2.9'.split())