File size: 2,406 Bytes
7b19e66 7888f8c bb2ddfa 72423ad 7b19e66 df1c5ff 7888f8c df1c5ff e40d876 df1c5ff e40d876 df1c5ff e40d876 df1c5ff 7b19e66 79a26ca e40d876 bb2ddfa 651837d 304bb90 79a26ca c52be9c bb2ddfa 7888f8c 7b19e66 df1c5ff 7b19e66 df1c5ff 7b19e66 df1c5ff c52be9c bb2ddfa 7b19e66 df1c5ff 7b19e66 8643c37 |
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 74 75 76 77 78 79 80 81 |
import gradio as gr
import os
from pdfitdown.pdfconversion import Converter
import fitz
from typing import List
from PIL import Image
from loadimg import load_img
import io
converter = Converter()
def convert_file_to_pdf(filename:str) -> str:
"""
Converts a markdown file to PDF format.
Args:
filename: str
The path to the markdown file to be converted.
Returns:
str: The file path of the generated PDF file.
"""
output_path = filename.name.rsplit('.', 1)[0] + '.pdf'
converter.convert(filename.name, output_path)
return output_path
def convert_file_to_img(image_file:str=None,txt:str="") -> List[Image.Image] :
"""
Convert an image file to PDF format.
Args:
image_file: A file object containing the image to be converted.
The file must be in a format supported by the converter
(e.g., PNG, JPG, JPEG).
Returns:
str: The file path of the generated PDF file. The output filename will be
the same as the input filename but with a .pdf extension.
"""
img_list = []
if txt != "":
img_list.append(load_img(txt,output_type="pil"))
if image_file is not None:
output_path = image_file.name.rsplit('.', 1)[0] + '.pdf'
converter.convert(image_file.name, output_path)
doc = fitz.open(output_path)
for page in doc:
page_bytes = page.get_pixmap().tobytes("png")
img_list.append(load_img(Image.open(io.BytesIO(page_bytes))).convert("RGB"))
doc.close()
return img_list
# Create individual interfaces
file_to_pdf = gr.Interface(
fn=convert_file_to_pdf,
inputs=gr.File(label="Upload README/Markdown file"),
outputs=gr.File(label="Converted PDF"),
title="File to PDF Converter",
description="Convert your files to PDF format"
)
image_to_pdf = gr.Interface(
fn=convert_file_to_img,
inputs=[gr.File(label="Upload Image"),gr.Textbox(label="base64, url")],
outputs=gr.Gallery(label="Converted Images"),
title="File to Images Converter",
description="Convert your images to an image format"
)
# Create tabbed interface
demo = gr.TabbedInterface(
[file_to_pdf, image_to_pdf],
["File to PDF", "File to Image"],
)
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860, debug=True, mcp_server=True) |