LLM-app/prepare.py at main · Frank-abcd/LLM-app · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import subprocess
import sys

def install(package):
    # 检查是否已安装
    result = subprocess.run(
        [sys.executable, "-m", "pip", "show", package],
        stdout=subprocess.DEVNULL,
        stderr=subprocess.DEVNULL
    )

    if result.returncode == 0:
        print(f"{package} 已存在，跳过安装")
        return

    try:
        #使用这些命令安装库
        subprocess.run([
            sys.executable, "-m", "pip", "install",
            "--trusted-host", "files.pythonhosted.org",
            "--trusted-host", "pypi.org",
            "--trusted-host", "pypi.python.org",
            "--trusted-host", "files.pythonhosted.org:443",
            "--trusted-host", "pypi.org:443",
            "--trusted-host", "pypi.python.org:443",
            "--trusted-host", "mirrors.aliyun.com",
            "--trusted-host", "mirrors.aliyun.com:443",
            package
        ], check=True)
        print(f"{package} 已成功安装")
    except subprocess.CalledProcessError as e:
        print(f"安装 {package} 时出错: {e}")

def prepare():
    install("openai")
    install("requests")
    install("python-docx")
    install("Pillow")
    install("faiss-cpu")
    install("sentence-transformers")
    install("python-pptx")  # 新增的pptx库安装

if __name__ == "__main__":
    prepare()