davideuler commited on
Commit
d3f93de
·
1 Parent(s): 7c278e2

reset session_state after new file uploaded

Browse files
Files changed (1) hide show
  1. pdf_translator_web.py +12 -0
pdf_translator_web.py CHANGED
@@ -232,8 +232,20 @@ def main():
232
  with st.sidebar:
233
  st.header("Settings")
234
 
 
 
235
  uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
236
 
 
 
 
 
 
 
 
 
 
 
237
  # Add source language selection
238
  source_lang_name = st.selectbox(
239
  "Source Language",
 
232
  with st.sidebar:
233
  st.header("Settings")
234
 
235
+ # Store previous file name to detect changes
236
+ previous_file = st.session_state.get('previous_file', None)
237
  uploaded_file = st.file_uploader("Choose a PDF file", type="pdf")
238
 
239
+ # Reset session state when a new file is uploaded
240
+ if uploaded_file is not None and (previous_file is None or uploaded_file.name != previous_file):
241
+ # Reset all relevant session state variables
242
+ st.session_state.current_page = 0
243
+ st.session_state.translation_started = True
244
+ st.session_state.all_translated = False
245
+ st.session_state.translated_doc = None
246
+ st.session_state.previous_file = uploaded_file.name
247
+ st.rerun()
248
+
249
  # Add source language selection
250
  source_lang_name = st.selectbox(
251
  "Source Language",