add progress bar

app.py

@@ -269,7 +269,7 @@ def matched_context(start_i, end_i, contexts_string, seperator='---'):
     return None
 
 
-def run_query(query):
+def run_query(query, progress_bar):
 #     if use_query_exp == 'yes':
 #         query_exp = paraphrase(f"question2question: {query}")
 #         st.markdown(f"""
@@ -298,7 +298,7 @@ def run_query(query):
             set(contexts_strict)
         )
         orig_docs = orig_docs_strict
-
+    progress_bar.progress(25)
 
     if len(contexts) == 0 or not ''.join(contexts).strip():
         return st.markdown("""
@@ -320,6 +320,7 @@ def run_query(query):
     else:
         contexts = contexts[:context_limit]
 
+    progress_bar.progress(50)
     if concat_passages == 'yes':
         context = '\n---'.join(contexts)
         model_results = qa_model(question=query, context=context, top_k=10)
@@ -329,6 +330,7 @@ def run_query(query):
 
     results = []
 
+    progress_bar.progress(75)
     for i, result in enumerate(model_results):
         if concat_passages == 'yes':
             matched = matched_context(result['start'], result['end'], context)
@@ -359,6 +361,7 @@ def run_query(query):
         sorted_result
     ))
 
+    progress_bar.progress(100)
     for r in sorted_result:
         ctx = remove_html(r["context"])
         for answer in r['texts']:
@@ -371,4 +374,5 @@ def run_query(query):
 query = st.text_input("Ask scientific literature a question", "")
 if query != "":
     with st.spinner('Loading...'):
-        run_query(query)
+        progress_bar = st.progress(0)
+        run_query(query, progress_bar)