Update pages/type_text.py

pages/type_text.py

@@ -24,9 +24,62 @@ os.getenv("HF_TOKEN")
 #st.title("Map internal descriptions to SBS codes with Sentence Transformer + Reasoning Models")
 #st.subheader("Select specific Chapter for quicker results")
 
-# Initialize session state if not already done
-if 'should_scroll' not in st.session_state:
-    st.session_state.should_scroll = False
+def get_device_map() -> str:
+    return 'cuda' if torch.cuda.is_available() else 'cpu'
+device = get_device_map()  # 'cpu'
+
+def on_click():
+    st.session_state.user_input = "" 
+
+def make_spinner(text = "In progress..."):
+    with st.spinner(text):
+        yield
+    
+#@st.cache
+def convert_df(df:pd.DataFrame):
+     return df.to_csv(index=False).encode('utf-8')
+
+#@st.cache
+def convert_json(df:pd.DataFrame):
+    result = df.to_json(orient="index")
+    parsed = json.loads(result)
+    json_string = json.dumps(parsed)
+    #st.json(json_string, expanded=True)
+    return json_string
+
+def auto_scroll_to_bottom():
+    # JavaScript to scroll to bottom
+    js_code = """
+    <script>
+        // Wait for the page to fully render
+        const scrollToBottom = () => {
+            // Get the main Streamlit iframe
+            const streamlitDoc = window.parent.document;
+            // Get the app container
+            const appContainer = streamlitDoc.querySelector('.main');
+            if (appContainer) {
+                // Scroll the app container to the bottom
+                appContainer.scrollTop = appContainer.scrollHeight;
+            } else {
+                // Fallback to scrolling the entire page
+                window.parent.scrollTo(0, streamlitDoc.body.scrollHeight);
+            }
+        };
+        
+        // Try immediately
+        scrollToBottom();
+        
+        // Also try after a short delay to ensure content is rendered
+        setTimeout(scrollToBottom, 200);
+        
+        // And after a longer delay just to be safe
+        setTimeout(scrollToBottom, 500);
+    </script>
+    """
+    
+    # Render the JavaScript code
+    html = f'<div style="display:none">{js_code}</div>'
+    st.components.v1.html(html, height=0)
 
 
 #df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows.csv") 
@@ -83,35 +136,12 @@ df_SBS = pd.read_csv("SBS_V2_0/Code_Sheet.csv", header=0, skip_blank_lines=False
 SBScorpus = df_SBS['Long_Description'].values.tolist()
 
 dictA = {"Score": [], "SBS Code": [], "SBS Description V2.0": []}
-st.session_state.should_scroll = True
 dfALL = pd.DataFrame.from_dict(dictA)
 
 
-def get_device_map() -> str:
-    return 'cuda' if torch.cuda.is_available() else 'cpu'
-device = get_device_map()  # 'cpu'
-
-def on_click():
-    st.session_state.user_input = "" 
-
-def make_spinner(text = "In progress..."):
-    with st.spinner(text):
-        yield
-    
-#@st.cache
-def convert_df(df:pd.DataFrame):
-     return df.to_csv(index=False).encode('utf-8')
-
-#@st.cache
-def convert_json(df:pd.DataFrame):
-    result = df.to_json(orient="index")
-    parsed = json.loads(result)
-    json_string = json.dumps(parsed)
-    #st.json(json_string, expanded=True)
-    return json_string
-
 #INTdesc_input = st.text_input("Type internal description", key="user_input") 
 INTdesc_input = st.text_input(r"$\textsf{\Large Type internal description}$", key="user_input")
+auto_scroll_to_bottom()
 
 placeholder, centerL_column, centerR_column, right_column = st.columns(4)
 #placeholder_clicked = placeholder.button("Perform some action", key="user_placeholder")
@@ -194,27 +224,9 @@ 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)
         dictA.update({"Score": "%.4f" % result[4]["score"], "SBS Code": df_SBS.loc[df_SBS["Long_Description"] == SBScorpus[result[4]["corpus_id"]],"SBS_Code_Hyphenated"].values[0], "SBS Description V2.0": SBScorpus[result[4]["corpus_id"]]})
         dfALL = pd.concat([dfALL, pd.DataFrame([dictA])], ignore_index=True)
-
-    if st.session_state.should_scroll:
-        # Reset the flag for next run
-        st.session_state.should_scroll = False
-        # Add the JavaScript for scrolling
-        st.components.v1.html(
-            """
-            <script>
-                // Scroll to bottom with highest priority
-                document.addEventListener("DOMContentLoaded", function(event) {
-                    window.parent.scrollTo(0, window.parent.document.body.scrollHeight);
-                });
-                // Also try immediately
-                window.parent.scrollTo(0, window.parent.document.body.scrollHeight);
-            </script>
-            """,
-            height=0
-        )    
-    
+ 
     st.dataframe(data=dfALL, hide_index=True)
-
+    auto_scroll_to_bottom()
 
     display_format = "ask REASONING MODEL: Which, if any, of the following SBS descriptions corresponds best to " + INTdesc_input +"? " 
     #st.write(display_format)
@@ -238,25 +250,7 @@ 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"]) 
-
-    if st.session_state.should_scroll:
-        # Reset the flag for next run
-        st.session_state.should_scroll = False
-        # Add the JavaScript for scrolling
-        st.components.v1.html(
-            """
-            <script>
-                // Scroll to bottom with highest priority
-                document.addEventListener("DOMContentLoaded", function(event) {
-                    window.parent.scrollTo(0, window.parent.document.body.scrollHeight);
-                });
-                // Also try immediately
-                window.parent.scrollTo(0, window.parent.document.body.scrollHeight);
-            </script>
-            """,
-            height=0
-        )    
-
+    auto_scroll_to_bottom()
     
     bs, b1, b2, b3, bLast = st.columns([0.75, 1.5, 1.5, 1.5, 0.75])
     with b1: