Update pages

Streamlit/pages/PDF_to_TSV.py

@@ -228,10 +228,13 @@ def upPage():
 
 
 def uploadZip():
-    st.write(st.session_state.general_text_dict['text'])
-    st.write(st.session_state.general_text_dict['text2'])
-    return st.file_uploader(
-        st.session_state.general_text_dict['file'], type=["zip"], key='file')
+    with st.form("Submit"):
+        st.write(st.session_state.general_text_dict['text'])
+        st.write(st.session_state.general_text_dict['text2'])
+        st.file_uploader(
+            st.session_state.general_text_dict['file'], type=["zip"], key='file')
+        st.form_submit_button(
+            st.session_state.general_text_dict['submit'], on_click=setSubmit())
 
 
 def askPDF(fileName):
@@ -256,21 +259,22 @@ def askPDF(fileName):
 
 # Page Code End
 
-if 'getFile' not in st.session_state:
-    st.session_state.getFile = False
-
 if st.session_state.page == 0:
-    if st.session_state.getFile == None:
-        st.session_state.zipDir = tempfile.TemporaryDirectory()
-        st.session_state.pdfDir = tempfile.TemporaryDirectory()
-        st.session_state.tsv = "authors\tsource\tpublication_year\tpublication_month\tpublication_day\ttitle\tabstract"
-
-        st.session_state.page = 1
-        extractAllPDF(st.session_state.zipDir.name, st.session_state.file)
-        st.session_state.len = len(os.listdir(st.session_state.zipDir.name))
-        st.session_state.nbDoc = 0
+    if st.session_state.submit:
+        st.session_state.submit = False
+        if st.session_state.file != None:
+            st.session_state.zipDir = tempfile.TemporaryDirectory()
+            st.session_state.pdfDir = tempfile.TemporaryDirectory()
+            st.session_state.tsv = "authors\tsource\tpublication_year\tpublication_month\tpublication_day\ttitle\tabstract"
+            st.session_state.page = 1
+            extractAllPDF(st.session_state.zipDir.name, st.session_state.file)
+            st.session_state.len = len(
+                os.listdir(st.session_state.zipDir.name))
+            st.session_state.nbDoc = 0
+        else:
+            uploadZip()
     else:
-        st.session_state.getFile = uploadZip()
+        uploadZip()
 
 
 if st.session_state.page == 2:

Streamlit/pages/PDF_to_TXT.py

@@ -107,6 +107,7 @@ if 'submit' not in st.session_state:
 
 if st.session_state.page == 0:
     if st.session_state.submit:
+        st.session_state.submit = False
         if st.session_state.file != None:
             st.session_state.page = 1
         else:

Streamlit/pages/TXT_to_TSV.py

@@ -161,6 +161,8 @@ def askTXT():
 
     with st.form("Submit"):
         st.write(st.session_state.general_text_dict['text'])
+        st.write(st.session_state.general_text_dict['text2'])
+        st.write(st.session_state.general_text_dict['text3'])
         col1, col2 = st.columns(2)
         st.session_state.author = ""
         with col1:
@@ -182,6 +184,7 @@ if 'submit' not in st.session_state:
 
 if st.session_state.page == 0:
     if st.session_state.submit:
+        st.session_state.submit = False
         if st.session_state.file != None:
             print(st.session_state.file)
             st.session_state.page = 1