making the search results more responsive

app.py

@@ -111,10 +111,17 @@ def get_similar_paper(
     
     ## Set up output elements
     
-    # set up elements to show 
+    # first the list of top papers, sentences to select from, paper_title, affinity
+    title = results[display_title[0]]['title'] # set default title as the top paper
+    url = results[display_title[0]]['url']
+    aff_score = results[display_title[0]]['doc_score']
+    title_out = """<a href="%s" target="_blank"><h5>%s</h5></a>"""%(url, title)
+    aff_score_out = '##### Affinity Score: %s'%aff_score
     out = [
-        gr.update(choices=display_title, interactive=True), # set of papers (radio)
-        gr.update(choices=input_sentences, interactive=True) # submission sentences 
+        gr.update(choices=display_title, value=display_title[0],  interactive=True), # set of papers (radio)
+        gr.update(choices=input_sentences, value=input_sentences[0], interactive=True), # submission sentences 
+        gr.update(value=title_out), # paper_title
+        gr.update(value=aff_score_out)  # affinity
     ]
     
     # set up elements to visualize upfront
@@ -162,7 +169,7 @@ def get_similar_paper(
 
     # add updates to the show more button
     out = out + summary_out + [gr.update(visible=True)] # make show more button visible
-    assert(len(out) == (top_num_info_show * 5 + 2) * top_papers_show + 3)
+    assert(len(out) == (top_num_info_show * 5 + 2) * top_papers_show + 5)
 
     out += [gr.update(value="""
                             <h3>Top three relevant papers by the reviewer <a href="%s" target="_blank">%s</a></h3>
@@ -198,6 +205,7 @@ def show_more(info):
         gr.update(visible=True), # title row
         gr.update(visible=True), # affinity row
         gr.update(visible=True), # highlight legend
+        gr.update(visible=True), # highlight abstract
     )
 
 def show_status():
@@ -210,7 +218,7 @@ def update_name(author_id_input):
     
     return gr.update(value=name)
 
-def change_output_highlight(selected_papers_radio, source_sent_choice, info={}):
+def change_sentence(selected_papers_radio, source_sent_choice, info={}):
     # change the output highlight based on the sentence selected from the submission
     if len(info.keys()) != 0: # if the info is not empty
         source_sents = info[selected_papers_radio]['source_sentences']
@@ -221,8 +229,9 @@ def change_output_highlight(selected_papers_radio, source_sent_choice, info={}):
     else:
         return
 
-def change_paper(selected_papers_radio, info={}):
+def change_paper(selected_papers_radio, source_sent_choice, info={}):
     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']
         abstract = info[selected_papers_radio]['abstract']
         aff_score = info[selected_papers_radio]['doc_score']
@@ -230,8 +239,9 @@ def change_paper(selected_papers_radio, info={}):
         url = info[selected_papers_radio]['url']
         title_out = """<a href="%s" target="_blank"><h5>%s</h5></a>"""%(url, title)
         aff_score_out = '##### Affinity Score: %s'%aff_score
-        return title_out, abstract, aff_score_out, highlights['0']
-
+        for i, s in enumerate(source_sents):
+            if source_sent_choice == s:
+                return title_out, abstract, aff_score_out, highlights[str(i)]
     else:
         return
 
@@ -285,7 +295,6 @@ R2P2 provides more information about each reviewer. It searches for the most rel
     ## PAPER1 
     with gr.Row():
         result1_desc = gr.Markdown(value='', visible=False)
-    # TODO hovering instructions
     with gr.Row():
         with gr.Column(scale=3):
             paper_title1 = gr.Markdown(value='', visible=False)
@@ -413,7 +422,7 @@ R2P2 provides more information about each reviewer. It searches for the most rel
                 paper_title = gr.Markdown(value='')
             with gr.Row(visible=False) as aff_row:
                 affinity = gr.Markdown(value='')
-            with gr.Row(): 
+            with gr.Row(visible=False) as hl_row: 
                 # highlighted text from paper
                 highlight = gr.components.Interpretation(paper_abstract) 
             
@@ -434,8 +443,10 @@ R2P2 provides more information about each reviewer. It searches for the most rel
             info
         ],
         outputs=[
-            selected_papers_radio,
-            source_sentences,
+            selected_papers_radio, # list of papers for show more section
+            source_sentences, # list of sentences for show more section
+            paper_title, # paper title for show more section
+            affinity, # paper affinity for show more section
             paper_title1, # paper info
             affinity1,
             sent_pair_score1_1,
@@ -495,12 +506,13 @@ R2P2 provides more information about each reviewer. It searches for the most rel
             title_row,
             aff_row,
             highlight_legend,
+            hl_row
         ]
     )
     
     # change highlight based on selected sentences from submission
     source_sentences.change(
-        fn=change_output_highlight,
+        fn=change_sentence,
         inputs=[
             selected_papers_radio,
             source_sentences,
@@ -514,6 +526,7 @@ R2P2 provides more information about each reviewer. It searches for the most rel
         fn=change_paper,
         inputs=[
             selected_papers_radio,
+            source_sentences,
             info,
         ],
         outputs= [
@@ -526,3 +539,4 @@ R2P2 provides more information about each reviewer. It searches for the most rel
     
 if __name__ == "__main__":
     demo.queue().launch()  # add ?__theme=light to force light mode
+