using state variables to store intermediary search results.

app.py

@@ -27,8 +27,9 @@ def get_similar_paper(
     abstract_text_input, 
     pdf_file_input, 
     author_id_input, 
-    num_papers_show=10
+    results={} # variable will be updated and returned
 ):
+    num_papers_show = 10 # number of top papers to show from the reviewer
     print('retrieving similar papers...')
     start = time.time()
     input_sentences = sent_tokenize(abstract_text_input)
@@ -53,7 +54,7 @@ def get_similar_paper(
         batch=50
     )
     
-    tmp = {
+    results = {
         'titles': titles,
         'abstracts': abstracts,
         'doc_scores': doc_scores
@@ -91,7 +92,7 @@ def get_similar_paper(
                 "interpretation": list(zip(info['all_words'], info[i]['scores']))
             }
 
-        tmp[display_title[aa]] = {
+        results[display_title[aa]] = {
             'title': tt,
             'abstract': ab,
             'doc_score': ds,
@@ -100,11 +101,11 @@ def get_similar_paper(
             'top_pairs': top_pairs_info
         }
        
-    # TODO better ways of saving intermediate results? user identifiers per session? 
-    pickle.dump(tmp, open('info.pkl', 'wb')) 
     end = time.time()
     print('done in [%0.2f] seconds'%(end - start)) 
     
+    ## Set up output elements
+    
     # set up elements to show 
     out = [
         gr.update(choices=display_title, interactive=True, visible=False), # set of papers (radio)
@@ -120,7 +121,7 @@ def get_similar_paper(
             gr.update(value=titles[i], visible=True), 
             gr.update(value=doc_scores[i], visible=True)
         ]
-        tp = tmp[display_title[i]]['top_pairs']
+        tp = results[display_title[i]]['top_pairs']
         for j in range(top_num_info_show):
             out_tmp += [
                 gr.update(value=tp[j]['score'], visible=True), 
@@ -135,6 +136,9 @@ def get_similar_paper(
     out = out + summary_out + [gr.update(visible=True)] # show more button
     assert(len(out) == (top_num_info_show * 5 + 2) * top_papers_show + 3)
     
+    # add the search results to pass on to the Gradio State varaible
+    out += [results]
+    
     return tuple(out)
         
 def show_more():
@@ -151,13 +155,11 @@ def update_name(author_id_input):
     
     return gr.update(value=name)
 
-def change_output_highlight(selected_papers_radio, source_sent_choice):
+def change_output_highlight(selected_papers_radio, source_sent_choice, info={}):
     # change the output highlight based on the sentence selected from the submission
-    fname = 'info.pkl'
-    if os.path.exists(fname):
-        tmp = pickle.load(open(fname, 'rb'))
-        source_sents = tmp[selected_papers_radio]['source_sentences']
-        highlights = tmp[selected_papers_radio]['highlight']
+    if len(info.keys()) != 0: # if the info is not empty
+        source_sents = info[selected_papers_radio]['source_sentences']
+        highlights = info[selected_papers_radio]['highlight']
         for i, s in enumerate(source_sents):
             #print('changing highlight')
             if source_sent_choice == s:
@@ -165,21 +167,19 @@ def change_output_highlight(selected_papers_radio, source_sent_choice):
     else:
         return
 
-def change_paper(selected_papers_radio):
-    # change the paper to show based on the paper selected
-    fname = 'info.pkl'
-    if os.path.exists(fname):
-        tmp = pickle.load(open(fname, 'rb'))
-        title = tmp[selected_papers_radio]['title']
-        abstract = tmp[selected_papers_radio]['abstract']
-        aff_score = tmp[selected_papers_radio]['doc_score']
-        highlights = tmp[selected_papers_radio]['highlight']
+def change_paper(selected_papers_radio, info={}):
+    if len(info.keys()) != 0: # if the info is not empty
+        title = info[selected_papers_radio]['title']
+        abstract = info[selected_papers_radio]['abstract']
+        aff_score = info[selected_papers_radio]['doc_score']
+        highlights = info[selected_papers_radio]['highlight']
         return title, abstract, aff_score, highlights['0']
 
     else:
         return
 
 with gr.Blocks() as demo:
+    info = gr.State({})  # cached search results as a State variable shared throughout
     
     # Text description about the app and disclaimer
     ### TEXT Description
@@ -228,7 +228,6 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
     with gr.Row():
         compute_btn = gr.Button('What Makes This a Good Match?')  
         
-        
     ### OVERVIEW
     # Paper title, score, and top-ranking sentence pairs -- two sentence pairs per paper, three papers
     # TODO blockfy similar components together and simplify 
@@ -350,7 +349,8 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
         inputs=[
             abstract_text_input, 
             pdf_file_input, 
-            author_id_input
+            author_id_input,
+            info
         ],
         outputs=[
             selected_papers_radio,
@@ -391,7 +391,8 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
             sent_pair_source3_2_hl,
             sent_pair_candidate3_2,
             sent_pair_candidate3_2_hl,
-            see_more_rel_btn
+            see_more_rel_btn,
+            info
         ]
     )      
     
@@ -412,7 +413,8 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
         fn=change_output_highlight,
         inputs=[
             selected_papers_radio,
-            source_sentences
+            source_sentences,
+            info
         ],
         outputs=highlight
     )
@@ -420,7 +422,10 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
     # change paper to show based on selected papers
     selected_papers_radio.change(
         fn=change_paper,
-        inputs=selected_papers_radio,
+        inputs=[
+            selected_papers_radio,
+            info,
+        ],
         outputs= [
             paper_title,
             paper_abstract,