Remove proceed button after finishing

app.py

@@ -70,6 +70,7 @@ 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']
 
@@ -101,8 +102,12 @@ def main():
     if few_documents:
         error_msg = st.empty()
         error_msg.warning(st.session_state['msg'])
-        msg = 'Prosseguir' if portuguese else 'Proceed'
-        if st.button(msg):
+        proceed = st.empty()
+        if portuguese:
+            proceed_button = proceed.button('Prosseguir')
+        else:
+            proceed_button = proceed.button('Proceed')
+        if proceed_button:
             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)
@@ -117,6 +122,7 @@ def main():
                     st.markdown(text.replace('\n', '\n\n'))
             
             error_msg.empty()
+            proceed.empty()
             st.session_state['few_documents'] = False
             few_documents = False