File size: 1,352 Bytes
9965554 741685e 9965554 741685e 9965554 741685e 9965554 |
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 |
import gradio as gr
import fitz
from PIL import Image
import io
def pdf_page_to_image(pdf_path, page_number):
# Open the PDF file
doc = fitz.open(pdf_path)
page_number -= 1
# Check if the page number is valid
if page_number < 0 or page_number >= len(doc):
raise gr.Error("Page number out of range")
page = doc.load_page(page_number)
pix = page.get_pixmap()
img_data = pix.tobytes("png") # Get the PNG image data
img = Image.open(io.BytesIO(img_data))
doc.close()
return img, {"example": "value"}
def get_pdf_length(pdf_path):
return len(fitz.open(pdf_path))
def update_number(pdf_path):
num_pages = get_pdf_length(pdf_path)
return gr.Number(info=f"Enter a number between 1 and {num_pages}", maximum=num_pages)
with gr.Blocks() as demo:
with gr.Row():
with gr.Column():
f = gr.File(label="Upload files", file_types=[".pdf"])
n = gr.Number(1, minimum=1, label="Page number")
b = gr.Button("Submit", variant="primary")
with gr.Column():
i = gr.Image()
o = gr.JSON(label="Prediction")
e = gr.Textbox(label="Edited prediction")
f.upload(update_number, f, n)
n.submit(pdf_page_to_image, [f, n], [i, o])
b.click(pdf_page_to_image, [f, n], [i, o])
demo.launch() |