Update app.py

app.py

@@ -1,36 +1,40 @@
-import os
 import cv2
 import numpy as np
+import os
 import gradio as gr
-import zipfile
 import gc
-from pdf2image import convert_from_bytes
-
-# ✅ Hugging Face Spaces Docker 可寫目錄
-BASE_DIR = "/tmp/myapp"
-FG_DIR = os.path.join(BASE_DIR, "foregrounds")
-OUTPUT_DIR = os.path.join(BASE_DIR, "output")
-ZIP_PATH = os.path.join(BASE_DIR, "results.zip")
+import zipfile
 
-# 背景圖
-BG_PATH = "background.jpg"
+# 常量設定
+MAX_FILES = 250
+BG_PATH = "background.jpg"  # 預設背景圖，請自行放置
+FG_DIR = "foregrounds"
+OUTPUT_DIR = "output"
+ZIP_PATH = "output/results.zip"
 TARGET_WIDTH, TARGET_HEIGHT = 2133, 1200
 POS1 = (1032, 0)
 POS2 = (4666, 0)
 
+# 初始化資料夾與權限檢查
 def check_write_permission(path):
-    os.makedirs(path, exist_ok=True)
-    if not os.access(path, os.W_OK):
-        raise RuntimeError(f"❌ 無法寫入資料夾：{path}")
+    if not os.path.exists(path):
+        try:
+            os.makedirs(path)
+        except Exception as e:
+            raise RuntimeError(f"❌ 無法建立目錄 '{path}'：{e}")
+    elif not os.access(path, os.W_OK):
+        raise RuntimeError(f"❌ 目錄 '{path}' 存在，但沒有寫入權限！")
 
 check_write_permission(FG_DIR)
 check_write_permission(OUTPUT_DIR)
 
+# 合成圖片
 def overlay_image(bg, fg, x, y):
     h, w = fg.shape[:2]
     bh, bw = bg.shape[:2]
     if x + w > bw or y + h > bh:
         raise ValueError(f"前景圖超出背景邊界：({x},{y},{w},{h}) > 背景({bw},{bh})")
+
     if fg.shape[2] == 4:
         alpha = fg[:, :, 3:] / 255.0
         bg[y:y+h, x:x+w] = alpha * fg[:, :, :3] + (1 - alpha) * bg[y:y+h, x:x+w]
@@ -38,20 +42,7 @@ def overlay_image(bg, fg, x, y):
         bg[y:y+h, x:x+w] = fg[:, :, :3]
     return bg
 
-def process_pdf_to_images(pdf_file):
-    with open(pdf_file.name, "rb") as f:
-        images = convert_from_bytes(f.read(), fmt="jpeg", size=(TARGET_WIDTH, TARGET_HEIGHT))
-    
-    fg_arrays = []
-    for img in images:
-        np_img = np.array(img.convert("RGB"))
-        fg = cv2.cvtColor(np_img, cv2.COLOR_RGB2BGR)
-        fg = cv2.resize(fg, (TARGET_WIDTH, TARGET_HEIGHT))
-        fg = fg.astype(np.uint8)
-        fg = np.dstack([fg, np.ones((TARGET_HEIGHT, TARGET_WIDTH), dtype=np.uint8)*255])  # fake alpha
-        fg_arrays.append(fg)
-    return fg_arrays
-
+# 壓縮成 zip
 def zip_output_files():
     with zipfile.ZipFile(ZIP_PATH, "w", zipfile.ZIP_DEFLATED) as zipf:
         for file in sorted(os.listdir(OUTPUT_DIR)):
@@ -59,63 +50,65 @@ def zip_output_files():
                 zipf.write(os.path.join(OUTPUT_DIR, file), arcname=file)
     return ZIP_PATH
 
-def process_inputs(fg_files, pdf_file):
+# 處理合成流程
+def process_images(fg_files):
     try:
         progress = gr.Progress()
         progress(0, desc="準備中...")
 
+        if len(fg_files) == 0:
+            return [], None, "請至少上傳一張前景圖"
+        if len(fg_files) > MAX_FILES:
+            raise gr.Error(f"最多支援 {MAX_FILES} 張圖片")
+
         bg = cv2.imread(BG_PATH)
         if bg is None:
             raise gr.Error("無法讀取背景圖（請確認 background.jpg 是否存在）")
 
+        # 清除舊結果
         for f in os.listdir(OUTPUT_DIR):
             if f.endswith(".jpg"):
                 os.remove(os.path.join(OUTPUT_DIR, f))
 
-        fg_images = []
+        results = []
+        for i, fg_file in enumerate(fg_files, 1):
+            progress(i / len(fg_files), f"合成第 {i} 張圖...")
 
-        if pdf_file is not None:
-            fg_images = process_pdf_to_images(pdf_file)
-        elif fg_files:
-            for file in fg_files:
-                with open(file.name, 'rb') as f:
-                    data = np.frombuffer(f.read(), np.uint8)
-                fg = cv2.imdecode(data, cv2.IMREAD_UNCHANGED)
+            try:
+                with open(fg_file.name, 'rb') as f:
+                    fg_data = np.frombuffer(f.read(), np.uint8)
+                fg = cv2.imdecode(fg_data, cv2.IMREAD_UNCHANGED)
                 if fg is None:
                     continue
-                fg = cv2.resize(fg, (TARGET_WIDTH, TARGET_HEIGHT))
-                if fg.shape[2] == 3:
-                    fg = np.dstack([fg, np.ones((TARGET_HEIGHT, TARGET_WIDTH), dtype=np.uint8)*255])
-                fg_images.append(fg)
 
-        if not fg_images:
-            return [], None, "請上傳 PDF 或 圖片"
+                fg = cv2.resize(fg, (TARGET_WIDTH, TARGET_HEIGHT), interpolation=cv2.INTER_AREA)
 
-        results = []
-        for i, fg in enumerate(fg_images, 1):
-            progress(i / len(fg_images), f"處理第 {i} 張圖...")
-            result = overlay_image(bg.copy(), fg, *POS1)
-            result = overlay_image(result, fg, *POS2)
-            out_path = os.path.join(OUTPUT_DIR, f"result_{i:03d}.jpg")
-            cv2.imwrite(out_path, result)
-            results.append(out_path)
+                result = overlay_image(bg.copy(), fg, *POS1)
+                result = overlay_image(result, fg, *POS2)
 
-            if i % 10 == 0:
-                gc.collect()
+                output_path = f"{OUTPUT_DIR}/result_{i:03d}.jpg"
+                cv2.imwrite(output_path, result)
+                results.append((output_path, f"Result {i:03d}"))
 
+                if i % 10 == 0:
+                    gc.collect()
+
+            except Exception as e:
+                print(f"處理 {fg_file.name} 失敗: {str(e)}")
+                continue
+
+        progress(1, "完成！")
         zip_file = zip_output_files()
-        return results, zip_file, f"成功處理 {len(results)} 張圖片"
+        return results, zip_file, f"成功處理 {len(results)}/{len(fg_files)} 張圖片，已打包為 ZIP 可下載"
 
     except Exception as e:
         raise gr.Error(f"處理出錯：{str(e)}")
 
-# Gradio UI
+# Gradio 介面
 with gr.Blocks() as demo:
-    gr.Markdown("## 🖼️ 超寬圖片合成工具（支援 PDF 或前景圖）")
+    gr.Markdown("## 🖼️ 寬幅圖片合成工具（背景固定為 7872x1200）")
 
-    with gr.Row():
-        fg_upload = gr.Files(label="上傳前景圖（JPG 或 PNG）", file_types=[".jpg", ".png"])
-        pdf_upload = gr.File(label="或上傳 PDF（每頁轉圖片）", file_types=[".pdf"])
+    fg_upload = gr.Files(label="上傳前景圖（JPG 或 PNG，自動套用背景）", file_types=[".jpg", ".png"])
     btn_run = gr.Button("開始合成", variant="primary")
 
     with gr.Row():
@@ -124,8 +117,8 @@ with gr.Blocks() as demo:
         log_output = gr.Textbox(label="日誌", interactive=False)
 
     btn_run.click(
-        fn=process_inputs,
-        inputs=[fg_upload, pdf_upload],
+        fn=process_images,
+        inputs=fg_upload,
         outputs=[gallery, zip_download, log_output],
         concurrency_limit=4
     )