Streamlitアプリの依存関係を自動チェック＆インストールしてくれるHome.pyを作った話 #Python – Qiita

import streamlit as st
import os
import ast
import pkg_resources
import subprocess
import sys

# --- stdlib_list の自動インストール ---
try:
    from stdlib_list import stdlib_list
except ModuleNotFoundError:
    with st.spinner("📦 stdlib_list をインストール中..."):
        subprocess.run([sys.executable, "-m", "pip", "install", "stdlib-list"], check=True)
    from stdlib_list import stdlib_list

# --- よくある import名 → pip install名 の変換辞書 ---
import_to_pip = {
    "sklearn": "scikit-learn",
    "cv2": "opencv-python",
    "PIL": "Pillow",
    "bs4": "beautifulsoup4",
    "yaml": "PyYAML",
    "Crypto": "pycryptodome",
    "Image": "Pillow",
    "matplotlib": "matplotlib",
    "seaborn": "seaborn",
    "numpy": "numpy",
    "pandas": "pandas",
    "tensorflow": "tensorflow",
    "torch": "torch",
    "plotly": "plotly",
    "flask": "flask",
    "fastapi": "fastapi",
}

# --- 設定 ---
PAGES_DIR = "pages"
PYTHON_VERSION = f"{sys.version_info.major}.{sys.version_info.minor}"

# --- 標準ライブラリ一覧を取得 ---
stdlibs = set(stdlib_list(PYTHON_VERSION))

# --- ファイルから import 文を抽出 ---
def extract_imports_from_file(file_path):
    with open(file_path, "r", encoding="utf-8") as f:
        tree = ast.parse(f.read(), filename=file_path)

    imports = set()
    for node in ast.walk(tree):
        if isinstance(node, ast.Import):
            for n in node.names:
                imports.add(n.name.split('.')[0])
        elif isinstance(node, ast.ImportFrom):
            if node.module:
                imports.add(node.module.split('.')[0])
    return imports

# --- パッケージがインストールされているか判定（変換対応） ---
def is_installed(import_name):
    pip_name = import_to_pip.get(import_name, import_name)
    if import_name in stdlibs:
        return True
    try:
        pkg_resources.get_distribution(pip_name)
        return True
    except pkg_resources.DistributionNotFound:
        return False

# --- メイン処理 ---
st.title("📦 ページ別ライブラリチェック＆インストーラー")

missing_packages = {}

# 各ページの.pyファイルを走査
for filename in os.listdir(PAGES_DIR):
    if filename.endswith(".py"):
        filepath = os.path.join(PAGES_DIR, filename)
        imported = extract_imports_from_file(filepath)

        not_installed = [pkg for pkg in imported if not is_installed(pkg)]
        if not_installed:
            missing_packages[filename] = not_installed

# --- 結果表示 ---
if not missing_packages:
    st.success("✅ すべてのページに必要なライブラリはインストール済みです 🎉")
else:
    st.subheader("📄 不足パッケージの確認とインストール")
    all_missing = set()
    for file, pkgs in missing_packages.items():
        with st.expander(f"📄 {file} に不足しているパッケージ"):
            for pkg in pkgs:
                pip_pkg = import_to_pip.get(pkg, pkg)
                all_missing.add(pip_pkg)
                col1, col2 = st.columns([3, 1])
                col1.write(f"- `{pkg}` ➜ `{pip_pkg}`")
                if col2.button(f"インストール: {pip_pkg}", key=f"{file}-{pkg}"):
                    with st.spinner(f"{pip_pkg} をインストール中..."):
                        result = subprocess.run(
                            [sys.executable, "-m", "pip", "install", pip_pkg],
                            capture_output=True, text=True
                        )
                        if result.returncode == 0:
                            st.success(f"{pip_pkg} をインストールしました ✅")
                        else:
                            st.error(f"{pip_pkg} のインストールに失敗しました ❌\n\n{result.stderr}")

    # --- requirements.txt への出力 ---
    st.subheader("📄 requirements.txt を自動生成")
    if st.button("📥 requirements.txt に保存"):
        try:
            with open("requirements.txt", "w", encoding="utf-8") as f:
                for pkg in sorted(all_missing):
                    f.write(f"{pkg}\n")
            st.success("✅ requirements.txt を保存しました！")
            st.code("\n".join(sorted(all_missing)), language="text")
        except Exception as e:
            st.error(f"❌ 保存に失敗しました: {e}")