Spaces:
Runtime error
Runtime error
File size: 2,987 Bytes
63135a7 8fafff8 63135a7 4c1af0e 80d71cb 4c1af0e 80d71cb ee4ec2a 4c1af0e 42fb932 4c1af0e 42fb932 80d71cb 4c1af0e 42fb932 4c1af0e 8fafff8 63135a7 4c1af0e 63135a7 f1ded69 e426ea3 ee4ec2a 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="URL",
lines=1,
max_lines=1,
elem_id="url-textbox",
)
button = gr.Button("Generate card", 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() |