added search progress indication

app.py

@@ -27,20 +27,21 @@ def get_similar_paper(
     abstract_text_input, 
     author_id_input, 
     results={}, # this state variable will be updated and returned
-    #progress=gr.Progress()
 ):
+    progress = gr.Progress()
+    
     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)
     
     # Get author papers from id
-    #progress(0.1, desc="Retrieving reviewer papers ...")
+    progress(0.1, desc="Retrieving reviewer papers ...")
     name, papers = get_text_from_author_id(author_id_input)
     
     # Compute Doc-level affinity scores for the Papers 
     # print('computing document scores...') 
-    #progress(0.5, desc="Computing document scores...")
+    progress(0.5, desc="Computing document scores...")
     # TODO detect duplicate papers?
     titles, abstracts, paper_urls, doc_scores = compute_document_score(
         doc_model, 
@@ -65,9 +66,10 @@ def get_similar_paper(
     
     display_title = ['[ %0.3f ] %s'%(s, t) for t, s in zip(titles, doc_scores)]
     end = time.time()
-    print('paper retrieval complete in [%0.2f] seconds'%(end - start))
+    retrieval_time = end - start
+    print('paper retrieval complete in [%0.2f] seconds'%(retrieval_time))
     
-    #progress(0.4, desc="Obtaining relevant information from the papers...")
+    progress(0.4, desc="Obtaining relevant information from the papers...")
     print('obtaining highlights..')
     start = time.time()
     input_sentences = sent_tokenize(abstract_text_input)
@@ -101,7 +103,8 @@ def get_similar_paper(
         }
        
     end = time.time()
-    print('done in [%0.2f] seconds'%(end - start)) 
+    highlight_time = end - start
+    print('done in [%0.2f] seconds'%(highlight_time)) 
     
     ## Set up output elements
     
@@ -137,6 +140,9 @@ def get_similar_paper(
     assert(len(out) == (top_num_info_show * 5 + 2) * top_papers_show + 3)
 
     out += [gr.update(visible=True), gr.update(visible=True)] # demarcation line between results
+
+    # progress status
+    out += [gr.update(value='Done (in %0.1f seconds)'%(retrieval_time+highlight_time), visible=True)]
     
     # add the search results to pass on to the Gradio State varaible
     out += [results]
@@ -144,6 +150,7 @@ def get_similar_paper(
     return tuple(out)
         
 def show_more():
+    # show the interactive part of the app
     return (
         gr.update(visible=True), # set of papers
         gr.update(visible=True), # submission sentences
@@ -151,6 +158,10 @@ def show_more():
         gr.update(visible=True), # abstract row
     )
 
+def show_status():
+    # show search status field when search button is clicked
+    return gr.update(visible=True)
+
 def update_name(author_id_input):
     # update the name of the author based on the id input
     name, _ = get_text_from_author_id(author_id_input)
@@ -233,7 +244,8 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
 
     # TODO indicate the progress when pressed
     with gr.Row():
-        search_status = gr.Textbox(label='Search Status', interactive=False, visible=True)
+        #search_status = gr.Label(label="Search Status", visible=True)
+        search_status = gr.Textbox(label='Search Status', interactive=False, visible=False)
         
     ### OVERVIEW
     # Paper title, score, and top-ranking sentence pairs -- two sentence pairs per paper, three papers
@@ -360,8 +372,15 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
         with gr.Column(scale=3): # highlighted text from paper
             highlight = gr.components.Interpretation(paper_abstract) 
     
+    
     ### EVENT LISTENERS
     
+    compute_btn.click(
+        fn=show_status,
+        inputs=[],
+        outputs=search_status
+    )
+    
     # retrieve similar papers and show top results
     compute_btn.click(
         fn=get_similar_paper,
@@ -412,6 +431,7 @@ Below we describe how to use the tool. Also feel free to check out the [video]()
             see_more_rel_btn,
             demarc1,
             demarc2,
+            search_status,
             info,
         ]
     )