Update app.py

app.py

@@ -85,35 +85,30 @@ def main():
 
   comprehension_model = load_comprehension_model()
 
-  query = st.text_input("Enter Query",'example query ', key="query")
+  query = st.text_input("Enter Query",'What are the corona viruses?', key="query")
   
-  st.write(query)
+  with st.spinner('Please wait...'):
+ 
+    query_embedding, results1 = fetch_stage1(query, model, list_of_articles)
+    results2 = fetch_stage2(results1, model, embeddings, query_embedding)
+    results3 = fetch_stage3(results2, query, embeddings, comprehension_model, spacy_nlp)
+
+    if results3:
+      count = 1
   
-  query_embedding, results1 = fetch_stage1(query, model, list_of_articles)
-  st.write('Hello1')
-
-  results2 = fetch_stage2(results1, model, embeddings, query_embedding)
-  st.write('Hello2')
-  results3 = fetch_stage3(results2, query, embeddings, comprehension_model, spacy_nlp)
-  
-  st.write('Hello3')
-  
-  if results3:
-    count = 1
-
-    for res in results3:
-      st.write('{}> {}'.format(count, res[2]))
-      st.write('Score: %.4f' % (res[1]))
-      st.write("From the article with title: {}".format(embeddings[res[0]][0]))
-      st.write("\n")
-      # print(count,". ", res[2], "(Score: %.4f)" % (res[1]))
-      # print("From the article with title: ", embeddings[res[0]][0])
-      # print("\n")
-      if count > 3:
-        break
-      count += 1
-  else:
-    st.info("There isn't any answer")
+      for res in results3:
+        st.write('{}> {}'.format(count, res[2]))
+        st.write('Score: %.4f' % (res[1]))
+        st.write("From the article with title: {}".format(embeddings[res[0]][0]))
+        st.write("\n")
+        
+        if count > 3:
+          break
+        count += 1
+    else:
+      st.info("There isn't any answer")
+      
+  st.success('Done!')
 
 
 if __name__ == '__main__':