Update app.py

app.py

@@ -31,14 +31,16 @@ def parse_page_range(range_str):
     return sorted(pages)
 
 # PDF轉圖片
-def pdf_to_images(pdf_file, selected_pages=None):
+def pdf_to_images(pdf_file, start_page=None, end_page=None):
     images = []
     with fitz.open(pdf_file.name) as 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)
+        total_pages = len(doc)
+        start = max(0, (start_page or 1) - 1)
+        end = min(end_page or total_pages, total_pages)
+        for page_num in range(start, end):
+            pix = doc[page_num].get_pixmap(matrix=fitz.Matrix(2, 2), colorspace=fitz.csRGB)
+            img = np.frombuffer(pix.samples, dtype=np.uint8).reshape((pix.height, pix.width, 3))
+            img = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)  # ⭐ 修正色彩順序
             images.append(img)
     return images