SImple paging for entry selection to accommodate long lists

src/app.py

@@ -38,6 +38,41 @@ if "audio_preview_active" not in st.session_state:
 if "uploaded_file" not in st.session_state:
     st.session_state.results_file = None
 
+if "selected_entry_idx" not in st.session_state:
+    st.session_state.selected_entry_idx = 0
+if "total_entry_count" not in st.session_state:
+    st.session_state.total_entry_count = 0
+if "entry_page_size" not in st.session_state:
+    st.session_state.entry_page_size = 20
+
+
+def get_current_page_slice():
+    ss = st.session_state
+    if ss.total_entry_count == 0:
+        return slice(0, 0)
+
+    page_first_entry = (
+        st.session_state.selected_entry_idx // ss.entry_page_size
+    ) * ss.entry_page_size
+    page_last_entry = min(page_first_entry + ss.entry_page_size, ss.total_entry_count)
+
+    return slice(page_first_entry, page_last_entry)
+
+
+def page_navigation():
+    ss = st.session_state
+    current_page_slice = get_current_page_slice()
+    has_next_page = current_page_slice.stop < ss.total_entry_count - 1
+    has_prev_page = current_page_slice.start >= ss.entry_page_size
+
+    col1, col2 = st.columns(2)
+    if col1.button("Prev Page", disabled=not has_prev_page):
+        ss.selected_entry_idx = current_page_slice.start - 1
+        st.rerun()
+    if col2.button("Next Page", disabled=not has_next_page):
+        ss.selected_entry_idx = current_page_slice.start + ss.entry_page_size
+        st.rerun()
+
 
 def on_file_upload():
     st.session_state.audio_cache = {}
@@ -89,7 +124,8 @@ def choose_input_file_from_leaderboard():
 @st.cache_data
 def read_results_csv(uploaded_file):
     with st.spinner("Loading results...", show_time=True):
-        return pd.read_csv(uploaded_file)
+        results_df = pd.read_csv(uploaded_file)
+    return results_df
 
 
 def display_rtl(html):
@@ -284,6 +320,7 @@ def main():
         # Load the data
         try:
             eval_results = read_results_csv(uploaded_file)
+            st.session_state.total_entry_count = len(eval_results)
             st.success("File uploaded successfully!")
 
             with st.sidebar:
@@ -327,6 +364,12 @@ def main():
             col1.button("← Prev", on_click=go_prev, use_container_width=True)
             col2.button("Next →", on_click=go_next, use_container_width=True)
 
+            # Use a container for better styling
+            entry_container = st.sidebar.container()
+            with entry_container:
+                page_navigation()
+                st.write(f"Total entries: {st.session_state.total_entry_count}")
+
             # Create a data table with entries and their WER
             entries_data = []
             for i in range(len(eval_results)):
@@ -340,15 +383,13 @@ def main():
                 entries_data.append({"Entry": f"Entry #{i+1}", "WER": wer_formatted})
 
             # Create a selection mechanism using radio buttons that look like a table
-            st.sidebar.write("Select an entry:")
-
-            # Use a container for better styling
-            entry_container = st.sidebar.container()
+            st.sidebar.write("Select an entry")
 
             # Create a radio button for each entry, styled to look like a table row
+            current_page_slice = get_current_page_slice()
             entry_container.radio(
                 "Select an entry",
-                options=list(range(len(eval_results))),
+                options=list(range(len(eval_results))[current_page_slice]),
                 format_func=lambda i: f"Entry #{i+1} ({entries_data[i]['WER']})",
                 label_visibility="collapsed",
                 key="selected_entry_idx",