gradio version update. adding more examples for the task. adding new fields for each paper.

app.py

@@ -53,7 +53,7 @@ def get_similar_paper(
     # print('computing document scores...') 
     #progress(0.5, desc="Computing document scores...")
     # TODO detect duplicate papers?
-    titles, abstracts, paper_urls, doc_scores = compute_document_score(
+    titles, abstracts, paper_urls, doc_scores, paper_years, paper_citations = compute_document_score(
         doc_model, 
         tokenizer,
         title_input,
@@ -68,7 +68,9 @@ def get_similar_paper(
         'titles': titles,
         'abstracts': abstracts,
         'urls': paper_urls,
-        'doc_scores': doc_scores
+        'doc_scores': doc_scores,
+        'years': paper_years,
+        'citations': paper_citations,
     }
     
     # Select top 10 papers to show
@@ -76,6 +78,8 @@ def get_similar_paper(
     abstracts = abstracts[:10]
     doc_scores = doc_scores[:10]
     paper_urls = paper_urls[:10]
+    paper_years = paper_years[:10]
+    paper_citations = paper_citations[:10]
     
     display_title = ['[ %0.3f ] %s'%(s, t) for t, s in zip(titles, doc_scores)]
     end = time.time()
@@ -119,7 +123,9 @@ def get_similar_paper(
             'source_sentences': input_sentences,
             'highlight': word_scores,
             'top_pairs': top_pairs_info,
-            'url': url
+            'url': url,
+            'year': paper_years[aa],
+            'citations': paper_citations[aa],
         }
     
     end = time.time()
@@ -127,12 +133,11 @@ def get_similar_paper(
     print('done in [%0.2f] seconds'%(highlight_time)) 
     
     ## Set up output elements
-    
     ## Components for Initial Part
     result1_desc_value = """
         <h3>Top %d relevant papers by the reviewer <a href="%s" target="_blank">%s</a></h3>
         
-        For each paper, top %d sentence pairs (one from the submission, one from the paper) with the highest relevance scores are shown.
+        For each paper, top %d sentence pairs (one from the submission on the left, one from the paper on the right) with the highest relevance scores are shown.
         
         **<span style="color:black;background-color:#65B5E3;">Blue highlights</span>**: phrases that appear in both sentences.
         """%(int(top_paper_slider), author_id_input, results['name'], int(top_pair_slider))
@@ -161,7 +166,7 @@ def get_similar_paper(
     title_out = """<a href="%s" target="_blank"><h5>%s</h5></a>"""%(url, title)
     aff_score_out = '##### Affinity Score: %s'%aff_score
     result2_desc_value = """
-        ##### Click a paper by %s (left, sorted by affinity scores), and a sentence from the submission (center), to see which parts of the paper are relevant (right).
+        ##### Click a paper by %s (left, sorted by affinity scores), and a sentence from the submission abstract (center), to see which parts of the paper's abstract are relevant (right).
         """%results['name']
     out3 = [
         gr.update(choices=display_title, value=display_title[0],  interactive=True), # set of papers (radio)
@@ -169,9 +174,11 @@ def get_similar_paper(
         gr.update(value=title_out), # paper_title
         gr.update(value=aff_score_out),  # affinity
         gr.update(value=result2_desc_value), # result 2 description (show more section)
-        gr.update(value=1, maximum=len(sent_tokenize(abstracts[0]))), # highlight slider to control
+        gr.update(value=2, maximum=len(sent_tokenize(abstracts[0]))), # highlight slider to control
     ]
     
+    torch.cuda.empty_cache()
+    
     ## Return by adding the State variable info
     return out1 + out2 + out3 + [results]
 
@@ -179,9 +186,12 @@ def setup_outputs(info, top_papers_show, top_num_info_show):
     titles = info['titles']
     doc_scores = info['doc_scores']
     paper_urls = info['urls']
+    paper_years = info['years']
+    paper_citations = info['citations']
     display_title = ['[ %0.3f ] %s'%(s, t) for t, s in zip(info['titles'], info['doc_scores'])]
     title = []
     affinity = []
+    citation_count = []
     sent_pair_score = []
     sent_text_query = []
     sent_text_candidate = []
@@ -191,22 +201,28 @@ def setup_outputs(info, top_papers_show, top_num_info_show):
     for i in range(top_papers_show):
         if i == 0:
             title.append(
-                gr.update(value="""<a href="%s" target="_blank"><h4>%s</h4></a>"""%(paper_urls[i], titles[i]), visible=True)
+                gr.update(value="""<a href="%s" target="_blank"><h4>%s (%s)</h4></a>"""%(paper_urls[i], titles[i], str(paper_years[i])), visible=True)
             )
             affinity.append(
                 gr.update(value="""#### Affinity Score: %0.3f 
                         <div class="help-tip">
-                                <p>Measures how similar the paper's abstract is to the submission abstract.</p>
+                            <p>Measures how similar the paper's abstract is to the submission abstract.</p>
                         </div>
                         """%doc_scores[i], visible=True) # document affinity
             )
+            citation_count.append(
+                 gr.update(value="""#### Citation Count: %d"""%paper_citations[i], visible=True) # document affinity
+            )
         else:
             title.append(
-                gr.update(value="""<a href="%s" target="_blank"><h4>%s</h4></a>"""%(paper_urls[i], titles[i]), visible=True)
+                gr.update(value="""<a href="%s" target="_blank"><h4>%s (%s)</h4></a>"""%(paper_urls[i], titles[i], str(paper_years[i])), visible=True)
             )
             affinity.append(
                 gr.update(value='#### Affinity Score: %0.3f'%doc_scores[i], visible=True) # document affinity
             )
+            citation_count.append(
+                 gr.update(value="""#### Citation Count: %d"""%paper_citations[i], visible=True) # document affinity
+            )
         demarc_lines.append(gr.Markdown.update(visible=True))
 
         # fill in the rest as 
@@ -238,6 +254,7 @@ def setup_outputs(info, top_papers_show, top_num_info_show):
     # mark others not visible
     title += [gr.Markdown.update(visible=False)] * (NUM_PAPERS_SHOW - top_papers_show)
     affinity += [gr.Markdown.update(visible=False)] * (NUM_PAPERS_SHOW - top_papers_show)
+    citation_count += [gr.Markdown.update(visible=False)] * (NUM_PAPERS_SHOW - top_papers_show)
     demarc_lines += [gr.Markdown.update(visible=False)] * (NUM_PAPERS_SHOW - top_papers_show)
     sent_pair_score += [gr.Markdown.update(visible=False)] * (NUM_PAPERS_SHOW - top_papers_show) * NUM_PAIRS_SHOW
     sent_text_query += [gr.Textbox.update(value='', visible=False)] * (NUM_PAPERS_SHOW - top_papers_show) * NUM_PAIRS_SHOW
@@ -249,7 +266,7 @@ def setup_outputs(info, top_papers_show, top_num_info_show):
     assert(len(affinity) == NUM_PAPERS_SHOW)
     assert(len(sent_pair_score) == NUM_PAIRS_SHOW * NUM_PAPERS_SHOW)
 
-    return title, affinity, demarc_lines, sent_pair_score, sent_text_query, sent_text_candidate, sent_hl_query, sent_hl_candidate
+    return title, affinity, citation_count, demarc_lines, sent_pair_score, sent_text_query, sent_text_candidate, sent_hl_query, sent_hl_candidate
 
 def show_more(info):
     # show the interactive part of the app
@@ -259,6 +276,7 @@ def show_more(info):
         gr.update(visible=True), # submission sentences
         gr.update(visible=True), # title row
         gr.update(visible=True), # affinity row
+        gr.update(visible=True), # citation row
         gr.update(visible=True), # highlight legend
         gr.update(visible=True), # highlight slider
         gr.update(visible=True), # highlight abstract
@@ -298,18 +316,21 @@ def change_paper(
     if len(info.keys()) != 0: # if the info is not empty
         source_sents = info[selected_papers_radio]['source_sentences']
         title = info[selected_papers_radio]['title']
+        year = info[selected_papers_radio]['year']
+        citation_count = info[selected_papers_radio]['citations']
         num_sents = info[selected_papers_radio]['num_cand_sents']
         abstract = info[selected_papers_radio]['abstract']
         aff_score = info[selected_papers_radio]['doc_score']
         highlights = info[selected_papers_radio]['highlight']
         url = info[selected_papers_radio]['url']
-        title_out = """<a href="%s" target="_blank"><h5>%s</h5></a>"""%(url, title)
+        title_out = """<a href="%s" target="_blank"><h5>%s (%s)</h5></a>"""%(url, title, str(year))
         aff_score_out = '##### Affinity Score: %s'%aff_score
+        citation_count_out = '##### Citation Count: %s'%citation_count
         idx = source_sents.index(source_sent_choice)
         if highlight_slider <= num_sents:
-            return title_out, abstract, aff_score_out, highlights[str(idx)][str(highlight_slider)], gr.update(value=highlight_slider, maximum=num_sents)
+            return title_out, abstract, aff_score_out, citation_count_out, highlights[str(idx)][str(highlight_slider)], gr.update(value=highlight_slider, maximum=num_sents)
         else: # if the slider is set to more than the current number of sentences, show the max number of highlights
-            return title_out, abstract, aff_score_out, highlights[str(idx)][str(num_sents)], gr.update(value=num_sents, maximum=num_sents)
+            return title_out, abstract, aff_score_out, citation_count_out, highlights[str(idx)][str(num_sents)], gr.update(value=num_sents, maximum=num_sents)
     else:
         return
 
@@ -334,7 +355,7 @@ def change_top_output(top_paper_slider, top_pair_slider, info={}):
     result1_desc_value = """
     <h3>Top %d relevant papers by the reviewer <a href="%s" target="_blank">%s</a></h3>
     
-    For each paper, top %d sentence pairs (one from the submission, one from the paper) with the highest relevance scores are shown.
+    For each paper, top %d sentence pairs (one from the submission on the left, one from the paper on the right) with the highest relevance scores are shown.
     
     **<span style="color:black;background-color:#65B5E3;">Blue highlights</span>**: phrases that appear in both sentences.
     """%(int(top_paper_slider), info['author_url'], info['name'], int(top_pair_slider))
@@ -363,9 +384,9 @@ It is for meta-reviewers, area chairs, program chairs, or anyone who oversees th
 <center><img src="file/tool-img.jpeg" width="70%" alt="general workflow"></center>
 
 #### How does it help?
-A typical meta-reviewer workflow lacks supportive information on **what makes the pre-selected candidate reviewers a good fit** for the submission. Only affinity scores between the reviewer and the paper are shown, without additional detail. 
+A typical meta-reviewer workflow lacks supportive information on **what makes the pre-selected candidate reviewers a good fit** for the submission. Only affinity scores between the reviewer and the paper are shown, without additional details on what makes them similar/different. 
 
-R2P2 provides more information about each reviewer. It searches for the **most relevant papers** among the reviewer's previous publications and **highlights relevant parts** within them. 
+R2P2 provides more information about each reviewer. Given a paper and a reviewer, it searches for the **most relevant papers** among the reviewer's previous publications and **highlights relevant parts** within them. 
     """
     # More details (video, addendum)
     more_details_instruction = """Check out <a href="https://drive.google.com/file/d/1Ex_-cOplBitO7riNGliecFc8H3chXUN-/view?usp=share_link", target="_blank">this video</a> for a quick introduction of what R2P2 is and how it can help. You can find more details <a href="file/details.html", target="_blank">here</a>, along with our privacy policy and disclaimer."""
@@ -374,6 +395,33 @@ R2P2 provides more information about each reviewer. It searches for the **most r
     gr.HTML(more_details_instruction)
     gr.Markdown("""---""")
     
+    # Add main example
+    example_title ="The Toronto Paper Matching System: An automated paper-reviewer assignment system"
+    example_submission = """One of the most important tasks of conference organizers is the assignment of papers to reviewers. Reviewers' assessments of papers is a crucial step in determining the conference program, and in a certain sense to shape the direction of a field. However this is not a simple task: large conferences typically have to assign hundreds of papers to hundreds of reviewers, and time constraints make the task impossible for one person to accomplish. Furthermore other constraints, such as reviewer load have to be taken into account, preventing the process from being completely distributed. We built the first version of a system to suggest reviewer assignments for the NIPS 2010 conference, followed, in 2012, by a release that better integrated our system with Microsoft's popular Conference Management Toolkit (CMT). Since then our system has been widely adopted by the leading conferences in both the machine learning and computer vision communities. This paper provides an overview of the system, a summary of learning models and methods of evaluation that we have been using, as well as some of the recent progress and open issues."""
+    example_reviewer = "https://www.semanticscholar.org/author/Nihar-B.-Shah/1737249"
+    
+    ## Add other examples for the task
+
+    # match 1
+    # example1_title = "VoroCNN: Deep convolutional neural network built on 3D Voronoi tessellation of protein structures"
+    # example1_submission = """Effective use of evolutionary information has recently led to tremendous progress in computational prediction of three-dimensional (3D) structures of proteins and their complexes. Despite the progress, the accuracy of predicted structures tends to vary considerably from case to case. Since the utility of computational models depends on their accuracy, reliable estimates of deviation between predicted and native structures are of utmost importance. Results For the first time we present a deep convolutional neural network (CNN) constructed on a Voronoi tessellation of 3D molecular structures. Despite the irregular data domain, our data representation allows to efficiently introduce both convolution and pooling operations of the network. We trained our model, called VoroCNN, to predict local qualities of 3D protein folds. The prediction results are competitive to the state of the art and superior to the previous 3D CNN architectures built for the same task. We also discuss practical applications of VoroCNN, for example, in the recognition of protein binding interfaces. Availability The model, data, and evaluation tests are available at https://team.inria.fr/nano-d/software/vorocnn/. Contact [email protected], [email protected]"""
+    # example1_reviewer = "https://www.semanticscholar.org/author/2025052385"
+
+    # # match 2
+    # example2_title = "Model-based Policy Optimization with Unsupervised Model Adaptation"
+    # example2_submission = """Model-based reinforcement learning methods learn a dynamics model with real data sampled from the environment and leverage it to generate simulated data to derive an agent. However, due to the potential distribution mismatch between simulated data and real data, this could lead to degraded performance. Despite much effort being devoted to reducing this distribution mismatch, existing methods fail to solve it explicitly. In this paper, we investigate how to bridge the gap between real and simulated data due to inaccurate model estimation for better policy optimization. To begin with, we first derive a lower bound of the expected return, which naturally inspires a bound maximization algorithm by aligning the simulated and real data distributions. To this end, we propose a novel model-based reinforcement learning framework AMPO, which introduces unsupervised model adaptation to minimize the integral probability metric (IPM) between feature distributions from real and simulated data. Instantiating our framework with Wasserstein-1 distance gives a practical model-based approach. Empirically, our approach achieves state-of-the-art performance in terms of sample efficiency on a range of continuous control benchmark tasks."""
+    # example2_reviewer = "https://www.semanticscholar.org/author/144974941"
+
+    # # match 3
+    # example3_title = "Sharp asymptotic and finite-sample rates of convergence of empirical measures in Wasserstein distance"
+    # example3_submission = """The Wasserstein distance between two probability measures on a metric space is a measure of closeness with applications in statistics, probability, and machine learning. In this work, we consider the fundamental question of how quickly the empirical measure obtained from $n$ independent samples from $\mu$ approaches $\mu$ in the Wasserstein distance of any order. We prove sharp asymptotic and finite-sample results for this rate of convergence for general measures on general compact metric spaces. Our finite-sample results show the existence of multi-scale behavior, where measures can exhibit radically different rates of convergence as $n$ grows."""
+    # example3_reviewer = "https://www.semanticscholar.org/author/27911143"
+    
+    # # match 4
+    # example4_title = "Deep Neural Networks for Estimation and Inference: Application to Causal Effects and Other Semiparametric Estimands"
+    # example4_submission = """We study deep neural networks and their use in semiparametric inference. We prove valid inference after first-step estimation with deep learning, a result new to the literature. We provide new rates of convergence for deep feedforward neural nets and, because our rates are sufficiently fast (in some cases minimax optimal), obtain valid semiparametric inference. Our estimation rates and semiparametric inference results handle the current standard architecture: fully connected feedforward neural networks (multi-layer perceptrons), with the now-common rectified linear unit activation function and a depth explicitly diverging with the sample size. We discuss other architectures as well, including fixed-width, very deep networks. We establish nonasymptotic bounds for these deep nets for nonparametric regression, covering the standard least squares and logistic losses in particular. We then apply our theory to develop semiparametric inference, focusing on treatment effects, expected welfare, and decomposition effects for concreteness. Inference in many other semiparametric contexts can be readily obtained. We demonstrate the effectiveness of deep learning with a Monte Carlo analysis and an empirical application to direct mail marketing."""
+    # example4_reviewer = "https://www.semanticscholar.org/author/3364789"
+    
     ### INPUT
     with gr.Row() as input_row:
         with gr.Column(scale=3):
@@ -388,15 +436,17 @@ R2P2 provides more information about each reviewer. It searches for the **most r
                 name = gr.Textbox(label='Confirm Reviewer Name', info='This will be automatically updated based on the reviewer profile link above', interactive=False)
                 author_id_input.change(fn=update_name, inputs=author_id_input, outputs=name)
                 
-    # Add examples
-    example_title ="The Toronto Paper Matching System: An automated paper-reviewer assignment system"
-    example_submission = """One of the most important tasks of conference organizers is the assignment of papers to reviewers. Reviewers' assessments of papers is a crucial step in determining the conference program, and in a certain sense to shape the direction of a field. However this is not a simple task: large conferences typically have to assign hundreds of papers to hundreds of reviewers, and time constraints make the task impossible for one person to accomplish. Furthermore other constraints, such as reviewer load have to be taken into account, preventing the process from being completely distributed. We built the first version of a system to suggest reviewer assignments for the NIPS 2010 conference, followed, in 2012, by a release that better integrated our system with Microsoft's popular Conference Management Toolkit (CMT). Since then our system has been widely adopted by the leading conferences in both the machine learning and computer vision communities. This paper provides an overview of the system, a summary of learning models and methods of evaluation that we have been using, as well as some of the recent progress and open issues."""
-    example_reviewer = "https://www.semanticscholar.org/author/Nihar-B.-Shah/1737249"
     gr.Examples(
-        examples=[[example_title, example_submission, example_reviewer]],
+        examples=[
+            [example_title, example_submission, example_reviewer],
+            # [example1_title, example1_submission, example1_reviewer],
+            # [example2_title, example2_submission, example2_reviewer],
+            # [example3_title, example3_submission, example3_reviewer],
+            # [example4_title, example4_submission, example4_reviewer],
+        ],
         inputs=[title_input, abstract_text_input, author_id_input],
         cache_examples=False,
-        label="Try out the following example input."
+        label="Try out the following example input. Click on a row to fill in the input fields accordingly."
     )
     
     with gr.Row():
@@ -409,16 +459,18 @@ R2P2 provides more information about each reviewer. It searches for the **most r
     # Paper title, score, and top-ranking sentence pairs 
     # a knob for controlling the number of output displayed
     with gr.Row():
-        with gr.Column(scale=5):
+        with gr.Column(scale=3):
             result1_desc = gr.Markdown(value='', visible=False)
         with gr.Column(scale=2):
-            with gr.Row():
-                top_paper_slider = gr.Slider(label='Top-K Papers by the Reviewer', value=3, minimum=3, step=1, maximum=NUM_PAPERS_SHOW, visible=False)
-            with gr.Row():
-                top_pair_slider = gr.Slider(label='Top-K Sentence Pairs per Paper', value=2, minimum=2, step=1, maximum=NUM_PAIRS_SHOW, visible=False)
+        #    with gr.Row():
+            top_paper_slider = gr.Slider(label='Number of papers to show', value=3, minimum=3, step=1, maximum=NUM_PAPERS_SHOW, visible=False)
+        with gr.Column(scale=2):
+            #with gr.Row():
+            top_pair_slider = gr.Slider(label='Number of sentence pairs to show', value=2, minimum=2, step=1, maximum=NUM_PAIRS_SHOW, visible=False)
 
     paper_title_up = []
     paper_affinity_up = []
+    citation_count = []
     sent_pair_score = []
     sent_text_query = []
     sent_text_candidate = []
@@ -434,6 +486,9 @@ R2P2 provides more information about each reviewer. It searches for the **most r
             with gr.Column(scale=3):
                 tt = gr.Markdown(value='', visible=False)
                 paper_title_up.append(tt)
+            with gr.Column(scale=1):
+                cc = gr.Markdown(value='', visible=False)
+                citation_count.append(cc)
             with gr.Column(scale=1):
                 aff = gr.Markdown(value='', visible=False)
                 paper_affinity_up.append(aff)
@@ -443,12 +498,14 @@ R2P2 provides more information about each reviewer. It searches for the **most r
                     sps = gr.Markdown(value='', visible=False)
                     sent_pair_score.append(sps)
                 with gr.Column(scale=5):
-                    stq = gr.Textbox(label='Sentence from Submission', visible=False)
+                    #stq = gr.Textbox(label='Sentence from Submission', visible=False)
+                    stq = gr.Textbox(label='', visible=False)
                     shq = gr.components.Interpretation(stq, visible=False)
                     sent_text_query.append(stq)
                     sent_hl_query.append(shq)
                 with gr.Column(scale=5):
-                    stc = gr.Textbox(label="Sentence from Reviewer's Paper", visible=False)
+                    #stc = gr.Textbox(label="Sentence from Reviewer's Paper", visible=False)
+                    stc = gr.Textbox(label="", visible=False)
                     shc = gr.components.Interpretation(stc, visible=False)
                     sent_text_candidate.append(stc)
                     sent_hl_candidate.append(shc)
@@ -458,7 +515,7 @@ R2P2 provides more information about each reviewer. It searches for the **most r
       
     ## Show more button
     with gr.Row():
-        see_more_rel_btn = gr.Button('Explore more', visible=False)
+        see_more_rel_btn = gr.Button('Not Enough Information? Explore More', visible=False)
     
     ### PAPER INFORMATION
     
@@ -474,7 +531,7 @@ R2P2 provides more information about each reviewer. It searches for the **most r
     """ 
     #---"""
     # show multiple papers in radio check box to select from
-    paper_abstract = gr.Textbox(label='Abstract', interactive=False, visible=False)
+    paper_abstract = gr.Textbox(label='', interactive=False, visible=False)
     with gr.Row():
         with gr.Column(scale=1):
             selected_papers_radio = gr.Radio(
@@ -510,6 +567,9 @@ R2P2 provides more information about each reviewer. It searches for the **most r
             with gr.Row(visible=False) as aff_row:
                 # selected paper's affinity score
                 affinity = gr.Markdown(value='')
+            with gr.Row(visible=False) as cite_row:
+                # selected paper's citation count 
+                citation = gr.Markdown(value='')
             with gr.Row(visible=False) as hl_row: 
                 # highlighted text from paper
                 highlight = gr.components.Interpretation(paper_abstract) 
@@ -526,7 +586,7 @@ R2P2 provides more information about each reviewer. It searches for the **most r
     ]
     
     init_result_components = \
-        paper_title_up + paper_affinity_up + demarc_lines + sent_pair_score + \
+        paper_title_up + paper_affinity_up + citation_count + demarc_lines + sent_pair_score + \
             sent_text_query + sent_text_candidate + sent_hl_query + sent_hl_candidate 
     
     explore_more_components = [
@@ -569,6 +629,7 @@ R2P2 provides more information about each reviewer. It searches for the **most r
             source_sentences,
             title_row,
             aff_row,
+            cite_row,
             highlight_legend,
             highlight_slider,
             hl_row,
@@ -600,6 +661,7 @@ R2P2 provides more information about each reviewer. It searches for the **most r
             paper_title,
             paper_abstract,
             affinity,
+            citation,
             highlight, 
             highlight_slider
         ]
@@ -642,4 +704,5 @@ R2P2 provides more information about each reviewer. It searches for the **most r
     )
     
 if __name__ == "__main__":
-    demo.queue().launch()  # add ?__theme=light to force light mode
+    #demo.queue().launch(debug=True)  # add ?__theme=light to force light mode
+    demo.queue().launch(share=True)  # add ?__theme=light to force light mode

input_format.py

@@ -1,67 +1,9 @@
-from pypdf import PdfReader
 from urllib.parse import urlparse
 import requests
 from semanticscholar import SemanticScholar
 
-### Input Formatting Module
+### Input Formatting
 
-## Input formatting for the given paper
-# Extracting text from a pdf or a link
-
-def get_text_from_pdf(file_path):
-    """
-    Convert a pdf to list of text files
-    """
-    reader = PdfReader(file_path)
-    text = []
-    for p in reader.pages:
-        t = p.extract_text()
-        text.append(t)
-    return text
-
-def get_text_from_url(url, file_path='paper.pdf'):
-    """
-    Get text of the paper from a url
-    """
-    ## Check for different URL cases
-    url_parts = urlparse(url)
-    # arxiv
-    if 'arxiv' in url_parts.netloc:
-        if 'abs' in url_parts.path:
-            # abstract page, change the url to pdf link
-            paper_id = url_parts.path.split('/')[-1]
-            url = 'https://www.arxiv.org/pdf/%s.pdf'%(paper_id)
-        elif 'pdf' in url_parts.path:
-            # pdf file, pass
-            pass
-        else:
-            raise ValueError('invalid url')
-    else:
-        raise ValueError('invalid url')
-
-    # download the file
-    download_pdf(url, file_path)
-
-    # get the text from the pdf file
-    text = get_text_from_pdf(file_path)
-    return text
-    
-def download_pdf(url, file_name):
-    """
-    Download the pdf file from given url and save it as file_name
-    """
-    # Send GET request
-    response = requests.get(url)
-
-    # Save the PDF
-    if response.status_code == 200:
-        with open(file_name, "wb") as f:
-            f.write(response.content)
-    elif response.status_code == 404:
-        raise ValueError('cannot download the file')
-    else:
-        print(response.status_code)
-        
 ## Input formatting for the given author (reviewer)
 # Extracting text from a link
 
@@ -71,8 +13,9 @@ def get_text_from_author_id(author_id, max_count=150):
     aid = str(author_id)
     if 'http' in aid: # handle semantic scholar url input
         aid = aid.split('/')
-        aid = aid[aid.index('author')+2]
-    url = "https://api.semanticscholar.org/graph/v1/author/%s?fields=url,name,paperCount,papers,papers.title,papers.abstract,papers.url"%aid
+        aid = aid[-1]
+        # aid = aid[aid.index('author')+2]
+    url = "https://api.semanticscholar.org/graph/v1/author/%s?fields=url,name,paperCount,papers,papers.title,papers.abstract,papers.url,papers.year,papers.citationCount"%aid
     r = requests.get(url)
     if r.status_code == 404:
         raise ValueError('Author link not found.')

requirements.txt

@@ -1,16 +1,11 @@
-gradio==3.20.1
-huggingface-hub==0.8.1
+gradio==3.24.1
+huggingface-hub
 nltk==3.7
-numpy==1.21.6
-py-pdf-parser==0.10.2
-py-rouge==1.1
-pypdf==3.3.0
-pyrogue==0.0.2
+numpy
 requests==2.28.1
-rouge-score==0.1.2
 semanticscholar==0.3.2
-sentence-transformers==2.2.0
+sentence-transformers==2.2.2
 torch==1.9.0
-transformers
+transformers==4.27.4
 urllib3==1.26.6
-tqdm
+tqdm

score.py

@@ -338,11 +338,15 @@ def compute_document_score(doc_model, tokenizer, query_title, query_abs, papers,
     titles = []
     abstracts = []
     urls = []
+    years = []
+    citations = []
     for p in papers:
         if p['title'] is not None and p['abstract'] is not None:
             titles.append(p['title'])
             abstracts.append(p['abstract'])
             urls.append(p['url'])
+            years.append(p['year'])
+            citations.append(p['citationCount'])
     if query_title == '':
         query = query_abs
     else:
@@ -355,5 +359,7 @@ def compute_document_score(doc_model, tokenizer, query_title, query_abs, papers,
     abstracts_sorted = [abstracts[x] for x in idx_sorted]
     scores_sorted = [scores[x] for x in idx_sorted]
     urls_sorted = [urls[x] for x in idx_sorted]
+    years_sorted = [years[x] for x in idx_sorted]
+    citations_sorted = [citations[x] for x in idx_sorted]
     
-    return titles_sorted, abstracts_sorted, urls_sorted, scores_sorted
+    return titles_sorted, abstracts_sorted, urls_sorted, scores_sorted, years_sorted, citations_sorted

style.css

@@ -44,7 +44,7 @@
 	box-shadow: 1px 1px 1px rgba(0, 0, 0, 0.2);
 	right: -4px;
 	color: #FFF;
-	font-size: 10px;
+	font-size: 11px;
 	line-height: 1.4;
 }