|
|
import os |
|
|
import streamlit as st |
|
|
import torch |
|
|
import pandas as pd |
|
|
|
|
|
def get_device_map() -> str: |
|
|
return 'cuda' if torch.cuda.is_available() else 'cpu' |
|
|
device = get_device_map() |
|
|
|
|
|
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; */ |
|
|
/* }} */ |
|
|
}} |
|
|
|
|
|
|
|
|
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 __name__ == "__main__": |
|
|
st.markdown(custom_css, unsafe_allow_html=True) |
|
|
st.set_page_config(page_title="Sticky Header test", layout="wide") |
|
|
setup_page() |
|
|
|
