Update app.py

app.py

@@ -354,113 +354,49 @@ if uploaded_file is not None:
             plt.xticks(rotation=45)
             st.pyplot(fig2)
 
-    st.subheader("Word Sequence Viewer")
+       st.subheader("Word Sequence Viewer")
     
-    if 'word_index' not in st.session_state:
-        st.session_state.word_index = 0
+    # Initialize session states
+    if 'current_folio' not in st.session_state:
+        st.session_state.current_folio = ''
+    if 'current_line' not in st.session_state:
+        st.session_state.current_line = ''
     
+    # Get available folios
     available_folios = sorted(set(line_data['folio'] for line_data in word_positions))
-    selected_folio = st.selectbox("Select Folio:", [''] + available_folios, key='folio_select')
-    
-    # Pre-process data with line information
-    filtered_words = []
-    line_boundaries = []  # Store indices where new lines start
-    current_line = None
+    selected_folio = st.selectbox("Select Folio:", [''] + available_folios, 
+                                 key='folio_select',
+                                 on_change=lambda: setattr(st.session_state, 'current_line', ''))
     
+    # Get available lines for selected folio
+    available_lines = []
     if selected_folio:
-        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 = [(word, chars, None) for word, chars in zip(words, chars_list)]
-    
-    if filtered_words:
-        current_word, current_chars, line_info = filtered_words[st.session_state.word_index]
-        
-        st.write(f"Word {st.session_state.word_index + 1} of {len(filtered_words)}")
-        
-        cols = st.columns(12)
-        for i in range(12):
-            with cols[i]:
-                char = current_chars[i] if i < len(current_chars) else ""
-                st.markdown(f"""
-                    <div style='
-                        width: 40px;
-                        height: 40px;
-                        border: 2px solid #ccc;
-                        display: flex;
-                        align-items: center;
-                        justify-content: center;
-                        font-size: 20px;
-                        background-color: {"#e6f3ff" if char else "white"};
-                        margin: 2px;
-                    '>
-                        {char}
-                    </div>
-                    """, unsafe_allow_html=True)
-        
-        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("◀ 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 Word", disabled=st.session_state.word_index <= 0):
-                st.session_state.word_index -= 1
-                st.rerun()
-        
-        with col3:
-            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("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:
-            if line_info:
-                st.write(f"Folio: {line_info['folio']}, Par: {line_info['par']}, Line: {line_info['line']}")
-
-    # Add this after the current word viewer
-    st.subheader("Current Line View")
-    
-    if filtered_words:
-        current_line_info = line_info
-        current_line_words = []
+        available_lines = [(line_data['par'], line_data['line']) 
+                          for line_data in word_positions 
+                          if line_data['folio'] == selected_folio]
+        available_lines = sorted(set(available_lines))
+    
+    # Line selector
+    selected_line = st.selectbox("Select Line:", 
+                                [''] + [f"Par {par}, Line {line}" for par, line in available_lines])
+    
+    # Display selected line's words
+    if selected_folio and selected_line:
+        par, line = map(int, selected_line.replace('Par ', '').replace('Line ', '').split(', '))
         
-        # Get all words from the current line
-        for word_data in word_positions:
-            if (word_data['folio'] == current_line_info['folio'] and 
-                word_data['par'] == current_line_info['par'] and 
-                word_data['line'] == current_line_info['line']):
-                current_line_words = word_data['words']
-                break
+        # Get words for selected line
+        line_words = next((line_data['words'] 
+                          for line_data in word_positions 
+                          if line_data['folio'] == selected_folio 
+                          and line_data['par'] == par 
+                          and line_data['line'] == line), [])
         
-        # Display each word in the line with its slot sequence
-        st.write(f"Line {current_line_info['line']} in Paragraph {current_line_info['par']}")
-        
-        for word, _, chars in current_line_words:
+        # Display each word in the line
+        for word, _, chars in line_words:
             st.write(f"Word: {word}")
-            word_cols = st.columns(12)
+            cols = st.columns(12)
             for i in range(12):
-                with word_cols[i]:
+                with cols[i]:
                     char = chars[i] if i < len(chars) else ""
                     st.markdown(f"""
                         <div style='
@@ -477,3 +413,56 @@ if uploaded_file is not None:
                             {char}
                         </div>
                         """, unsafe_allow_html=True)
+
+    st.subheader("Current Line View")
+    
+    st.subheader("Line Viewer")
+    
+    # Get available folios
+    available_folios = sorted(set(line_data['folio'] for line_data in word_positions))
+    selected_folio = st.selectbox("Select Folio for Line View:", [''] + available_folios)
+    
+    # Get available lines for selected folio
+    if selected_folio:
+        available_lines = [(line_data['par'], line_data['line']) 
+                          for line_data in word_positions 
+                          if line_data['folio'] == selected_folio]
+        available_lines = sorted(set(available_lines))
+        
+        # Line selector
+        selected_line = st.selectbox("Select Line:", 
+                                   [''] + [f"Par {par}, Line {line}" for par, line in available_lines])
+        
+        # Display selected line's words
+        if selected_line:
+            par, line = map(int, selected_line.replace('Par ', '').replace('Line ', '').split(', '))
+            
+            # Get words for selected line
+            line_words = next((line_data['words'] 
+                             for line_data in word_positions 
+                             if line_data['folio'] == selected_folio 
+                             and line_data['par'] == par 
+                             and line_data['line'] == line), [])
+            
+            # Display each word in the line
+            for word, _, chars in line_words:
+                st.write(f"Word: {word}")
+                cols = st.columns(12)
+                for i in range(12):
+                    with cols[i]:
+                        char = chars[i] if i < len(chars) else ""
+                        st.markdown(f"""
+                            <div style='
+                                width: 40px;
+                                height: 40px;
+                                border: 2px solid #ccc;
+                                display: flex;
+                                align-items: center;
+                                justify-content: center;
+                                font-size: 20px;
+                                background-color: {"#e6f3ff" if char else "white"};
+                                margin: 2px;
+                            '>
+                                {char}
+                            </div>
+                            """, unsafe_allow_html=True)