Upload 7 files

Dockerfile

@@ -0,0 +1,27 @@
+# 使用官方的 Python 基础镜像
+FROM python:3.9-slim
+
+# 设置工作目录
+WORKDIR /app
+
+# 将依赖文件复制到工作目录
+COPY requirements.txt ./
+
+# 安装依赖
+# --no-cache-dir 减少镜像大小
+# --trusted-host pypi.python.org --trusted-host files.pythonhosted.org --trusted-host pypi.org 是为了解决某些网络环境下可能的SSL问题
+RUN pip install --no-cache-dir --trusted-host pypi.python.org --trusted-host files.pythonhosted.org --trusted-host pypi.org -r requirements.txt
+
+# 将当前目录下的所有文件（app.py, .icc 文件等）复制到工作目录
+COPY . .
+
+# 暴露 Streamlit 默认端口
+EXPOSE 8501
+
+# 设置健康检查 (Hugging Face Spaces 会用到)
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+
+# 运行 Streamlit 应用
+# 使用 exec form of CMD, and ensure server binds to 0.0.0.0
+ENTRYPOINT ["streamlit", "run"]
+CMD ["app.py", "--server.port=8501", "--server.address=0.0.0.0"]

README.md

@@ -1,10 +1,43 @@
 ---
-title: Hdr
-emoji: 📊
-colorFrom: green
-colorTo: indigo
+title: 发光青蛙表情包 HDR 制作
+emoji: 🐸️
+colorFrom: indigo
+colorTo: blue
 sdk: docker
+app_port: 8501
 pinned: false
+# Python Version (for information, Dockerfile controls the actual version)
+python_version: 3.9
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🖼️ ICC Profile Embedder
+
+这是一个 Streamlit Web 应用，允许用户上传一个或多个图片文件，并嵌入选定的 ICC 颜色配置文件。
+
+## 功能
+
+*   **选择 ICC Profile**: 用户可以从以下预设的 ICC 配置文件中选择一个：
+    *   原版青蛙 - 默认
+    *   超亮版本
+    *   多兼容版本
+*   **图片上传**: 支持上传 JPG, PNG, TIFF 格式的图片。
+*   **ICC 嵌入**: 将选定的 ICC Profile 嵌入到每一张上传的图片中。
+*   **结果展示**: 处理后的图片会在各自的标签页 (Tab) 中展示。
+*   **单张下载**: 每个处理后的图片都有独立的下载按钮。
+*   **批量下载**: 如果上传并处理了多张图片，提供一个 ZIP 压缩包下载所有处理后图片的功能。
+
+## 如何使用
+
+1.  在页面顶部选择一个你想要嵌入的 ICC 配置文件。
+2.  点击 "Browse files" 或拖拽图片到上传区域以上传一个或多个图片。
+3.  应用会自动处理图片，并在下方的标签页中显示结果。
+4.  你可以点击每个图片下方的下载按钮下载单张图片，或者点击页面底部的 "下载全部 ZIP" 按钮下载所有处理后的图片。
+
+## 技术栈
+
+*   Streamlit
+*   Pillow (PIL Fork)
+*   Docker
+
+---
+由 [aianyu] 创建 (可选)

app.py

@@ -0,0 +1,206 @@
+import streamlit as st
+from PIL import Image, UnidentifiedImageError
+import io
+import os
+import zipfile
+
+# --- 配置 ---
+ICC_PROFILE_OPTIONS = {
+    "原版青蛙": "qingwa.icc",
+    "超亮版本": "6172.icc",
+    "多兼容版本": "2020.icc",
+}
+DEFAULT_ICC_KEY = "原版青蛙"
+SUPPORTED_FORMATS = ["jpg", "jpeg", "png", "tif", "tiff"]
+
+
+# --- 辅助函数 ---
+
+def get_icc_profile_bytes(selected_icc_filename):
+    script_dir = os.path.dirname(__file__)
+    icc_profile_path = os.path.join(script_dir, selected_icc_filename)
+    if not os.path.exists(icc_profile_path):
+        st.error(f"错误: ICC 配置文件 '{selected_icc_filename}' 未在脚本目录 '{script_dir}' 中找到。")
+        return None
+    try:
+        with open(icc_profile_path, "rb") as f:
+            return f.read()
+    except Exception as e:
+        st.error(f"读取 ICC 文件 '{selected_icc_filename}' 时出错: {e}")
+        return None
+
+
+def embed_icc_profile(image_bytes, icc_profile_bytes, original_filename):
+    if icc_profile_bytes is None:
+        st.error(f"由于无法加载ICC配置文件，未能处理图片 '{original_filename}'。")
+        return None, None
+    try:
+        img = Image.open(io.BytesIO(image_bytes))
+        img_format = img.format if img.format else "PNG"
+
+        original_mode = img.mode
+        exif_data = img.info.get("exif")
+
+        if img.mode == 'P':
+            has_transparency = "transparency" in img.info
+            img = img.convert('RGBA' if has_transparency else 'RGB')
+
+        output_image_bytes = io.BytesIO()
+        params = {"icc_profile": icc_profile_bytes}
+        if exif_data:
+            params["exif"] = exif_data
+
+        if img_format.upper() == "JPEG":
+            params["quality"] = 95
+            params["optimize"] = True
+
+        try:
+            img.save(output_image_bytes, format=img_format, **params)
+        except Exception as e_save:
+            st.warning(
+                f"无法以格式 '{img_format}' (模式: {original_mode}) 保存 '{original_filename}' (原因: {e_save})。尝试转换为兼容模式并以PNG格式保存...")
+
+            img_reloaded = Image.open(io.BytesIO(image_bytes))
+            target_mode = 'RGB'
+            if img_reloaded.mode in ['LA', 'RGBA', 'PA'] or \
+                    (img_reloaded.mode == 'P' and "transparency" in img_reloaded.info):
+                target_mode = 'RGBA'
+
+            img_converted = img_reloaded
+            if img_reloaded.mode != target_mode and img_reloaded.mode not in ['RGB', 'RGBA', 'L', 'CMYK', 'LAB']:
+                st.caption(
+                    f"提示: 将图像 '{original_filename}' 从 {img_reloaded.mode} 转换为 {target_mode} 以便用PNG保存。")
+                img_converted = img_reloaded.convert(target_mode)
+            elif img_reloaded.mode not in ['RGB', 'RGBA']:
+                img_converted = img_reloaded.convert(target_mode)
+
+            output_image_bytes = io.BytesIO()
+            png_params = {"icc_profile": icc_profile_bytes}
+            img_converted.save(output_image_bytes, format="PNG", **png_params)
+            img_format = "PNG"
+
+        output_image_bytes.seek(0)
+        return output_image_bytes, img_format
+
+    except UnidentifiedImageError:
+        st.error(f"无法识别文件 '{original_filename}' 为有效图片。")
+        return None, None
+    except Exception as e:
+        st.error(f"处理图片 '{original_filename}' 时发生错误: {e}")
+        return None, None
+
+
+# --- Streamlit 应用界面 ---
+st.set_page_config(page_title="发光青蛙表情包 HDR 制作")
+st.title("🐸 发光青蛙表情包 HDR 制作")
+
+# ICC 文件选择 (使用单选按钮)
+st.markdown("#### 1. 选择要嵌入的 ICC 配置文件：")
+selected_icc_display_name = st.radio(
+    label="选择 ICC Profile:",  # label 可以设为空字符串如果你觉得上面的 markdown 标题足够
+    options=list(ICC_PROFILE_OPTIONS.keys()),
+    index=list(ICC_PROFILE_OPTIONS.keys()).index(DEFAULT_ICC_KEY),  # 设置默认选项
+    key="icc_radio_selector",
+    horizontal=True,  # 让单选按钮水平排列
+    label_visibility="collapsed"  # 如果上面的 markdown 标题已足够，可以隐藏 radio 的 label
+)
+# 根据用户选择的显示名称获取实际的文件名
+SELECTED_ICC_FILENAME = ICC_PROFILE_OPTIONS[selected_icc_display_name]
+icc_profile_base_name = SELECTED_ICC_FILENAME.split('.')[0]
+
+st.info(f"当前选定的 ICC 配置文件：**{selected_icc_display_name}**")
+
+# 文件上传组件
+st.markdown("#### 2. 上传图片文件：")
+uploaded_files = st.file_uploader(
+    "请上传一个或多个图片文件:",
+    type=SUPPORTED_FORMATS,
+    accept_multiple_files=True,
+    key="file_uploader_main",
+    label_visibility="collapsed"  # 隐藏 label，因为上面的 markdown 标题已足够
+)
+
+if uploaded_files:
+    icc_profile_data = get_icc_profile_bytes(SELECTED_ICC_FILENAME)
+
+    if icc_profile_data is None:
+        st.error(f"无法加载选定的 ICC 配置文件 '{SELECTED_ICC_FILENAME}'。请检查文件是否存在。处理已中止。")
+        st.stop()
+
+    processed_files_info = []
+
+    with st.status(f"正在使用 '{selected_icc_display_name}' 处理图片...", expanded=True) as status_container:
+        for i, uploaded_file in enumerate(uploaded_files):
+            status_container.write(f"⏳ 处理: {uploaded_file.name} ({i + 1}/{len(uploaded_files)})")
+            image_bytes = uploaded_file.getvalue()
+
+            processed_image_bytes, processed_format = embed_icc_profile(image_bytes, icc_profile_data,
+                                                                        uploaded_file.name)
+
+            if processed_image_bytes:
+                base, ext = os.path.splitext(uploaded_file.name)
+                actual_ext = f".{processed_format.lower()}" if processed_format else ext
+                new_filename = f"{base}_{icc_profile_base_name}_embedded{actual_ext}"
+                mime_type = f"image/{processed_format.lower()}" if processed_format else uploaded_file.type
+
+                processed_files_info.append({
+                    "name": new_filename,
+                    "data": processed_image_bytes,
+                    "mime": mime_type,
+                    "original_name": uploaded_file.name,
+                    "id": i
+                })
+                status_container.write(f"✅ 完成: {new_filename}")
+            else:
+                status_container.write(f"❌ 失败: {uploaded_file.name}")
+        status_container.update(label="所有文件处理完毕!", state="complete", expanded=False)
+
+    if processed_files_info:
+        st.markdown("#### 3. 处理结果：")
+
+        tab_labels = [item["name"] for item in processed_files_info]
+
+        if not tab_labels:
+            st.info("没有成功处理的图片可供展示。")
+        else:
+            image_tabs = st.tabs(tab_labels)
+
+            for i, item in enumerate(processed_files_info):
+                with image_tabs[i]:
+                    st.image(item["data"], use_container_width=True)
+
+                    download_data_buffer = io.BytesIO(item["data"].getvalue())
+                    download_data_buffer.seek(0)
+
+                    st.download_button(
+                        label=f"下载 {item['name']}",
+                        data=download_data_buffer,
+                        file_name=item["name"],
+                        mime=item["mime"],
+                        key=f"download_tab_{item['original_name']}_{item['id']}_{icc_profile_base_name}"
+                    )
+    else:
+        if uploaded_files:
+            st.warning("所有上传的文件都未能成功处理。")
+
+    if len(processed_files_info) > 1:
+        st.markdown("#### 下载所有处理过的文件 (.zip)")
+
+        zip_buffer = io.BytesIO()
+        with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zf:
+            for item in processed_files_info:
+                item_data_for_zip = io.BytesIO(item["data"].getvalue())
+                item_data_for_zip.seek(0)
+                zf.writestr(item["name"], item_data_for_zip.read())
+
+        zip_buffer.seek(0)
+        st.download_button(
+            label=f"下载全部 ({len(processed_files_info)} 张图片) .zip",
+            data=zip_buffer,
+            file_name=f"processed_images_{icc_profile_base_name}_batch.zip",
+            mime="application/zip",
+            key=f"download_all_zip_{icc_profile_base_name}"
+        )
+
+elif not uploaded_files:
+    st.info("请选择 ICC 配置文件并上传图片文件开始处理。")

requirements.txt

@@ -0,0 +1,2 @@
+streamlit
+Pillow