pip_check_v2.py

pip_check_v2.py

import importlib.metadata
import subprocess
import sys

def check_and_install(package_or_file, file=False):
    if file:
        with open(package_or_file, 'r') as file:
            packages = file.readlines()
    else:
        packages = [package_or_file]

    for package in packages:
        package = package.strip()
        try:
            importlib.metadata.version(package)
            print(f"Package {package} installed ok ... ✅")
        except importlib.metadata.PackageNotFoundError:
            subprocess.check_call([sys.executable, '-m', 'pip', 'install', package])

# Usage
#check_and_install("langchain")  # For a single package
#check_and_install("requirements.txt", file=True)  # For a file