import gradio as gr from description import * from reference_string_parsing import * from controlled_summarization import * from dataset_extraction import * from controlled_summarization import recommended_kw import requests # Example Usage # url = "https://arxiv.org/pdf/2305.14996.pdf" # dest_folder = "./examples/" # download_pdf(url, dest_folder) with gr.Blocks(css="#htext span {white-space: pre-line}") as demo: gr.Markdown("# Gradio Demo for SciAssist") with gr.Tabs(): # Controlled Summarization with gr.TabItem("Controlled Summarization"): with gr.Box(): gr.Markdown(ctrlsum_file_md) with gr.Row(): with gr.Column(): ctrlsum_url = gr.Textbox(label="PDF URL", max_lines=1) ctrlsum_file = gr.File(label="Input File") ctrlsum_str = gr.TextArea(label="Input String", max_lines=5) with gr.Column(): # ctrlsum_file_beams = gr.Number(label="Number of beams for beam search", value=1, precision=0) # ctrlsum_file_sequences = gr.Number(label="Number of generated summaries", value=1, precision=0) ctrlsum_file_length = gr.Radio(label="Length", value=0, choices=[0, 50, 100, 200], info="Set the maximum limit for the length of text generated in the output summary. Length 0 implies no restriction on the summary length.") kw = gr.Radio(visible=False) ctrlsum_file_keywords = gr.Textbox(label="Keywords", max_lines=1) with gr.Row(): ctrlsum_file_btn = gr.Button("Generate") ctrlsum_file_output = gr.Textbox( elem_id="htext", label="Summary", show_label=True, show_copy_button=True ) ctrlsum_file_examples = gr.Examples( examples=[["examples/H01-1042_body.txt", 50, "automatic evaluation technique", "", ""], ["examples/H01-1042.pdf", 0, "automatic evaluation technique", "", ""]], inputs=[ctrlsum_file, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_str, ctrlsum_url ]) ctrlsum_file_btn.click( fn=ctrlsum_for_file, inputs=[ctrlsum_file, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_str, ctrlsum_url], outputs=[ctrlsum_file_output, ctrlsum_str, ctrlsum_file] ) def clear(): return None, 0, None, None, gr.Radio(visible=False) def update_url(url): if url in recommended_kw.keys(): keywords = recommended_kw[url] if keywords != None: return None, None, gr.Radio(choices=keywords[:3], label="Recommended Keywords", visible=True, interactive=True) return None, None, gr.Radio(visible=False) ctrlsum_file.upload(clear, inputs=None, outputs=[ctrlsum_str, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_url, kw]) ctrlsum_url.input(update_url, inputs=ctrlsum_url, outputs=[ctrlsum_str, ctrlsum_file, kw]) ctrlsum_str.input(clear, inputs=None, outputs=[ctrlsum_url, ctrlsum_file_length, ctrlsum_file_keywords, ctrlsum_file, kw]) def select_kw(env: gr.SelectData): return env.value kw.select(select_kw, None, ctrlsum_file_keywords) # Reference String Parsing with gr.TabItem("Reference String Parsing"): gr.Markdown(rsp_title_md) with gr.Box(): gr.Markdown(rsp_str_md) with gr.Row(): with gr.Column(): rsp_str = gr.Textbox(label="Input String") with gr.Column(): rsp_str_dehyphen = gr.Checkbox(label="dehyphen") with gr.Row(): rsp_str_btn = gr.Button("Parse") rsp_str_output = gr.HighlightedText( elem_id="htext", label="The Result of Parsing", combine_adjacent=True, adjacent_separator=" ", ) rsp_str_examples = gr.Examples(examples=[[ "Waleed Ammar, Matthew E. Peters, Chandra Bhagavat- ula, and Russell Power. 2017. The ai2 system at semeval-2017 task 10 (scienceie): semi-supervised end-to-end entity and relation extraction. In ACL workshop (SemEval).", True], [ "Isabelle Augenstein, Mrinal Das, Sebastian Riedel, Lakshmi Vikraman, and Andrew D. McCallum. 2017. Semeval-2017 task 10 (scienceie): Extracting keyphrases and relations from scientific publications. In ACL workshop (SemEval).", False]], inputs=[rsp_str, rsp_str_dehyphen]) with gr.Box(): gr.Markdown(rsp_file_md) with gr.Row(): with gr.Column(): rsp_file = gr.File(label="Input File") rsp_file_dehyphen = gr.Checkbox(label="dehyphen") with gr.Row(): rsp_file_btn = gr.Button("Parse") rsp_file_output = gr.HighlightedText( elem_id="htext", label="The Result of Parsing", combine_adjacent=True, adjacent_separator=" ", ) rsp_file_examples = gr.Examples(examples=[["examples/N18-3011_ref.txt", False],["examples/BERT_paper.pdf", True]], inputs=[rsp_file, rsp_file_dehyphen]) rsp_file_btn.click( fn=rsp_for_file, inputs=[rsp_file, rsp_file_dehyphen], outputs=rsp_file_output ) rsp_str_btn.click( fn=rsp_for_str, inputs=[rsp_str, rsp_str_dehyphen], outputs=rsp_str_output ) # Dataset Extraction with gr.TabItem("Dataset Mentions Extraction"): gr.Markdown(de_title_md) with gr.Box(): gr.Markdown(de_str_md) with gr.Row(): with gr.Column(): de_str = gr.Textbox(label="Input String") with gr.Row(): de_str_btn = gr.Button("Extract") de_str_output = gr.HighlightedText( elem_id="htext", label="The Result of Extraction", combine_adjacent=True, adjacent_separator=" ", ) de_str_examples = gr.Examples(examples=[["The impact of gender identity on emotions was examined by researchers using a subsample from the National Longitudinal Study of Adolescent Health. The study aimed to investigate the direct effects of gender identity on emotional experiences and expression. By focusing on a subsample of the larger study, the researchers were able to hone in on the specific relationship between gender identity and emotions. Through their analysis, the researchers sought to determine whether gender identity could have a significant and direct impact on emotional well-being. The findings of the study have important implications for our understanding of the complex interplay between gender identity and emotional experiences, and may help to inform future interventions and support for individuals who experience gender-related emotional distress."], ["The possibility of genotype-environment interaction for memory performance and change was examined in 150 monozygotic twin pairs from the Swedish Adoption Twin Study of Aging and the National Comorbidity Survey. They aimed to explore how genetic and environmental factors could interact to affect cognitive performance in aging individuals. Through their analysis, the researchers hoped to gain a better understanding of the complex interplay between nature and nurture in determining cognitive outcomes. By investigating the unique characteristics of monozygotic twins, who share identical genetic material, the study was able to isolate the role of environmental factors in shaping cognitive abilities over time. The findings from this research have important implications for our understanding of the complex interplay between genetics and the environment in shaping cognitive outcomes in aging individuals."]], inputs=[de_str]) with gr.Box(): gr.Markdown(de_file_md) with gr.Row(): with gr.Column(): de_file = gr.File(label="Input File") with gr.Row(): de_file_btn = gr.Button("Extract") de_file_output = gr.HighlightedText( elem_id="htext", label="The Result of Extraction", combine_adjacent=True, adjacent_separator=" ", ) de_file_examples = gr.Examples(examples=[["examples/127.txt"]], inputs=[de_file]) de_file_btn.click( fn=de_for_file, inputs=[de_file], outputs=de_file_output ) de_str_btn.click( fn=de_for_str, inputs=[de_str], outputs=de_str_output ) demo.queue().launch(share=False)