File size: 1,538 Bytes
8a469fd |
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 45 46 47 48 49 50 |
import launch
from scripts.physton_prompt.get_lang import get_lang
packages = {
"chardet": "chardet",
"fastapi": "fastapi",
"execjs": "PyExecJS",
"lxml": "lxml",
"tqdm": "tqdm",
"pathos": "pathos",
"cryptography": "cryptography",
# The following packages are required for translation service. If you do not need translation service, you can remove them.
# 以下是翻译所需的包，如果不需要翻译服务，可以删除掉它们。
"openai": "openai",
"boto3": "boto3",
"aliyunsdkcore": "aliyun-python-sdk-core",
"aliyunsdkalimt": "aliyun-python-sdk-alimt",
}
def get_packages_state():
states = []
for package_name in packages:
package = packages[package_name]
item = {
'name': package_name,
'package': package,
'state': False
}
if launch.is_installed(package) or launch.is_installed(package_name):
item['state'] = True
states.append(item)
return states
def install_package(name, package):
result = {'state': False, 'message': ''}
try:
launch.run_pip(f"install {package}", f"sd-webui-prompt-all-in-one: {name}")
result['state'] = True
result['message'] = get_lang('install_success', {'0': package})
except Exception as e:
print(e)
print(f'Warning: Failed to install {package}, some preprocessors may not work.')
result['message'] = get_lang('install_failed', {'0': package}) + '\n' + str(e)
return result
|