| import os | |
| import gradio as gr | |
| import fitz | |
| def merge(pdfs): | |
| os.makedirs("out", exist_ok = True) | |
| os.chdir("./out") | |
| result = fitz.open() | |
| for pdf in pdfs: | |
| with fitz.open(pdf) as mfile: | |
| result.insert_pdf(mfile) | |
| result.save("merge.pdf") | |
| file = "merge.pdf" | |
| return file | |
| demo = gr.Interface( | |
| fn = merge, | |
| inputs = gr.Files(file_types = ["text",".pdf"]), | |
| outputs = "file", | |
| theme = 'nuttea/Softblue', | |
| allow_flagging = "never" | |
| ) | |
| demo.queue(concurrency_count = 10) | |
| demo.launch() | |