Update app.py

app.py

@@ -70,7 +70,6 @@ def main():
     if 'few_documents' not in st.session_state:
         st.session_state['few_documents'] = False
         few_documents = False
-        Timer.reset()
     else:
         few_documents = st.session_state['few_documents']
 
@@ -100,6 +99,7 @@ def main():
 
 
     if few_documents:
+        Timer.reset()
         error_msg = st.empty()
         error_msg.warning(st.session_state['msg'])
         proceed = st.empty()
@@ -108,6 +108,8 @@ def main():
         else:
             proceed_button = proceed.button('Proceed')
         if proceed_button:
+            error_msg.empty()
+            proceed.empty()
             query = translate(query_pt, 'pt', 'en') if portuguese else query
             text = extract(query, search_model=search_model, extracted_documents=st.session_state['documents'])
             summary = summarize(text, summ_model, tokenizer)
@@ -121,8 +123,6 @@ def main():
                 with result.expander(f'Paragraphs used in summarization'):
                     st.markdown(text.replace('\n', '\n\n'))
             
-            error_msg.empty()
-            proceed.empty()
             st.session_state['few_documents'] = False
             few_documents = False