Merge branch 'develop'

pic_to_header/app.py

@@ -1,65 +1,111 @@
 import streamlit as st
-import os
-from core import process_header_image
+from PIL import Image
+import numpy as np
+from modules.image_processor import process_image, prepare_image, convert_to_pil
+from modules.mask_manager import MaskManager
+from modules.utils import create_download_button
 
 def main():
-    st.markdown("""
-
-<div align="center">
-
-# Pic-to-Header
+    # ワイドモードの設定
+    st.set_page_config(layout="wide")
 
-![Pic-to-Header Result](https://raw.githubusercontent.com/Sunwood-ai-labs/pic-to-header/refs/heads/main/assets/result.png)
-
-[![GitHub license](https://img.shields.io/github/license/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/blob/main/LICENSE)
-[![GitHub stars](https://img.shields.io/github/stars/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/stargazers)
-[![GitHub issues](https://img.shields.io/github/issues/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/issues)
+    st.markdown("""
+    <div align="center">
+    
+    # Pic-to-Header
 
-![Python](https://img.shields.io/badge/python-3670A0?style=for-the-badge&logo=python&logoColor=ffdd54)
-![Streamlit](https://img.shields.io/badge/Streamlit-FF4B4B?style=for-the-badge&logo=Streamlit&logoColor=white)
-![OpenCV](https://img.shields.io/badge/opencv-%23white.svg?style=for-the-badge&logo=opencv&logoColor=white)
+    <img src="https://raw.githubusercontent.com/Sunwood-ai-labs/pic-to-header/refs/heads/main/assets/result.png" width="50%">
 
-</div>
+    [![GitHub license](https://img.shields.io/github/license/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/blob/main/LICENSE)
+    [![GitHub stars](https://img.shields.io/github/stars/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/stargazers)
+    [![GitHub issues](https://img.shields.io/github/issues/Sunwood-ai-labs/pic-to-header)](https://github.com/Sunwood-ai-labs/pic-to-header/issues)
 
-Pic-to-Headerは、マスク画像と入力画像を使用してヘッダー画像を生成するPythonアプリケーションです。
+    ![Python](https://img.shields.io/badge/python-3670A0?style=for-the-badge&logo=python&logoColor=ffdd54)
+    ![Streamlit](https://img.shields.io/badge/Streamlit-FF4B4B?style=for-the-badge&logo=Streamlit&logoColor=white)
+    ![OpenCV](https://img.shields.io/badge/opencv-%23white.svg?style=for-the-badge&logo=opencv&logoColor=white)
+    </div>
+    """, unsafe_allow_html=True)
 
-                """, unsafe_allow_html=True)
+    st.write("Pic-to-Headerは、マスク画像と入力画像を使用してヘッダー画像を生成するPythonアプリケーションです。")
     st.write("マスク画像と入力画像をアップロードして、ヘッダー画像を生成します。")
 
-    input_image = st.file_uploader("入力画像をアップロード", type=["png", "jpg", "jpeg"])
-    mask_image = st.file_uploader("マスク画像をアップロード", type=["png", "jpg", "jpeg"])
-
-    if input_image is not None and mask_image is not None:
-        if st.button("ヘッダー画像を生成"):
-            # 一時ファイルとして保存
-            input_path = f"temp_input.{input_image.name.split('.')[-1]}"
-            mask_path = f"temp_mask.{mask_image.name.split('.')[-1]}"
-            output_path = "output_header.png"
-
-            with open(input_path, "wb") as f:
-                f.write(input_image.getbuffer())
-            with open(mask_path, "wb") as f:
-                f.write(mask_image.getbuffer())
-
-            # 画像処理
-            process_header_image(input_path, mask_path, output_path)
-
-            # 結果を表示
-            st.image(output_path, caption="生成されたヘッダー画像")
+    # 2段組レイアウトの作成（左側を狭く、右側を広く）
+    control_column, display_column = st.columns([1, 2])
+
+    with control_column:
+        # マスク管理インスタンスの作成
+        mask_manager = MaskManager()
+
+        # マスク画像の取得方法を選択
+        mask_source = st.radio(
+            "マスク画像の取得方法を選択",
+            ["プリセットから選択", "URLから取得", "ファイルをアップロード"]
+        )
+
+        mask_image = None
+        if mask_source == "プリセットから選択":
+            preset_name = st.selectbox(
+                "プリセットを選択",
+                mask_manager.get_preset_names()
+            )
+            mask_image = mask_manager.get_preset_mask(preset_name)
+        
+        elif mask_source == "URLから取得":
+            mask_url = st.text_input("マスク画像のURLを入力")
+            if mask_url:
+                mask_image = mask_manager.load_mask_from_url(mask_url)
+        
+        else:  # ファイルをアップロード
+            mask_file = st.file_uploader("マスク画像をアップロード", type=["png", "jpg", "jpeg"])
+            if mask_file is not None:
+                mask_image = Image.open(mask_file)
+
+        # 透明度の調整
+        alpha = st.slider("マスクの透明度", 0.0, 1.0, 1.0)
+
+        # ファイルアップロード（複数ファイル対応）
+        uploaded_files = st.file_uploader(
+            "画像をアップロードしてください",
+            type=["png", "jpg", "jpeg", "webp"],
+            accept_multiple_files=True
+        )
+        if mask_image:
+            # マスク画像のプレビュー表示
+            st.write("選択中のマスク画像:")
+            st.image(mask_image, caption="マスク画像", use_column_width=True)
             
-            # ダウンロードボタンを追加
-            with open(output_path, "rb") as file:
-                btn = st.download_button(
-                    label="ヘッダー画像をダウンロード",
-                    data=file,
-                    file_name="header_image.png",
-                    mime="image/png"
-                )
-
-            # 一時ファイルを削除
-            os.remove(input_path)
-            os.remove(mask_path)
-            os.remove(output_path)
-
+    with display_column:
+
+        if uploaded_files:
+            st.write("処理結果:")
+            # 各画像の処理と表示
+            for idx, uploaded_file in enumerate(uploaded_files):
+                # 水平に2列で表示するための設定
+                img_col1, img_col2 = st.columns(2)
+                
+                # 元画像の表示
+                with img_col1:
+                    st.write(f"元画像 {idx + 1}:")
+                    st.image(uploaded_file, use_column_width=True)
+
+                # 処理後の画像の表示
+                with img_col2:
+                    st.write(f"処理後 {idx + 1}:")
+                    input_image = prepare_image(uploaded_file)
+                    if input_image is not None:
+                        result = process_image(input_image, np.array(mask_image), alpha)
+                        result_pil = convert_to_pil(result)
+                        st.image(result_pil, use_column_width=True)
+                        
+                        # ダウンロードボタンの作成
+                        create_download_button(
+                            result_pil,
+                            f"header_{uploaded_file.name}"
+                        )
+                
+                # 区切り線を追加（最後の画像以外）
+                if idx < len(uploaded_files) - 1:
+                    st.markdown("---")
+                        
 if __name__ == "__main__":
     main()

pic_to_header/modules/__init__.py

@@ -0,0 +1 @@
+

pic_to_header/modules/image_processor.py

@@ -0,0 +1,59 @@
+import cv2
+import numpy as np
+from PIL import Image
+import io
+
+def process_image(image, mask_image, alpha=0.5):
+    """画像処理を行う関数"""
+    # PILイメージをnumpy配列に変換
+    if isinstance(image, Image.Image):
+        image = np.array(image)
+    if isinstance(mask_image, Image.Image):
+        mask_image = np.array(mask_image)
+
+    # 画像のチャンネル数を確認し、必要に応じて変換
+    if len(image.shape) == 2:  # グレースケール
+        image = cv2.cvtColor(image, cv2.COLOR_GRAY2BGRA)
+    elif image.shape[2] == 3:  # BGR
+        image = cv2.cvtColor(image, cv2.COLOR_BGR2BGRA)
+    elif image.shape[2] == 4 and image.dtype == np.uint8:
+        pass  # すでにBGRA形式
+    
+    if len(mask_image.shape) == 2:  # グレースケール
+        mask_image = cv2.cvtColor(mask_image, cv2.COLOR_GRAY2BGRA)
+    elif mask_image.shape[2] == 3:  # BGR
+        mask_image = cv2.cvtColor(mask_image, cv2.COLOR_BGR2BGRA)
+    elif mask_image.shape[2] == 4 and mask_image.dtype == np.uint8:
+        pass  # すでにBGRA形式
+
+    # マスク画像のリサイズ
+    mask_image = cv2.resize(mask_image, (image.shape[1], image.shape[0]))
+
+    # マスク画像のアルファチャンネルを取得
+    mask_alpha = mask_image[:, :, 3]
+
+    # 入力画像のアルファチャンネルを更新
+    # マスクのアルファ値を使って元の画像のアルファ値を減算
+    image[:, :, 3] = np.maximum(image[:, :, 3] - (mask_alpha * alpha).astype(np.uint8), 0)
+
+    return image
+
+def convert_to_pil(image):
+    """numpy配列をPIL Imageに変換"""
+    if isinstance(image, np.ndarray):
+        return Image.fromarray(image)
+    return image
+
+def prepare_image(uploaded_file):
+    """アップロードされたファイルを画像として準備"""
+    if uploaded_file is None:
+        return None
+    
+    image_bytes = uploaded_file.read()
+    image = Image.open(io.BytesIO(image_bytes))
+    
+    # RGBAモードに変換
+    if image.mode != 'RGBA':
+        image = image.convert('RGBA')
+    
+    return image

pic_to_header/modules/mask_manager.py

@@ -0,0 +1,38 @@
+import requests
+from PIL import Image
+import io
+import streamlit as st
+
+class MaskManager:
+    def __init__(self):
+        self.default_mask_url = "https://raw.githubusercontent.com/Sunwood-ai-labs/pic-to-header/refs/heads/main/assets/mask.png"
+        self.presets = {
+            "デフォルトマスク": self.default_mask_url,
+        }
+
+    @staticmethod
+    @st.cache_data
+    def load_mask_from_url(url):
+        """URLから画像を読み込む"""
+        try:
+            response = requests.get(url)
+            response.raise_for_status()
+            image = Image.open(io.BytesIO(response.content))
+            return image
+        except Exception as e:
+            st.error(f"マスク画像の読み込みに失敗しました: {str(e)}")
+            return None
+
+    def get_preset_names(self):
+        """プリセット名の一覧を取得"""
+        return list(self.presets.keys())
+
+    def get_preset_mask(self, preset_name):
+        """プリセット名に対応するマスク画像を取得"""
+        if preset_name in self.presets:
+            return self.load_mask_from_url(self.presets[preset_name])
+        return None
+
+    def add_preset(self, name, url):
+        """新しいプリセットを追加"""
+        self.presets[name] = url

pic_to_header/modules/utils.py

@@ -0,0 +1,23 @@
+import streamlit as st
+from PIL import Image
+import io
+
+def convert_image_to_bytes(image):
+    """画像をバイト列に変換"""
+    if image is None:
+        return None
+    
+    img_byte_arr = io.BytesIO()
+    image.save(img_byte_arr, format='PNG')
+    return img_byte_arr.getvalue()
+
+def create_download_button(image, filename):
+    """ダウンロードボタンを作成"""
+    if image is not None:
+        image_bytes = convert_image_to_bytes(image)
+        st.download_button(
+            label="画像をダウンロード",
+            data=image_bytes,
+            file_name=filename,
+            mime="image/png"
+        )