Update app.py

app.py

@@ -354,43 +354,34 @@ if uploaded_file is not None:
             plt.xticks(rotation=45)
             st.pyplot(fig2)
 
-
     st.subheader("Word Sequence Viewer")
     
-    # Initialize session state
     if 'word_index' not in st.session_state:
         st.session_state.word_index = 0
     
-    # Pre-process data for faster access
     available_folios = sorted(set(line_data['folio'] for line_data in word_positions))
     selected_folio = st.selectbox("Select Folio:", [''] + available_folios, key='folio_select')
     
-    # Add jump-to input
-    col1, col2 = st.columns([2, 1])
-    with col1:
-        selected_folio = st.selectbox("Select Folio:", [''] + available_folios, key='folio_select')
-    with col2:
-        jump_to = st.number_input("Jump to word:", min_value=1, max_value=len(filtered_words) if filtered_words else 1)
-        if jump_to != st.session_state.word_index + 1:
-            st.session_state.word_index = jump_to - 1
-    
-    # Filter words more efficiently
+    # Pre-process data with line information
     filtered_words = []
+    line_boundaries = []  # Store indices where new lines start
+    current_line = None
+    
     if selected_folio:
-        filtered_words = [(word, chars) 
-                         for line_data in word_positions 
-                         if line_data['folio'] == selected_folio 
-                         for word, _, chars in line_data['words']]
+        for line_data in word_positions:
+            if line_data['folio'] == selected_folio:
+                if current_line != (line_data['par'], line_data['line']):
+                    line_boundaries.append(len(filtered_words))
+                    current_line = (line_data['par'], line_data['line'])
+                filtered_words.extend([(word, chars, line_data) for word, _, chars in line_data['words']])
     else:
-        filtered_words = list(zip(words, chars_list))
+        filtered_words = [(word, chars, None) for word, chars in zip(words, chars_list)]
     
     if filtered_words:
-        current_word, current_chars = filtered_words[st.session_state.word_index]
+        current_word, current_chars, line_info = filtered_words[st.session_state.word_index]
         
-        # Display word info
         st.write(f"Word {st.session_state.word_index + 1} of {len(filtered_words)}")
         
-        # Display character grid
         cols = st.columns(12)
         for i in range(12):
             with cols[i]:
@@ -411,29 +402,39 @@ if uploaded_file is not None:
                     </div>
                     """, unsafe_allow_html=True)
         
-        # Navigation with page size options
+        def get_next_line_index():
+            for boundary in line_boundaries:
+                if boundary > st.session_state.word_index:
+                    return boundary
+            return len(filtered_words) - 1
+            
+        def get_prev_line_index():
+            for boundary in reversed(line_boundaries):
+                if boundary < st.session_state.word_index:
+                    return boundary
+            return 0
+        
         col1, col2, col3, col4 = st.columns([1,1,1,2])
         with col1:
-            if st.button("◀◀ -10", disabled=st.session_state.word_index < 10):
-                st.session_state.word_index = max(0, st.session_state.word_index - 10)
+            if st.button("◀ Previous Line", disabled=st.session_state.word_index <= 0):
+                st.session_state.word_index = get_prev_line_index()
                 st.rerun()
         
         with col2:
-            if st.button("◀ Previous", disabled=st.session_state.word_index <= 0):
+            if st.button("Previous Word", disabled=st.session_state.word_index <= 0):
                 st.session_state.word_index -= 1
                 st.rerun()
         
         with col3:
-            if st.button("Next ▶", disabled=st.session_state.word_index >= len(filtered_words) - 1):
+            if st.button("Next Word", disabled=st.session_state.word_index >= len(filtered_words) - 1):
                 st.session_state.word_index += 1
                 st.rerun()
                 
-            if st.button("+10 ▶▶", disabled=st.session_state.word_index >= len(filtered_words) - 10):
-                st.session_state.word_index = min(len(filtered_words) - 1, st.session_state.word_index + 10)
+            if st.button("Next Line ▶", disabled=st.session_state.word_index >= len(filtered_words) - 1):
+                st.session_state.word_index = get_next_line_index()
                 st.rerun()
         
         with col4:
-            line_info = next((line_data for line_data in word_positions 
-                            if any(word == current_word for word, _, _ in line_data['words'])), None)
             if line_info:
                 st.write(f"Folio: {line_info['folio']}, Par: {line_info['par']}, Line: {line_info['line']}")
+