Update app.py

app.py

@@ -10,16 +10,18 @@ dataset = load_dataset(dataset_repo_id, split='train', streaming=True)
 # Create an iterator from the dataset
 dataset_iter = iter(dataset)
 
-# Read the last entry from the dataset
-try:
-    last_entry = next(dataset_iter)  # Use dataset_iter here
-except StopIteration:
-    last_entry = {'standup_taker': 'Yiannis', 'date': '2023-01-01'}
-
 # Initialize login state
 if 'logged_in' not in st.session_state:
     st.session_state.logged_in = False
 
+# Initialize last_entry in session_state
+if 'last_entry' not in st.session_state:
+    try:
+        st.session_state.last_entry = next(dataset_iter)
+    except StopIteration:
+        st.session_state.last_entry = {'standup_taker': 'Yiannis', 'date': '2023-01-01'}
+
+
 # Login Screen
 if not st.session_state.logged_in:
     st.title("Login to Stand-Up Taker App")
@@ -36,12 +38,9 @@ if not st.session_state.logged_in:
 # Main App
 if st.session_state.logged_in:
     st.title("Stand-Up Taker App")
-    st.write("This app determines who takes the stand-up on Tuesdays and Thursdays.")
-    
-    # Read the last entry from the dataset
-    last_entry = next(dataset)
-    last_standup_taker = last_entry['standup_taker']
-    last_date = last_entry['date']
+
+    last_standup_taker = st.session_state.last_entry['standup_taker']
+    last_date = st.session_state.last_entry['date']
     
     # Button to check today's stand-up taker
     if st.button("Who is taking the stand-up today?"):