Update verifier.py

verifier.py

@@ -1,63 +1,64 @@
-import streamlit as st
-import firebase_admin
-from firebase_admin import credentials, firestore
-import pandas as pd
-
-# Initialize Firebase
-cred = credentials.Certificate('serviceAccountKey.json')
-if not firebase_admin._apps:
-    firebase_admin.initialize_app(cred)
-
-# Create a Firestore client
-db = firestore.client()
-
-# Fetch and display submissions
-def fetch_data(selected_fields):
-    docs = db.collection('pdf_uploads').stream()
-    data = []
-    for doc in docs:
-        doc_dict = doc.to_dict()
-        filtered_data = {field: doc_dict.get(field) for field in selected_fields if field in doc_dict}
-        data.append(filtered_data)
-    return data
-
-fields_to_fetch = ['filename', 'pdf_url', 'text']
-
-# Create an empty placeholder for the data table and file details
-data_placeholder = st.empty()
-
-# Check if a file has already been selected and displayed
-if 'selected_file' in st.session_state:
-    # When a file is selected, display details and hide the table
-    st.title(f"File Details: {st.session_state.selected_file}")
-    st.write(f"Text: {st.session_state.selected_text}")
-else:
-    # Automatically load and display the data table on page load
-    data = fetch_data(fields_to_fetch)
-    if data:
-        df = pd.DataFrame(data)
-
-        # Use the placeholder to display the DataFrame
-        with data_placeholder:
-            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(f"View {row['filename']}", key=button_key):
-                    # Update the session state with the selected file details
-                    st.session_state.selected_file = row['filename']
-                    st.session_state.selected_text = row['text']
-
-                    # Clear the data table and show file details
-                    data_placeholder.empty()
-
-                    # Display the file details
-                    st.write(f"Selected File: {st.session_state.selected_file}")
-                    st.write(f"Text: {st.session_state.selected_text}")
+def verifier_page():
+    import streamlit as st
+    import firebase_admin
+    from firebase_admin import credentials, firestore
+    import pandas as pd
+    
+    # Initialize Firebase
+    cred = credentials.Certificate('serviceAccountKey.json')
+    if not firebase_admin._apps:
+        firebase_admin.initialize_app(cred)
+    
+    # Create a Firestore client
+    db = firestore.client()
+    
+    # Fetch and display submissions
+    def fetch_data(selected_fields):
+        docs = db.collection('pdf_uploads').stream()
+        data = []
+        for doc in docs:
+            doc_dict = doc.to_dict()
+            filtered_data = {field: doc_dict.get(field) for field in selected_fields if field in doc_dict}
+            data.append(filtered_data)
+        return data
+    
+    fields_to_fetch = ['filename', 'pdf_url', 'text']
+    
+    # Create an empty placeholder for the data table and file details
+    data_placeholder = st.empty()
+    
+    # Check if a file has already been selected and displayed
+    if 'selected_file' in st.session_state:
+        # When a file is selected, display details and hide the table
+        st.title(f"File Details: {st.session_state.selected_file}")
+        st.write(f"Text: {st.session_state.selected_text}")
+    else:
+        # Automatically load and display the data table on page load
+        data = fetch_data(fields_to_fetch)
+        if data:
+            df = pd.DataFrame(data)
+    
+            # Use the placeholder to display the DataFrame
+            with data_placeholder:
+                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(f"View {row['filename']}", key=button_key):
+                        # Update the session state with the selected file details
+                        st.session_state.selected_file = row['filename']
+                        st.session_state.selected_text = row['text']
+    
+                        # Clear the data table and show file details
+                        data_placeholder.empty()
+    
+                        # Display the file details
+                        st.write(f"Selected File: {st.session_state.selected_file}")
+                        st.write(f"Text: {st.session_state.selected_text}")