Update verifier.py

verifier.py

@@ -28,46 +28,15 @@ def verifier_page():
 
     # 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 = ""
+        st.session_state.selected_file = ""
+    #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 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.write(f"Selected File: {st.session_state.selected_file}")
-        #st.write(f"AI Response: {st.session_state.selected_ai}")
-    else:
-        # Automatically load and display the data table on page load
-        data = fetch_data(fields_to_fetch)
-        
-        if data:
-            df = pd.DataFrame(data)
-    
-            # Display the DataFrame
-            st.dataframe(df)
-
-            # Add buttons for each row in the DataFrame
-            for index, row in df.iterrows():
-                col1, col2 = st.columns([4, 1])
-                with col1:
-                    st.write(row['filename'])
-    
-                with col2:
-                    # Button to view details of each row
-                    button_key = f"view_{index}"
-                    if st.button("See details", key=button_key):
-                        # Update the session state with the selected file details
-                        st.session_state.selected_file = row['text']
-                        #st.session_state.selected_text = row['text']
 
-                        # Clear existing content on button click
-                        st.rerun()
 
-        if st.session_state.selected_file:
+    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)
@@ -102,6 +71,34 @@ def verifier_page():
             st.write(f"Selected File: {st.session_state.selected_file}")
             #st.write(f"AI Response: {st.session_state.selected_ai}")
 
+        
+    else:
+        # Automatically load and display the data table on page load
+            data = fetch_data(fields_to_fetch)
+            if data:
+                df = pd.DataFrame(data)
+        
+                # Display the DataFrame
+            st.dataframe(df)
+    
+                # Add buttons for each row in the DataFrame
+            for index, row in df.iterrows():
+                col1, col2 = st.columns([4, 1])
+                with col1:
+                    st.write(row['filename'])
+        
+                with col2:
+                    # Button to view details of each row
+                    button_key = f"view_{index}"
+                    if st.button("See details", key=button_key):
+                        # Update the session state with the selected file details
+                        st.session_state.selected_file = row['text']
+                        #st.session_state.selected_text = row['text']
+    
+                        # Clear existing content on button click
+                        st.rerun()
+                            
+
 # Helper function to extract text from PDF
 def extract_text_from_pdf(uploaded_file, start_page, end_page):
     if uploaded_file is None: