Update verifier.py

verifier.py

@@ -26,8 +26,16 @@ def verifier_page():
     
     fields_to_fetch = ['filename', 'pdf_url', 'text']
 
+    # Initialize session state variables if not present
+    if 'selected_file' not in st.session_state:
+        st.session_state.selected_file = None
+    if 'selected_text' not in st.session_state:
+        st.session_state.selected_text = ""
+    if 'selected_ai' not in st.session_state:
+        st.session_state.selected_ai = ""
+
     # Check if a file has already been selected and displayed
-    if 'selected_file' in st.session_state:
+    if st.session_state.selected_file:
         # When a file is selected, clear everything and display the details only
         st.empty()  # Clear all content before displaying details
         st.title(f"Selected File: {st.session_state.selected_file}")
@@ -57,9 +65,9 @@ def verifier_page():
                         st.session_state.selected_text = row['text']
 
                         # Clear existing content on button click
-                        st.rerun()
+                        st.experimental_rerun()
 
-        if 'selected_file' in st.session_state:
+        if st.session_state.selected_file:
             # Process the selected file and generate AI response
             pdf_path = 'VCS-Standard.pdf'
             start_page = 0  # Start extracting from the first page (0-based index)