minor

app.py

@@ -177,6 +177,11 @@ def start_haystack():
 
     return pipe
 
+def run_statement(statement: str, retriever_top_k: int = 5):
+    pipe = start_haystack()
+    run = check_statement(statement, retriever_top_k)
+    return run
+
 @st.cache_resource
 def check_statement(pipe, statement: str, retriever_top_k: int = 5):
     """Run query and verify statement"""
@@ -243,8 +248,6 @@ def main():
         run_pressed or statement != st.session_state.statement
     )
 
-    pipe = start_haystack()
-
     # Get results for query
     if run_query and statement:
         time_start = time.time()
@@ -252,7 +255,7 @@ def main():
         st.session_state.statement = statement
         with st.spinner("   Procurando a Similaridade no banco de sentenças..."):
             try:
-                st.session_state.results = check_statement(pipe, statement, RETRIEVER_TOP_K)
+                st.session_state.results = run_statement(statement, RETRIEVER_TOP_K)
                 print(f"S: {statement}")
                 time_end = time.time()
                 print(time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime()))