From c0414a91c1bd9a3bdc5b3467dd459fdfcfef1b54 Mon Sep 17 00:00:00 2001
From: Andrew Weisman <andrew.weisman@nih.gov>
Date: Thu, 26 Sep 2024 13:13:45 -0400
Subject: [PATCH] This seems to be working for both published and unpublished
 dashboards, though I'm a bit nervous about the consistency of the
 installation of 'annoy'. But the last two times it worked. This adds
 Python-based installation of packages after environment.yml/startup.sh have
 been used to set up the environments.

---
 Multiplex_Analysis_Web_Apps.py |  3 ++
 install_missing_packages.py    | 99 ++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)
 create mode 100644 install_missing_packages.py

diff --git a/Multiplex_Analysis_Web_Apps.py b/Multiplex_Analysis_Web_Apps.py
index 37448d8..2f37a89 100644
--- a/Multiplex_Analysis_Web_Apps.py
+++ b/Multiplex_Analysis_Web_Apps.py
@@ -7,6 +7,9 @@
 import numpy as np
 import subprocess
 import platform_io
+import install_missing_packages
+
+install_missing_packages.live_package_installation()
 
 # Note if any of the following imports having "  # slow" are not commented out, there is a delay in running the forking test
 from pages2 import data_import_and_export
diff --git a/install_missing_packages.py b/install_missing_packages.py
new file mode 100644
index 0000000..e7ec24c
--- /dev/null
+++ b/install_missing_packages.py
@@ -0,0 +1,99 @@
+import subprocess
+import importlib
+
+def is_mamba_installed():
+    try:
+        # Run the 'mamba --version' command
+        result = subprocess.run(['mamba', '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+        # Check if the command was successful
+        if result.returncode == 0:
+            print("&&&& Mamba is installed.")
+            return True
+        else:
+            print("&&&& Mamba is not installed.")
+            return False
+    except FileNotFoundError:
+        # The command was not found
+        print("&&&& Mamba is not installed.")
+        return False
+
+def install_with_mamba(packages):
+    print(f"&&&& Attempting to install {', '.join(packages)} with mamba.")
+    try:
+        # Run the 'mamba install <packages>' command
+        result = subprocess.run(['mamba', 'install', '-y'] + packages, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+        # Check if the command was successful
+        if result.returncode == 0:
+            print(f"&&&& {', '.join(packages)} have been installed successfully with mamba.")
+            print(result.stdout)
+        else:
+            print(f"&&&& Failed to install {', '.join(packages)} with mamba.")
+            print(result.stderr)
+    except Exception as e:
+        print(f"&&&& An error occurred while trying to install {', '.join(packages)} with mamba: {e}")
+
+def install_with_conda(packages):
+    print(f"&&&& Attempting to install {', '.join(packages)} with conda.")
+    try:
+        # Run the 'conda install <packages>' command
+        result = subprocess.run(['conda', 'install', '-y'] + packages, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+        # Check if the command was successful
+        if result.returncode == 0:
+            print(f"&&&& {', '.join(packages)} have been installed successfully with conda.")
+            print(result.stdout)
+        else:
+            print(f"&&&& Failed to install {', '.join(packages)} with conda.")
+            print(result.stderr)
+    except Exception as e:
+        print(f"&&&& An error occurred while trying to install {', '.join(packages)} with conda: {e}")
+
+def install_with_pip(packages):
+    print(f"&&&& Attempting to install {', '.join(packages)} with pip.")
+    try:
+        # Run the 'pip install <packages>' command
+        result = subprocess.run(['pip', 'install'] + packages, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+        
+        # Check if the command was successful
+        if result.returncode == 0:
+            print(f"&&&& {', '.join(packages)} have been installed successfully with pip.")
+            print(result.stdout)
+        else:
+            print(f"&&&& Failed to install {', '.join(packages)} with pip.")
+            print(result.stderr)
+    except Exception as e:
+        print(f"&&&& An error occurred while trying to install {', '.join(packages)} with pip: {e}")
+
+def live_package_installation():
+    packages_to_install = ['hnswlib', 'parc', 'sklearn_ann', 'annoy', 'pyNNDescent']  # last two probably only needed for published dashboards
+    installers_to_use = ['mamba', 'conda', 'pip']
+
+    for package in packages_to_install:
+        try:
+            importlib.import_module(package.lower())
+            print(f"&&&& {package} is already installed.")
+        except ImportError:
+            print(f"&&&& {package} is not installed.")
+            for installer in installers_to_use:
+                print(f"&&&& Trying to install {package} using {installer}.")
+                if installer == 'mamba':
+                    if is_mamba_installed():
+                        install_with_mamba([package])
+                    else:
+                        print(f"&&&& mamba is not installed. Trying the next installer.")
+                        continue
+                elif installer == 'conda':
+                    install_with_conda([package])
+                elif installer == 'pip':
+                    install_with_pip([package])
+                
+                try:
+                    importlib.import_module(package.lower())
+                    print(f"&&&& {package} has been successfully installed using {installer}.")
+                    break
+                except ImportError:
+                    print(f"&&&& {package} was not successfully installed with {installer}.")
+            else:
+                print(f"&&&& {package} could not be installed after trying all installers.")