Spaces:
Runtime error
Runtime error
import gradio as gr | |
import urllib.request | |
import subprocess | |
import os | |
import glob | |
def extract_figure(url): | |
# download PDF file from URL | |
urllib.request.urlretrieve(url, "input.pdf") | |
# extract first figure from PDF using pdffigures2 | |
subprocess.run(["java", "-jar", "pdffigures2.jar", "input.pdf", "-m", "figures_"]) | |
all_pngs = glob.glob("*.png") | |
print(all_pngs) | |
# get path to first figure | |
figure_path = "figures_input-Figure1-1.png" | |
# # read first figure from file | |
# with open(figure_path, "rb") as f: | |
# figure_bytes = f.read() | |
# # delete downloaded file and figure file | |
# os.remove("input.pdf") | |
# os.remove(figure_path) | |
# return first figure | |
return figure_path | |
def run(): | |
# define input and output interfaces | |
inputs = gr.inputs.Textbox(label="Enter URL of PDF file:") | |
outputs = gr.outputs.Image(label="First figure in PDF:", type="filepath") | |
# create interface | |
interface = gr.Interface(fn=extract_figure, inputs=inputs, outputs=outputs, title="Extract First Figure from PDF", description="Enter the URL of a PDF file and the first figure in the file will be extracted and displayed.") | |
# launch interface | |
interface.launch(server_name="0.0.0.0", server_port=7860) | |
if __name__ == "__main__": | |
run() | |