Update app_test_header.py

app_test_header.py

@@ -1,57 +1,19 @@
-import os
 import streamlit as st
 import torch
-import pandas as pd
+import uuid
+import nest_asyncio
+import asyncio
 
-def get_device_map() -> str:
-    return 'cuda' if torch.cuda.is_available() else 'cpu'
-device = get_device_map()  # 'cpu'
+torch.classes.__path__ = []
 
-custom_css = f"""
-    <style>
-        /* Making the Streamlit header sticky */
-        {HEADER_SELECTOR} {{
-            position: -webkit-sticky !important; /* For Safari */
-            position: sticky !important;
-            top: 0 !important;
-            z-index: 9999 !important; /* Very high z-index */
-            background-color: #90EE90 !important; /* Or your app's header background color */
-            /* Add a subtle shadow to make it feel more distinct when content scrolls under */
-            /* box-shadow: 0 2px 4px -1px rgba(0,0,0,0.1); */
-            /*.reportview-container .main .block-container{{ */
-            /*    padding-top: 0rem; */
-            /* }} */
-        }}
+if 'device_id' not in st.session_state:
+    st.session_state.device_id = str(uuid.uuid4())
 
-
-def setup_page():
-    with st.spinner("In progress..."):
-        hf_token = os.getenv('HF_TOKEN')
-        df_chapters = pd.read_csv("SBS_V2_0/Chapter_Index_Rows_with_total.csv") 
-        st.image("images/SBS_Chapter_Index.png", use_container_width=True) 
-        st.subheader("Select specific Chapter for quicker results")
-        st.write(df_chapters.head())    
-
-        /* Adding padding to the main content area to prevent overlap with the sticky header */
-        {MAIN_CONTENT_SELECTOR} {{
-            padding-top: 10px !important; /* {HEADER_HEIGHT} !important; */
-            /* background-color: yellow !important; */
-            /* border-style: solid !important; */
-            /* border-color: red !important; */
-        }}
-
-        /* Optional: If your app is set to wide mode and the header isn't spanning full width */
-        /* This might be needed if the sticky positioning affects its width calculation. */
-        /*
-        #{HEADER_SELECTOR} {{
-        #    width: 100% !important;
-        #}}
-        #*/
-    </style>
-    """
-        
+if "feedback_key" not in st.session_state:
+    st.session_state.feedback_key = 0
 
 if __name__ == "__main__":
-    st.markdown(custom_css, unsafe_allow_html=True) 
-    st.set_page_config(page_title="Sticky Header test", layout="wide")
-    setup_page()
+   st.set_page_config(page_title="Test Sticky Header", layout="wide")
+
+   nest_asyncio.apply()
+   asyncio.run(main())