add password for hf space

config.py

@@ -15,6 +15,8 @@ class Config(BaseSettings):
     embedding_version: str
     embedding_dir: str
     model_name: str
+    # HF
+    password: str
 
 
 load_dotenv()
@@ -26,4 +28,5 @@ config = Config(
     embedding_version=getenv("EMBEDDING_VERSION"),
     embedding_dir=getenv("EMBEDDING_DIR"),
     model_name=getenv("MODEL_NAME"),
+    password=getenv("PASSWORD"),
 )

main.py

@@ -44,21 +44,28 @@ def _get_embeddings(text: str):
 
 st.title("PubMed Embeddings")
 st.subheader("Search for a PubMed article and get its id.")
-    
-text = st.text_input("Search for a PubMed article", "Epidemiology of COVID-19")
+   
+
+with st.text_input("Password", type="password") as password:
+    if password == config.password:
+        st.write("Password correct!")
+        text = st.text_input("Search for a PubMed article", "Epidemiology of COVID-19")
 
-with st.spinner("Loading Embedding Model..."):
-    pinecone.init(api_key=config.pinecone_api_key, env=config.pinecone_env)
-    if "index" not in st.session_state:
-        st.session_state.index = pinecone.Index(config.pinecone_index)
-    if "tokenizer" not in st.session_state:          
-        st.session_state.tokenizer = AutoTokenizer.from_pretrained(config.model_name)
-    if "model" not in st.session_state:
-        st.session_state.model = AutoModel.from_pretrained(config.model_name)
+        with st.spinner("Loading Embedding Model..."):
+            pinecone.init(api_key=config.pinecone_api_key, env=config.pinecone_env)
+            if "index" not in st.session_state:
+                st.session_state.index = pinecone.Index(config.pinecone_index)
+            if "tokenizer" not in st.session_state:          
+                st.session_state.tokenizer = AutoTokenizer.from_pretrained(config.model_name)
+            if "model" not in st.session_state:
+                st.session_state.model = AutoModel.from_pretrained(config.model_name)
 
-if st.button("Search"):
-    with st.spinner("Searching..."):
-        results = search(text)
+        if st.button("Search"):
+            with st.spinner("Searching..."):
+                results = search(text)
 
-    for res in results["matches"]:
-        st.write(f"{res['id']} - confidence: {res['score']:.2f}")
+            for res in results["matches"]:
+                st.write(f"{res['id']} - confidence: {res['score']:.2f}")
+    else:
+        st.write("Password incorrect!")
+