Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import fitz  # PyMuPDF
 from PIL import Image
 
-# 參數設定
+# 設定常數
 OUTPUT_DIR = "output"
 ZIP_PATH = os.path.join(OUTPUT_DIR, "results.zip")
 BG_PATH = "background.jpg"
@@ -11,14 +11,14 @@ TARGET_WIDTH, TARGET_HEIGHT = 2133, 1200
 POS1 = (1032, 0)
 POS2 = (4666, 0)
 
-# 確保目錄可寫入
+# 檢查並建立資料夾
 def ensure_dir(path):
     os.makedirs(path, exist_ok=True)
     if not os.access(path, os.W_OK):
-        raise RuntimeError(f"❌ 無法寫入資料夾：{path}")
+        raise RuntimeError(f"❌ 無法寫入：{path}")
 ensure_dir(OUTPUT_DIR)
 
-# PDF轉圖片
+# PDF 轉圖片（PyMuPDF）
 def pdf_to_images(file):
     images = []
     doc = fitz.open(file.name)
@@ -28,7 +28,7 @@ def pdf_to_images(file):
         images.append(img)
     return images
 
-# 合成
+# 圖像合成功能
 def overlay_image(bg, fg_img, x, y):
     fg = cv2.resize(np.array(fg_img), (TARGET_WIDTH, TARGET_HEIGHT), interpolation=cv2.INTER_AREA)
     if fg.shape[2] == 4:
@@ -41,11 +41,11 @@ def overlay_image(bg, fg_img, x, y):
         bg[y:y+TARGET_HEIGHT, x:x+TARGET_WIDTH] = fg
     return bg
 
-# 合成與輸出
+# 主處理邏輯
 def process_and_combine(images, prefix):
     bg = cv2.imread(BG_PATH)
     if bg is None:
-        raise gr.Error("❌ 找不到 background.jpg")
+        raise gr.Error("❌ 無法讀取 background.jpg")
 
     for f in os.listdir(OUTPUT_DIR):
         if f.endswith(".jpg"):
@@ -63,46 +63,45 @@ def process_and_combine(images, prefix):
             gc.collect()
 
     if not results:
-        raise gr.Error("⚠️ 沒有生成圖片")
+        raise gr.Error("⚠️ 沒有產生圖片")
 
     with zipfile.ZipFile(ZIP_PATH, "w", zipfile.ZIP_DEFLATED) as zipf:
         for f in sorted(os.listdir(OUTPUT_DIR)):
             if f.endswith(".jpg"):
                 zipf.write(os.path.join(OUTPUT_DIR, f), arcname=f)
 
-    return results, ZIP_PATH, f"✅ 合成完成，共 {len(results)} 張圖片"
+    return results, ZIP_PATH, f"✅ 共合成 {len(results)} 張圖片，已打包 ZIP"
 
-# 主函數
+# 處理輸入
 def process_inputs(files, prefix):
     prefix = prefix.strip() or "result_"
     images = []
-
     for file in files:
         name = file.name.lower()
         if name.endswith(".pdf"):
             images.extend(pdf_to_images(file))
-        elif name.endswith((".jpg", ".jpeg", ".png")):
+        elif name.endswith((".png", ".jpg", ".jpeg")):
             img = Image.open(file).convert("RGBA")
             images.append(img)
         else:
             raise gr.Error(f"不支援的檔案格式：{name}")
     return process_and_combine(images, prefix)
 
-# Gradio 舊版介面
-iface = gr.Interface(
-    fn=process_inputs,
-    inputs=[
-        gr.inputs.File(label="📎 上傳 PDF 或圖片", type="file", multiple=True),
-        gr.inputs.Textbox(label="檔名前綴（可選）", default="result_"),
-    ],
-    outputs=[
-        gr.outputs.Gallery(label="🖼️ 合成結果"),
-        gr.outputs.File(label="⬇️ 下載 ZIP"),
-        gr.outputs.Textbox(label="📄 處理日誌")
-    ],
-    title="PDF 圖像合成工具",
-    description="📄🖼️ 支援上傳 PDF 或圖像，合成到固定背景圖中"
-)
+# Gradio Blocks UI
+with gr.Blocks(title="PDF與圖片合成工具") as demo:
+    gr.Markdown("### 📄🖼️ PDF / 圖片 雙輸入合成工具（背景為 7872x1200）")
+
+    with gr.Row():
+        files = gr.Files(label="📎 上傳 PDF 或圖片", file_types=[".pdf", ".jpg", ".jpeg", ".png"])
+        prefix = gr.Textbox(label="輸出檔名前綴", placeholder="例如：poster_", value="result_")
+        run_btn = gr.Button("🚀 開始合成", variant="primary")
+
+    with gr.Row():
+        gallery = gr.Gallery(label="🖼️ 預覽", columns=3, height="auto")
+        zip_file = gr.File(label="⬇️ 下載 ZIP")
+        log = gr.Textbox(label="📄 日誌", interactive=False)
+
+    run_btn.click(fn=process_inputs, inputs=[files, prefix], outputs=[gallery, zip_file, log])
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()