import io
import fitz
from PIL import Image
path = 'Test.pdf'
doc = fitz.open(path, filetype="pdf")
page_count = doc.page_count
if page_count:
 for page_no in range(page_count):
 blocks = doc[page_no].getText('dict')['blocks']
 for ind, block in enumerate(blocks):
 if block['type'] == 1:
 try:
 image = Image.open(io.BytesIO(block['image']))
 image.save(open(f"test.{block['ext']}", "wb"))
 except Exception as e:
 print(e)

pix = fitz.Pixmap(doc, 1088)
mask = fitz.Pixmap(doc, 6181)
pix.set_alpha(mask.samples)

pix1088 = fitz.Pixmap(doc,1088)
mask = fitz.Pixmap(doc, 6181)
if pix1088.alpha:
 temp = fitz.Pixmap(pix1088, 0) # make temp pixmap w/o the alpha
 pix1088 = None # release storage
 pix1088 = temp
pix = fitz.Pixmap(pix1088, mask) # now compose final pixmap
pix.save("image1088.png")

for page_num in range(len(doc)):
 page = doc[page_num]
 images = page.get_images(full=True)
 print(f"Page {page_num + 1}: Found {len(images)} images")
 for img_idx, img in enumerate(images):
 try:
 xref = img[0]
 # Get the image XObject dictionary
 img_dict = doc.xref_object(xref, compressed=True)
 pix = fitz.Pixmap(doc, xref)
 # Try to get the soft mask (transparency mask)
 smask_xref = None
 for line in img_dict.splitlines():
 if "/SMask" in line:
 # Extract the xref number after /SMask
 # Example line: "/SMask 123 0 R"
 parts = line.strip().split()
 if len(parts) >= 2 and parts[0] == "/SMask":
 try:
 smask_xref = int(parts[1])
 break
 except:
 pass
 
 if smask_xref:
 # Extract main image
 img_pix = fitz.Pixmap(doc, xref)
 img_np = np.frombuffer(img_pix.samples, dtype=np.uint8)
 img_np = img_np.reshape((img_pix.height, img_pix.width, img_pix.n))
 img_pix = None
 # Extract mask image
 mask_pix = fitz.Pixmap(doc, smask_xref)
 mask_np = np.frombuffer(mask_pix.samples, dtype=np.uint8)
 mask_np = mask_np.reshape((mask_pix.height, mask_pix.width))
 mask_pix = None
 # Combine image + alpha mask into RGBA
 if img_np.shape[2] == 3:
 rgba_np = np.dstack((img_np, mask_np))
 else:
 rgba_np = img_np # fallback
 pil_img = Image.fromarray(rgba_np, mode="RGBA")
 else:
 # Convert CMYK or grayscale to RGB if needed
 if pix.n >= 4 or pix.alpha or pix.colorspace != fitz.csRGB:
 pix = fitz.Pixmap(fitz.csRGB, pix)
 img_bytes = pix.tobytes("png")
 #pix = None
 pil_img = Image.open(BytesIO(img_bytes))
 # Flatten alpha if present
 if pil_img.mode in ("RGBA", "LA"):
 background = Image.new("RGB", pil_img.size, (255, 255, 255))
 background.paste(pil_img, mask=pil_img.getchannel("A")) # Use the last channel as alpha mask
 pil_img = background
 else:
 pil_img = pil_img.convert("RGB")
 # Skip small images
 if pil_img.width < 300 or pil_img.height < 300:
 continue
 # Skip near blank images
 stat = ImageStat.Stat(pil_img)
 if max(stat.stddev) < 1.0:
 continue
 image_list.append(pil_img)
 except Exception as e:
 print(f"Error processing image {img_idx + 1} on page {page_num + 1}: {e}")
if image_list:
 image_list[0].save(
 output_pdf_path,
 save_all=True,
 append_images=image_list[1:],
 resolution=100.0
 )
 print(f"Saved {len(image_list)} images into '{output_pdf_path}'")
else:
 print("No valid images found to save.")