|
|
""" |
|
|
图片处理工具函数 |
|
|
""" |
|
|
|
|
|
import base64 |
|
|
from typing import List |
|
|
from PIL import Image |
|
|
|
|
|
|
|
|
def get_image_list(file_paths: List[str]) -> List[Image.Image]: |
|
|
"""从文件路径列表获取PIL图片列表用于展示""" |
|
|
if not file_paths: |
|
|
return [] |
|
|
|
|
|
images = [] |
|
|
for path in file_paths: |
|
|
try: |
|
|
img = Image.open(path) |
|
|
images.append(img) |
|
|
except Exception: |
|
|
continue |
|
|
return images |
|
|
|
|
|
|
|
|
def create_image_html(file_paths: List[str]) -> str: |
|
|
"""创建图片展示的HTML代码""" |
|
|
if not file_paths: |
|
|
return "<div id='image-display-area'><div class='no-images'>No image uploaded - will generate from text only</div></div>" |
|
|
|
|
|
html_items = [] |
|
|
for i, path in enumerate(file_paths): |
|
|
try: |
|
|
|
|
|
with open(path, "rb") as img_file: |
|
|
img_data = base64.b64encode(img_file.read()).decode() |
|
|
img_ext = path.split('.')[-1].lower() |
|
|
if img_ext in ['jpg', 'jpeg']: |
|
|
mime_type = 'image/jpeg' |
|
|
elif img_ext == 'png': |
|
|
mime_type = 'image/png' |
|
|
elif img_ext == 'gif': |
|
|
mime_type = 'image/gif' |
|
|
elif img_ext == 'webp': |
|
|
mime_type = 'image/webp' |
|
|
else: |
|
|
mime_type = 'image/jpeg' |
|
|
|
|
|
img_src = f"data:{mime_type};base64,{img_data}" |
|
|
|
|
|
html_items.append(f""" |
|
|
<div class="image-item" data-index="{i}"> |
|
|
<img src="{img_src}" alt="Uploaded image {i + 1}"> |
|
|
<div class="delete-btn" onclick="deleteImageByIndex({i})" data-index="{i}">×</div> |
|
|
</div> |
|
|
""") |
|
|
except: |
|
|
continue |
|
|
|
|
|
if not html_items: |
|
|
return "<div id='image-display-area'><div class='no-images'>No image uploaded - will generate from text only</div></div>" |
|
|
|
|
|
html_content = f""" |
|
|
<div id='image-display-area'> |
|
|
{''.join(html_items)} |
|
|
</div> |
|
|
""" |
|
|
|
|
|
return html_content |
|
|
|
