showing multiple top papers from the author

app.py

@@ -52,14 +52,16 @@ def get_similar_paper(
         'abstracts': abstracts,
         'doc_scores': doc_scores
     }
-    pickle.dump(tmp, open('tmp_paperinfo.pkl', 'wb'))
+    pickle.dump(tmp, open('paper_info.pkl', 'wb'))
     
     # Select top K choices of papers to show
     titles = titles[:num_papers_show]
     abstracts = abstracts[:num_papers_show]
     doc_scores = doc_scores[:num_papers_show]
     
-    return titles[0], abstracts[0], doc_scores[0], gr.update(choices=input_sentences, interactive=True), gr.update(visible=True)
+    display_title = ['[ %0.3f ] %s'%(s, t) for t, s in zip(titles, doc_scores)]
+    
+    return gr.update(choices=display_title, interactive=True, visible=True), gr.update(choices=input_sentences, interactive=True), gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
 def get_highlights(
     abstract_text_input,
@@ -101,18 +103,32 @@ def update_name(author_id_input):
     return gr.update(value=name)
 
 def change_output_highlight(source_sent_choice):
+    fname = 'highlight_info.pkl'
     # change the output highlight based on the sentence selected from the submission
-    if os.path.exists('highlight_info.pkl'):
-        tmp = pickle.load(open('highlight_info.pkl', 'rb'))
+    if os.path.exists(fname):
+        tmp = pickle.load(open(fname, 'rb'))
         source_sents = tmp['source_sentences']
         highlights = tmp['highlight']
         for i, s in enumerate(source_sents):
-            print('changing highlight!')
+            #print('changing highlight')
             if source_sent_choice == s:
                 return highlights[str(i)]
     else:
         return
 
+def change_paper(selected_papers_radio):
+    # change the paper to show
+    fname = 'paper_info.pkl'
+    if os.path.exists(fname):
+        tmp = pickle.load(open(fname, 'rb'))
+        for title, abstract, aff_score in zip(tmp['titles'], tmp['abstracts'], tmp['doc_scores']):
+            display_title = '[ %0.3f ] %s'%(aff_score, title)
+            if display_title == selected_papers_radio:
+                #print('changing paper')
+                return title, abstract, aff_score
+    else:
+        return
+
 with gr.Blocks() as demo:
     
     ### INPUT
@@ -130,24 +146,22 @@ with gr.Blocks() as demo:
     with gr.Row():
         compute_btn = gr.Button('Search Similar Papers from the Reviewer')  
             
-
-    # with gr.Row(visible=False) as reviewer_name_info:
-    #     name = gr.Textbox(label='Reveiwer Author Name')
-    # with gr.Row():
-    #     with gr.Tabs():
-    #         for tt in range(num_papers_show):
-    #             with gr.TabItem('Paper %d'%(tt+1)):
-    
-    # TODO handle multiple papers
+    # show multiple papers in radio check box to select from
+    with gr.Row():
+        selected_papers_radio = gr.Radio(
+            choices=[], # will be udpated with the button click
+            visible=False, # also will be updated with the button click
+            label='Selected Top Papers from the Reviewer'
+        )
     
     ### PAPER INFORMATION 
-    with gr.Row():
+    with gr.Row(visible=False) as title_row:
         with gr.Column(scale=3):
             paper_title = gr.Textbox(label='Title', interactive=False)
         with gr.Column(scale=1):
             affinity= gr.Number(label='Affinity', interactive=False, value=0)
-    with gr.Row():
-        paper_abstract = gr.Textbox(label='Abstract', interactive=False)
+    with gr.Row(visibe=False) as abstract_row:
+        paper_abstract = gr.Textbox(label='Abstract', interactive=False, visible=False)
         
     with gr.Row(visible=False) as explain_button_row:
         explain_btn = gr.Button('Show Relevant Parts from Selected Paper')
@@ -172,11 +186,11 @@ with gr.Blocks() as demo:
             author_id_input
         ],
         outputs=[
-            paper_title,
-            paper_abstract,
-            affinity,
+            selected_papers_radio,
             source_sentences,
-            explain_button_row
+            title_row,
+            paper_abstract,
+            explain_button_row,
         ]
     )      
     
@@ -195,6 +209,16 @@ with gr.Blocks() as demo:
         inputs=source_sentences,
         outputs=highlight
     )
+    
+    selected_papers_radio.change(
+        fn=change_paper,
+        inputs=selected_papers_radio,
+        outputs= [
+            paper_title,
+            paper_abstract,
+            affinity
+        ]
+    )
 
 if __name__ == "__main__":
     demo.launch() 

input_format.py

@@ -24,8 +24,6 @@ def get_text_from_url(url, file_path='paper.pdf'):
     """
     Get text of the paper from a url
     """
-    # TODO check for other valid urls (e.g. semantic scholar)
-
     ## Check for different URL cases
     url_parts = urlparse(url)
     # arxiv