Update pages/type_text.py

pages/type_text.py

@@ -23,6 +23,12 @@ 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
+
+
 #df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows.csv") 
 df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv") 
 
@@ -77,6 +83,7 @@ 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)
 
 
@@ -171,8 +178,6 @@ pipe = load_pipe()
 INTdesc_embedding = model.encode(INTdesc_input)
 SBScorpus_embeddings = model.encode(SBScorpus)
 
-
-
 if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model"):     
     HF_model_results = util.semantic_search(INTdesc_embedding, SBScorpus_embeddings)
     HF_model_results_sorted = sorted(HF_model_results, key=lambda x: x[1], reverse=True)
@@ -189,22 +194,27 @@ 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)
 
-    # JavaScript component too automatically scroll to bottom of page
-    st.components.v1.html(
-        """
-        <script>
-            window.onload = function() {
-                window.scrollTo(0, document.body.scrollHeight);
-            }
-            // For immediate execution as well
-            window.scrollTo(0, document.body.scrollHeight);
-        </script>
-        """,
-        height=0
-    )
 
     display_format = "ask REASONING MODEL: Which, if any, of the following SBS descriptions corresponds best to " + INTdesc_input +"? " 
     #st.write(display_format)
@@ -229,19 +239,24 @@ if INTdesc_input is not None and st.button("Map to SBS codes", key="run_st_model
     )
     st.write(outputs[0]["generated_text"][-1]["content"]) 
 
-    # JavaScript component too automatically scroll to bottom of page
-    st.components.v1.html(
-        """
-        <script>
-            window.onload = function() {
-                window.scrollTo(0, document.body.scrollHeight);
-            }
-            // For immediate execution as well
-            window.scrollTo(0, document.body.scrollHeight);
-        </script>
-        """,
-        height=0
-    )
+    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
+        )    
+
     
     bs, b1, b2, b3, bLast = st.columns([0.75, 1.5, 1.5, 1.5, 0.75])
     with b1: