|
|
import gradio as gr |
|
|
import os |
|
|
from pdfitdown.pdfconversion import Converter |
|
|
|
|
|
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: |
|
|
""" |
|
|
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. |
|
|
""" |
|
|
output_path = image_file.name.replace(os.path.splitext(image_file.name)[1], '.pdf') |
|
|
converter.convert(image_file.name, output_path) |
|
|
return output_path |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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", file_types=["image"]), |
|
|
outputs=gr.File(label="Converted PDF"), |
|
|
title="Image to PDF Converter", |
|
|
description="Convert your images to PDF format" |
|
|
) |
|
|
|
|
|
|
|
|
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) |
