Update pages/type_text.py

pages/type_text.py

@@ -1,4 +1,5 @@
 import streamlit as st 
+import streamlit.components.v1 as components
 import pandas as pd
 from io import StringIO
 import json
@@ -22,55 +23,49 @@ def make_spinner(text = "In progress..."):
     with st.spinner(text):
         yield
 
-def scroll_past_status_bar():
-    st.markdown("""
-    <script>
-        // This function finds the correct elements in Streamlit's DOM structure
-        function scrollPastStatusBar() {
-            try {
-                // Get the document of the parent (Streamlit app)
-                const doc = window.parent.document;
-                
-                // Find all the major containers
-                const appFrame = doc.querySelector('.stApp');
-                const mainSection = appFrame.querySelector('section.main');
-                const blockContainer = mainSection.querySelector('.block-container');
-                const stStatusWidget = doc.querySelector('.stStatusWidget');
-                
-                if (blockContainer && stStatusWidget) {
-                    // Get the height of the status widget
-                    const statusHeight = stStatusWidget.offsetHeight;
-                    
-                    // Calculate the position to scroll to
-                    // This will scroll to just above the status bar
-                    const scrollPosition = blockContainer.scrollHeight - window.innerHeight + statusHeight + 20;
-                    
-                    // Perform the scroll on the appropriate container
-                    if (mainSection.scrollTo) {
-                        mainSection.scrollTo({
-                            top: scrollPosition,
+def scroll_to_results():
+    # Create a hidden HTML component with JavaScript that specifically targets
+    # Streamlit's structure and accounts for the fixed status bar
+    components.html(
+        """
+        <script>
+            // Wait for the page to be fully loaded
+            window.onload = function() {
+                // Function to scroll with offset for status bar
+                function scrollToResults() {
+                    try {
+                        // Get the Streamlit app frame
+                        const streamlitApp = window.parent.document.querySelector('.stApp');
+                        if (!streamlitApp) return;
+                        
+                        // Find the main content area
+                        const mainContent = streamlitApp.querySelector('.main .block-container');
+                        if (!mainContent) return;
+                        
+                        // Calculate scroll position: full scroll minus space for status bar
+                        const statusBarHeight = 60; // Adjust based on your status bar height
+                        const scrollTarget = mainContent.scrollHeight - statusBarHeight;
+                        
+                        // Scroll the content area
+                        mainContent.scrollTo({
+                            top: scrollTarget,
                             behavior: 'smooth'
                         });
+                    } catch (e) {
+                        console.error("Error scrolling:", e);
                     }
-                    
-                    // Also try scrolling the window as a fallback
-                    window.parent.scrollTo({
-                        top: document.body.scrollHeight,
-                        behavior: 'smooth'
-                    });
                 }
-            } catch (e) {
-                console.error("Scroll error:", e);
-            }
-        }
-        
-        // Set multiple attempts with delays
-        setTimeout(scrollPastStatusBar, 100);
-        setTimeout(scrollPastStatusBar, 300);
-        setTimeout(scrollPastStatusBar, 600);
-    </script>
-    """, unsafe_allow_html=True)
-
+                
+                // Execute scroll attempts with delays
+                setTimeout(scrollToResults, 100);
+                setTimeout(scrollToResults, 500);
+                setTimeout(scrollToResults, 1000);
+            };
+        </script>
+        """,
+        height=0,
+        width=0,
+    )
 
 #@st.cache
 def convert_df(df:pd.DataFrame):
@@ -226,10 +221,11 @@ if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model
         dfALL = pd.concat([dfALL, pd.DataFrame([dictA])], ignore_index=True)
  
     st.dataframe(data=dfALL, hide_index=True)
-    # Add extra space at the bottom to ensure results aren't hidden by status bar
-    st.markdown("<div style='height: 70px;'></div>", unsafe_allow_html=True)    
-    # Call the scroll function
-    scroll_past_status_bar()
+    # Call the scroll function after displaying results
+    scroll_to_results()
+    # Add a small spacer to ensure there's room below the results
+    # (this helps prevent the last bit of content from being hidden by the status bar)
+    st.markdown("<div style='height: 60px;'></div>", unsafe_allow_html=True)
 
     display_format = "ask REASONING MODEL: Which, if any, of the following SBS descriptions corresponds best to " + INTdesc_input +"? " 
     #st.write(display_format)
@@ -253,11 +249,12 @@ if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model
         max_new_tokens=256,
     )
     st.write(outputs[0]["generated_text"][-1]["content"]) 
-    # Add extra space at the bottom to ensure results aren't hidden by status bar
-    st.markdown("<div style='height: 70px;'></div>", unsafe_allow_html=True)
-    # Call the scroll function
-    scroll_past_status_bar()
-    
+    # Call the scroll function after displaying results
+    scroll_to_results()
+    # Add a small spacer to ensure there's room below the results
+    # (this helps prevent the last bit of content from being hidden by the status bar)
+    st.markdown("<div style='height: 60px;'></div>", unsafe_allow_html=True)
+        
     bs, b1, b2, b3, bLast = st.columns([0.75, 1.5, 1.5, 1.5, 0.75])
     with b1:
         #csvbutton = download_button(results, "results.csv", "📥 Download .csv")