Update app.py

app.py

@@ -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}")
 ensure_dir(OUTPUT_DIR)
 
-# PDF轉圖片
+# PDF 轉圖片
 def pdf_to_images(pdf_file):
     images = []
     with fitz.open(pdf_file.name) as doc:
@@ -28,14 +28,14 @@ def pdf_to_images(pdf_file):
             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]
     return bg
 
-# 合成 + 壓縮
+# 主處理流程
 def process_and_combine(images, prefix):
     bg_pil = Image.open(BG_PATH).convert("RGB")
     bg = np.array(bg_pil)
@@ -64,30 +64,25 @@ def process_and_combine(images, prefix):
     return results, ZIP_PATH, f"✅ 共合成 {len(results)} 張圖，已打包 ZIP"
 
 # 主流程
-def process_inputs(files, prefix):
+def process_inputs(file_list, prefix):
     prefix = prefix.strip() or "result_"
     images = []
-    for file in files:
+    for file in file_list:
         name = file.name.lower()
         if name.endswith(".pdf"):
             images.extend(pdf_to_images(file))
         elif name.endswith((".png", ".jpg", ".jpeg")):
-            img = Image.open(file)
-            if img.mode != "RGB":
-                img = img.convert("RGB")
+            img = Image.open(file).convert("RGB")
             images.append(img)
         else:
             raise gr.Error(f"不支援的檔案格式：{name}")
     return process_and_combine(images, prefix)
 
-# Gradio UI with Interface
-def ui_interface(files, prefix):
-    return process_inputs(files, prefix)
-
+# UI
 demo = gr.Interface(
-    fn=ui_interface,
+    fn=process_inputs,
     inputs=[
-        gr.File(label="📎 上傳 PDF 或圖片", file_types=[".pdf", ".png", ".jpg", ".jpeg"], multiple=True),
+        gr.Files(label="📎 上傳 PDF 或圖片", file_types=[".pdf", ".png", ".jpg", ".jpeg"]),
         gr.Textbox(label="檔名前綴（可選）", placeholder="例如：poster_")
     ],
     outputs=[