diff --git a/install.py b/install.py index a51bc97..a0acdf9 100644 --- a/install.py +++ b/install.py @@ -7,7 +7,8 @@ req_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), "requiremen def dist2package(dist: str): return ({ "python-dotenv": "dotenv", - "Pillow": "PIL" + "Pillow": "PIL", + "pillow-avif-plugin": "pillow_avif" }).get(dist, dist) # copy from controlnet, thanks diff --git a/scripts/iib/api.py b/scripts/iib/api.py index c524c23..8da3f0e 100644 --- a/scripts/iib/api.py +++ b/scripts/iib/api.py @@ -3,7 +3,7 @@ import os from pathlib import Path import shutil import sqlite3 -import pillow_avif + from scripts.iib.dir_cover_cache import get_top_4_media_info from scripts.iib.tool import ( @@ -60,6 +60,12 @@ import urllib.parse from scripts.iib.fastapi_video import range_requests_response, close_video_file_reader from scripts.iib.parsers.index import parse_image_info + +try: + import pillow_avif +except Exception as e: + logger.error(e) + index_html_path = os.path.join(cwd, "vue/dist/index.html") # 在app.py也被使用