Update app.py

app.py

@@ -2,7 +2,6 @@
 import gradio as gr
 from PIL import Image
 import io
-import numpy as np
 
 
 def merge_images_to_a4(img1, img2, img3=None):
@@ -24,7 +23,7 @@ def merge_images_to_a4(img1, img2, img3=None):
     valid_images = [img for img in [img1, img2, img3] if img is not None]
     
     if len(valid_images) == 0:
-        return a4_image, None  # Return blank page
+        return a4_image  # Return blank page
     
     elif len(valid_images) == 1:
         # Only 1 image: Center display
@@ -72,12 +71,7 @@ def merge_images_to_a4(img1, img2, img3=None):
             y = (A4_HEIGHT - img_resized.height) // 2
             a4_image.paste(img_resized, (x, y))
     
-    # Save as JPG to bytes for download
-    jpg_bytes = io.BytesIO()
-    a4_image.save(jpg_bytes, format='JPEG', quality=95)
-    jpg_bytes.seek(0)
-    
-    return a4_image, jpg_bytes.getvalue()
+    return a4_image
 
 
 def resize_to_fit(img, max_width, max_height):
@@ -108,7 +102,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         **使用說明：**
         - 上傳 **2 張圖片** → 左右並排
         - 上傳 **3 張圖片** → 橫向排列（3 列 1 行，從左到右）
-        - 點擊「下載 JPG」按鈕取得高品質 JPG 檔案
+        - 合併後可直接右鍵儲存圖片（JPG 格式，高品質）
         """
     )
     
@@ -121,17 +115,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     
     image_output = gr.Image(
         type="pil", 
-        label="✅ 合併結果預覽 (A4 橫向)", 
-        interactive=False
+        label="✅ 合併結果 (A4 橫向 - JPG 格式)", 
+        interactive=False,
+        format="jpg"  # Force JPG output
     )
     
-    download_file = gr.File(label="📥 下載 JPG 檔案", interactive=False)
-    
     # Button click event
     merge_button.click(
         fn=merge_images_to_a4,
         inputs=[image_input1, image_input2, image_input3],
-        outputs=[image_output, download_file]
+        outputs=image_output
     )
     
     gr.Markdown(