Upload 2 files

app.py

@@ -0,0 +1,238 @@
+import gradio as gr
+import numpy as np
+from PIL import Image
+import io
+
+def resize_images_to_match(images, resize_mode="height"):
+    """
+    調整圖片尺寸以匹配拼接需求
+    resize_mode: "height" - 垂直邊等長(水平拼接), "width" - 水平邊等長(垂直拼接)
+    """
+    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 []
+    
+    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:  # width mode
+        # 找到最小寬度，保持寬高比調整
+        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"):
+    """
+    拼接圖片
+    direction: "horizontal" - 水平拼接, "vertical" - 垂直拼接
+    spacing: 圖片間距(像素)
+    background_color: 背景顏色
+    """
+    if not images or len(images) == 0:
+        return None
+    
+    # 過濾掉None值
+    valid_images = [img for img in images if img is not None]
+    if not valid_images:
+        return None
+    
+    if len(valid_images) == 1:
+        return valid_images[0]
+    
+    if direction == "horizontal":
+        # 水平拼接
+        total_width = sum(img.size[0] for img in valid_images) + spacing * (len(valid_images) - 1)
+        max_height = max(img.size[1] for img in valid_images)
+        
+        # 創建新圖片
+        result = Image.new('RGB', (total_width, max_height), background_color)
+        
+        x_offset = 0
+        for img in valid_images:
+            # 垂直居中放置
+            y_offset = (max_height - img.size[1]) // 2
+            result.paste(img, (x_offset, y_offset))
+            x_offset += img.size[0] + spacing
+            
+    else:  # vertical
+        # 垂直拼接
+        max_width = max(img.size[0] for img in valid_images)
+        total_height = sum(img.size[1] for img in valid_images) + spacing * (len(valid_images) - 1)
+        
+        # 創建新圖片
+        result = Image.new('RGB', (max_width, total_height), background_color)
+        
+        y_offset = 0
+        for img in valid_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 process_images(image1, image2, image3, image4, image5, image6, 
+                  resize_mode, concat_direction, spacing, bg_color):
+    """
+    處理圖片拼接的主函數
+    """
+    # 收集所有上傳的圖片
+    uploaded_images = [image1, image2, image3, image4, image5, image6]
+    
+    # 過濾掉None值
+    valid_images = [img for img in uploaded_images if img is not None]
+    
+    if len(valid_images) == 0:
+        return None, "❌ 請至少上傳一張圖片"
+    
+    if len(valid_images) == 1:
+        return valid_images[0], f"✅ 只有一張圖片，無需拼接"
+    
+    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, "❌ 拼接失败"
+        
+        # 返回結果和統計信息
+        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} 像素"
+        
+        return result, info
+        
+    except Exception as e:
+        return None, f"❌ 處理過程中出現錯誤: {str(e)}"
+
+# 創建Gradio界面
+def create_interface():
+    with gr.Blocks(title="圖片拼接比較工具", theme=gr.themes.Soft()) as iface:
+        gr.Markdown("""
+        # 🖼️ 圖片拼接比較工具
+        
+        上傳多張PNG/JPG圖片，自動調整尺寸並拼接成比較圖。
+        
+        ### 使用說明：
+        1. **上傳圖片**：最多可上傳6張圖片
+        2. **選擇調整模式**：等高度適合水平拼接，等寬度適合垂直拼接
+        3. **設定參數**：調整間距和背景顏色
+        4. **點擊處理**：生成拼接後的比較圖
+        """)
+        
+        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(
+                        minimum=0, 
+                        maximum=50, 
+                        value=2, 
+                        step=1,
+                        label="圖片間距 (像素)"
+                    )
+                    
+                    bg_color = gr.Dropdown(
+                        choices=["white", "black", "gray", "transparent"],
+                        value="white",
+                        label="背景顏色"
+                    )
+                
+                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)
+        
+        # 綁定處理函數
+        process_btn.click(
+            fn=process_images,
+            inputs=[
+                image1, image2, image3, image4, image5, image6,
+                resize_mode, concat_direction, spacing, bg_color
+            ],
+            outputs=[result_image, result_info]
+        )
+        
+        gr.Markdown("""
+        ### 💡 小貼士：
+        - **等高度模式**：所有圖片調整為相同高度，適合水平對比
+        - **等寬度模式**：所有圖片調整為相同寬度，適合垂直對比  
+        - 支持常見圖片格式：PNG, JPG, JPEG, WEBP
+        - 處理過程保持圖片原始寬高比
+        - 可下載處理後的結果圖片
+        """)
+    
+    return iface
+
+# 啟動應用
+if __name__ == "__main__":
+    app = create_interface()
+    app.launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True
+    )

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+Pillow
+numpy