Update app.py

app.py

@@ -11,34 +11,48 @@ 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}")
 ensure_dir(OUTPUT_DIR)
 
-# PDF 轉圖片
-def pdf_to_images(pdf_file):
+# 解析頁碼範圍字串（如 "1-3,5,7"）
+def parse_page_range(range_str):
+    if not range_str.strip():
+        return None
+    pages = set()
+    for part in range_str.split(","):
+        if '-' in part:
+            start, end = map(int, part.split("-"))
+            pages.update(range(start, end + 1))
+        else:
+            pages.add(int(part))
+    return sorted(pages)
+
+# PDF轉圖片
+def pdf_to_images(pdf_file, selected_pages=None):
     images = []
     with fitz.open(pdf_file.name) as doc:
-        for page in doc:
+        for i, page in enumerate(doc, 1):
+            if selected_pages and i not in selected_pages:
+                continue
             pix = page.get_pixmap(matrix=fitz.Matrix(2, 2), colorspace=fitz.csRGB)
             img = Image.frombytes("RGB", [pix.width, pix.height], pix.samples)
             images.append(img)
     return images
 
-# 合成邏輯
+# 合成圖片
 def overlay_image(bg, fg_img, x, y):
-    fg = fg_img.resize((TARGET_WIDTH, TARGET_HEIGHT), resample=Image.BILINEAR)
-    fg = np.array(fg)
-    bg[y:y+TARGET_HEIGHT, x:x+TARGET_WIDTH] = fg[:, :, :3]
+    fg = cv2.resize(np.array(fg_img), (TARGET_WIDTH, TARGET_HEIGHT), interpolation=cv2.INTER_AREA)
+    bg[y:y+TARGET_HEIGHT, x:x+TARGET_WIDTH] = fg
     return bg
 
-# 主處理流程
+# 處理與合成流程
 def process_and_combine(images, prefix):
-    bg_pil = Image.open(BG_PATH).convert("RGB")
-    bg = np.array(bg_pil)
+    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"):
@@ -49,7 +63,7 @@ def process_and_combine(images, prefix):
         combined = overlay_image(bg.copy(), img, *POS1)
         combined = overlay_image(combined, img, *POS2)
         output_path = os.path.join(OUTPUT_DIR, f"{prefix}{i:03d}.jpg")
-        cv2.imwrite(output_path, cv2.cvtColor(combined, cv2.COLOR_RGB2BGR))
+        cv2.imwrite(output_path, combined)
         results.append((output_path, f"{prefix}{i:03d}"))
         if i % 10 == 0:
             gc.collect()
@@ -61,16 +75,18 @@ def process_and_combine(images, prefix):
         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)} 張圖，已打包 ZIP"
 
-# 主流程
-def process_inputs(file_list, prefix):
+    return results, ZIP_PATH, f"✅ 成功合成 {len(results)} 張圖，已打包 ZIP"
+
+# 主邏輯：上傳處理
+def process_inputs(files, prefix, page_range):
     prefix = prefix.strip() or "result_"
+    selected_pages = parse_page_range(page_range)
     images = []
-    for file in file_list:
+    for file in files:
         name = file.name.lower()
         if name.endswith(".pdf"):
-            images.extend(pdf_to_images(file))
+            images.extend(pdf_to_images(file, selected_pages))
         elif name.endswith((".png", ".jpg", ".jpeg")):
             img = Image.open(file).convert("RGB")
             images.append(img)
@@ -78,20 +94,21 @@ def process_inputs(file_list, prefix):
             raise gr.Error(f"不支援的檔案格式：{name}")
     return process_and_combine(images, prefix)
 
-# UI
+# Gradio UI
 demo = gr.Interface(
     fn=process_inputs,
     inputs=[
         gr.Files(label="📎 上傳 PDF 或圖片", file_types=[".pdf", ".png", ".jpg", ".jpeg"]),
-        gr.Textbox(label="檔名前綴（可選）", placeholder="例如：poster_")
+        gr.Textbox(label="檔名前綴（可選）", placeholder="例如：poster_"),
+        gr.Textbox(label="選擇頁碼（PDF）：例如 1-3,5,7，留空表示全部", placeholder="1-5,7")
     ],
     outputs=[
         gr.Gallery(label="🖼️ 預覽", columns=3, height="auto"),
         gr.File(label="⬇️ 下載 ZIP"),
-        gr.Textbox(label="📄 日誌")
+        gr.Textbox(label="📄 處理日誌")
     ],
     title="PDF 圖片合成工具",
-    description="📄🖼️ 將 PDF 或圖片套用背景合成為長圖，背景為固定寬度 7872x1200。"
+    description="📄🖼️ 將 PDF 或圖片套用固定背景合成為寬圖，背景為 7872x1200，支援 PDF 選擇頁數與多檔處理。"
 )
 
 if __name__ == "__main__":