| import fitz | |
| import os | |
| pdf1_path = r"C:\Users\admin\Downloads\BẢNG GIÁ IN NHANH 14.03.2026.pdf" | |
| pdf2_path = r"C:\Users\admin\Downloads\BẢNG GIÁ IN GIA CÔNG 26.02.2026.pdf" | |
| os.makedirs("scripts/extracted_images", exist_ok=True) | |
| # Extract PDF 1 Page 1 | |
| doc1 = fitz.open(pdf1_path) | |
| page1 = doc1[0] | |
| pix1 = page1.get_pixmap(dpi=150) | |
| pix1.save("scripts/extracted_images/in_nhanh_page1.png") | |
| print("Saved in_nhanh_page1.png") | |
| # Extract PDF 2 Page 2 | |
| doc2 = fitz.open(pdf2_path) | |
| page2 = doc2[1] # 0-indexed page 2 | |
| pix2 = page2.get_pixmap(dpi=150) | |
| pix2.save("scripts/extracted_images/in_gia_cong_page2.png") | |
| print("Saved in_gia_cong_page2.png") | |
| # Also let's extract page 1 and page 3 of PDF 2 just in case | |
| doc2[0].get_pixmap(dpi=100).save("scripts/extracted_images/in_gia_cong_page1.png") | |
| doc2[2].get_pixmap(dpi=100).save("scripts/extracted_images/in_gia_cong_page3.png") | |
| print("Saved in_gia_cong_page1.png and page3.png") | |