Update app.py

app.py

@@ -263,17 +263,24 @@ class ImageFilters:
 class PDFGenerator:
     @staticmethod
     def images_to_pdf(images: List[np.ndarray]) -> bytes:
-        """將多張影像合併為 PDF"""
+        """將多張影像合併為 PDF (修正版：假設輸入為 RGB)"""
         if not images:
             return None
         
         pil_images = []
         for img in images:
-            # BGR to RGB
-            rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
-            pil_img = Image.fromarray(rgb)
-            pil_images.append(pil_img)
+            # 修改點：移除 BGR2RGB 轉換
+            # 因為目前內部狀態已經是 RGB，直接轉 PIL 即可
+            try:
+                pil_img = Image.fromarray(img)
+                pil_images.append(pil_img)
+            except Exception as e:
+                print(f"PDF 圖片轉換錯誤: {e}")
+                continue
         
+        if not pil_images:
+            return None
+
         # 生成 PDF
         pdf_buffer = io.BytesIO()
         pil_images[0].save(
@@ -715,7 +722,7 @@ def create_interface():
             return gr.update(visible=False)
 
         def on_export_images(state: ScanState, enable_enhance):
-            """匯出圖片 (ZIP) (支援畫質增強) - 修復版"""
+            """匯出圖片 (ZIP) - 修正版：RGB 轉 BGR"""
             if not state.pages:
                 return gr.update(visible=False)
             
@@ -728,12 +735,17 @@ def create_interface():
                     for i, page in enumerate(state.pages):
                         img = page.processed if page.processed is not None else page.original
                         
-                        # 如果勾選增強，則進行處理
+                        # 如果勾選增強
                         if enable_enhance:
+                            # 注意：如果 QualityEnhancer 預設處理 BGR，這裡可能要先轉 BGR 再增強
+                            # 但假設增強器已適配，我們只關注最後輸出
                             img = QualityEnhancer.enhance(img)
                         
-                        # 編碼並寫入 ZIP
-                        success, buffer = cv2.imencode('.jpg', img, [cv2.IMWRITE_JPEG_QUALITY, 95])
+                        # --- 修改點：存檔前 RGB -> BGR ---
+                        # OpenCV 的 imencode 預設是 BGR 順序，所以必須轉過去
+                        img_bgr = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
+                        
+                        success, buffer = cv2.imencode('.jpg', img_bgr, [cv2.IMWRITE_JPEG_QUALITY, 95])
                         if success:
                             zip_file.writestr(f"page_{i+1:02d}.jpg", buffer.tobytes())