Spaces:
Sleeping
Sleeping
File size: 2,373 Bytes
20185a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import gradio as gr
import fitz
import numpy as np
import tempfile
import os
def parse_page_selection(selection):
pages = set()
for part in selection.split(','):
if '-' in part:
start, end = map(int, part.split('-'))
pages.update(range(start - 1, end)) # Convert to 0-based index
else:
pages.add(int(part) - 1) # Convert to 0-based index
return sorted(pages)
def invert_colors(pix):
img = np.frombuffer(pix.samples, dtype=np.uint8)
img = 255 - img # Invert colors
return fitz.Pixmap(pix.colorspace, pix.width, pix.height, img.tobytes(), pix.alpha)
def invert_pdf_pages(input_pdf, output_pdf, page_selection):
pages_to_invert = parse_page_selection(page_selection)
doc = fitz.open(input_pdf)
for page_num in pages_to_invert:
if 0 <= page_num < len(doc):
page = doc[page_num]
pix = page.get_pixmap()
inverted_pix = invert_colors(pix)
img_rect = page.rect
page.clean_contents()
page.insert_image(img_rect, stream=inverted_pix.tobytes())
doc.save(output_pdf)
doc.close()
def invert_pdf_document(input_file, page_selection):
# If input_file is a file-like object, write its content to a temp file.
# Otherwise, assume it's a file path.
if hasattr(input_file, "read"):
with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_in:
tmp_in.write(input_file.read())
tmp_in_path = tmp_in.name
cleanup = True
else:
tmp_in_path = input_file
cleanup = False
# Create a temporary file to save the inverted PDF.
with tempfile.NamedTemporaryFile(delete=False, suffix=".pdf") as tmp_out:
tmp_out_path = tmp_out.name
invert_pdf_pages(tmp_in_path, tmp_out_path, page_selection)
# Clean up temporary input file if it was created.
if cleanup:
os.remove(tmp_in_path)
return tmp_out_path
iface = gr.Interface(
fn=invert_pdf_document,
inputs=[
gr.File(label="Input PDF"),
gr.Textbox(label="Page Selection", value="1-12,14-20,22-32,56,66-78,82-97")
],
outputs=gr.File(label="Output PDF"),
title="PDF Color Inverter",
description="Upload a PDF and specify the pages to invert colors."
)
if __name__ == "__main__":
iface.launch(share=True) |