Spaces:
Runtime error
Runtime error
File size: 3,004 Bytes
63135a7 8fafff8 63135a7 4c1af0e 80d71cb 4c1af0e 80d71cb ee4ec2a 7f4a2af ee4ec2a 4c1af0e 42fb932 4c1af0e 42fb932 80d71cb 4c1af0e 42fb932 4c1af0e 8fafff8 63135a7 4c1af0e 63135a7 f1ded69 e426ea3 f3bd71a e426ea3 f3bd71a e426ea3 63135a7 8fafff8 63135a7 f1ded69 |
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 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 |
import os
from jinja2 import Environment, FileSystemLoader, select_autoescape
from get_paperinfo_fromurls import get_paperinfo_fromurls
import gradio as gr
class CARDS_TEMPLATE(object):
def __init__(self, path_to_template, template_filename):
self.path_to_template = path_to_template
self.template_filename = template_filename
self.template = self._get_template()
self.rendered_html = None
def _get_template(self):
env = Environment(
autoescape=select_autoescape(
enabled_extensions=('html'),
default_for_string=True,
),
loader=FileSystemLoader(self.path_to_template)
)
return env.get_template(self.template_filename)
def render(self, paper_details_iterator):
self.rendered_html = self.template.render(paper_details=paper_details_iterator)
def save_html(self, output_dir=None, output_htmlfile=None):
with open(os.path.join(output_dir, output_htmlfile), "w") as f:
f.write(self.rendered_html)
template_file = "htmlcard.html"
template_path = ""
card_template = CARDS_TEMPLATE(
path_to_template = template_path,
template_filename = template_file,
)
CSS = """
#url-textbox {
padding: 0 !important;
font-size: 16px;
}
.gradio-container {
background-color: transparent;
}
.gradio-container .gr-button-primary {
background: #b31b1b;
border: 1px solid #b31b1b;
border-radius: 8px;
color: white;
font-weight: bold;
font-size: 16px;
}
#htel {
justify-content: center;
}
"""
def create_html_card(arxiv_link):
paper_details = get_paperinfo_fromurls(arxiv_link)
card_template.render(paper_details_iterator=paper_details)
return card_template.rendered_html
demo = gr.Blocks(css=CSS)
with demo:
gr.Markdown("# DocQuery: Document Query Engine")
gr.Markdown(
"DocQuery (created by [Impira](https://impira.com)) uses LayoutLMv1 fine-tuned on DocVQA, a document visual question"
" answering dataset, as well as SQuAD, which boosts its English-language comprehension."
" To use it, simply upload an image or PDF, type a question, and click 'submit', or "
" click one of the examples to load them."
" DocQuery is MIT-licensed and available on [Github](https://github.com/impira/docquery)."
)
with gr.Column():
with gr.Row():
text = gr.Textbox(
show_label=False,
placeholder="Paste arXiv link (abs of pdf)",
lines=1,
max_lines=1,
elem_id="url-textbox",
)
button = gr.Button("Generate", variant="primary")
with gr.Row():
card = gr.HTML(elem_id="htel")
button.click(
fn=create_html_card,
inputs=[text],
outputs=[card]
)
if __name__ == "__main__":
demo.launch() |