import gradio as gr import numpy as np from PIL import Image import io import tempfile import os def resize_images_to_match(images, resize_mode="height"): if not images or len(images) == 0: return [] valid_images = [img for img in images if img is not None] if not valid_images: return [] if resize_mode == "height": target_height = min(img.size[1] for img in valid_images) resized_images = [] for img in valid_images: aspect_ratio = img.size[0] / img.size[1] new_width = int(target_height * aspect_ratio) resized_img = img.resize((new_width, target_height), Image.Resampling.LANCZOS) resized_images.append(resized_img) else: target_width = min(img.size[0] for img in valid_images) resized_images = [] for img in valid_images: aspect_ratio = img.size[1] / img.size[0] new_height = int(target_width * aspect_ratio) resized_img = img.resize((target_width, new_height), Image.Resampling.LANCZOS) resized_images.append(resized_img) return resized_images def concatenate_images(images, direction="horizontal", spacing=0, background_color="white"): if not images or len(images) == 0: return None valid_images = [img for img in images if img is not None] if not valid_images: return None rgb_images = [] for img in valid_images: if img.mode != 'RGB': if img.mode in ('RGBA', 'LA'): background = Image.new('RGB', img.size, 'white') background.paste(img, mask=img.split()[-1] if img.mode == 'RGBA' else None) rgb_images.append(background) else: rgb_images.append(img.convert('RGB')) else: rgb_images.append(img) if len(rgb_images) == 1: return rgb_images[0] if direction == "horizontal": total_width = sum(img.size[0] for img in rgb_images) + spacing * (len(rgb_images) - 1) max_height = max(img.size[1] for img in rgb_images) result = Image.new('RGB', (total_width, max_height), background_color) x_offset = 0 for img in rgb_images: y_offset = (max_height - img.size[1]) // 2 result.paste(img, (x_offset, y_offset)) x_offset += img.size[0] + spacing else: max_width = max(img.size[0] for img in rgb_images) total_height = sum(img.size[1] for img in rgb_images) + spacing * (len(rgb_images) - 1) result = Image.new('RGB', (max_width, total_height), background_color) y_offset = 0 for img in rgb_images: x_offset = (max_width - img.size[0]) // 2 result.paste(img, (x_offset, y_offset)) y_offset += img.size[1] + spacing return result def save_as_jpg(image, quality=95): if image is None: return None if image.mode != 'RGB': if image.mode in ('RGBA', 'LA'): background = Image.new('RGB', image.size, 'white') background.paste(image, mask=image.split()[-1]) image = background else: image = image.convert('RGB') buffer = io.BytesIO() image.save(buffer, format='JPEG', quality=quality, optimize=True) buffer.seek(0) return Image.open(buffer) def process_images( image1, image2, image3, image4, image5, image6, resize_mode, concat_direction, spacing, bg_color, jpg_quality ): uploaded_images = [image1, image2, image3, image4, image5, image6] valid_images = [img for img in uploaded_images if img is not None] if len(valid_images) == 0: return None, "❌ 請至少上傳一張圖片", None if len(valid_images) == 1: return valid_images[0], f"✅ 只有一張圖片，無需拼接", None try: if resize_mode == "等高度 (水平拼接)": resized_images = resize_images_to_match(valid_images, "height") direction = "horizontal" else: resized_images = resize_images_to_match(valid_images, "width") direction = "vertical" if concat_direction != "自動 (根據調整模式)": direction = "horizontal" if concat_direction == "水平拼接" else "vertical" result = concatenate_images(resized_images, direction, spacing, bg_color) if result is None: return None, "❌ 拼接失敗", None result = save_as_jpg(result, quality=jpg_quality) temp_path = os.path.join(tempfile.gettempdir(), "result.jpg") result.save(temp_path, format="JPEG", quality=jpg_quality) info = f"✅ 成功拼接 {len(valid_images)} 張圖片\n" info += f"📐 最終尺寸: {result.size[0]} x {result.size[1]} 像素\n" info += f"🔄 調整模式: {resize_mode}\n" info += f"➡️ 拼接方向: {'水平' if direction == 'horizontal' else '垂直'}\n" info += f"📏 間距: {spacing} 像素\n" info += f"💾 輸出品質: {jpg_quality}%" return result, info, temp_path except Exception as e: return None, f"❌ 處理過程中出現錯誤: {str(e)}", None def create_interface(): with gr.Blocks(title="圖片拼接比較工具", theme=gr.themes.Soft()) as iface: gr.Markdown("# 🖼️ 圖片拼接比較工具") with gr.Row(): with gr.Column(scale=2): gr.Markdown("### 📁 上傳圖片") with gr.Row(): image1 = gr.Image(type="pil", label="圖片 1") image2 = gr.Image(type="pil", label="圖片 2") image3 = gr.Image(type="pil", label="圖片 3") with gr.Row(): image4 = gr.Image(type="pil", label="圖片 4") image5 = gr.Image(type="pil", label="圖片 5") image6 = gr.Image(type="pil", label="圖片 6") gr.Markdown("### ⚙️ 處理設定") with gr.Row(): resize_mode = gr.Dropdown( choices=["等高度 (水平拼接)", "等寬度 (垂直拼接)"], value="等高度 (水平拼接)", label="尺寸調整模式" ) concat_direction = gr.Dropdown( choices=["自動 (根據調整模式)", "水平拼接", "垂直拼接"], value="自動 (根據調整模式)", label="拼接方向" ) with gr.Row(): spacing = gr.Slider(0, 50, value=2, step=1, label="圖片間距 (像素)") bg_color = gr.Dropdown( choices=["white", "black", "gray"], value="white", label="背景顏色" ) jpg_quality = gr.Slider(60, 100, value=95, step=5, label="JPG 輸出品質 (%)") process_btn = gr.Button("🚀 開始處理", variant="primary", size="lg") with gr.Column(scale=2): gr.Markdown("### 🎯 處理結果") result_image = gr.Image(type="pil", label="拼接結果") result_info = gr.Textbox(label="處理信息", lines=6, max_lines=10) result_file = gr.File(label="下載 JPG 檔案") process_btn.click( fn=process_images, inputs=[image1, image2, image3, image4, image5, image6, resize_mode, concat_direction, spacing, bg_color, jpg_quality], outputs=[result_image, result_info, result_file] ) return iface if __name__ == "__main__": app = create_interface() app.launch(server_name="0.0.0.0", server_port=7860, share=True)