updating login feature

app.py

@@ -1,4 +1,6 @@
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
 import uuid
 import streamlit as st
@@ -20,7 +22,8 @@ st.markdown(
 )
 
 # --- Hugging Face Login ---
-default_hf_token = os.getenv("HUGGINGFACE_API_KEY") or ""
+# Use session state for hf_token if it exists, otherwise fallback to env.
+default_hf_token = st.session_state.get("hf_token", os.getenv("HUGGINGFACE_API_KEY") or "")
 hf_token = st.sidebar.text_input("Enter your Hugging Face API Token", type="password", value=default_hf_token)
 
 if "hf_logged_in" not in st.session_state:
@@ -33,6 +36,7 @@ if st.sidebar.button("Login to Hugging Face"):
             user_info = whoami()
             st.sidebar.success(f"Logged in as: {user_info['name']}")
             st.session_state.hf_logged_in = True
+            st.session_state.hf_token = hf_token  # Persist the API key in session state.
         except Exception as e:
             st.sidebar.error(f"Login failed: {e}")
             st.session_state.hf_logged_in = False

pages/Client_Response.py

@@ -1,6 +1,5 @@
 import os
-from dotenv import load_dotenv
-load_dotenv()
+
 
 import streamlit as st
 import pandas as pd

pages/Evaluation_Report.py

@@ -1,6 +1,4 @@
 import os
-from dotenv import load_dotenv
-load_dotenv()
 
 import streamlit as st
 import pandas as pd

pages/Select_Best.py

@@ -1,6 +1,4 @@
 import os
-from dotenv import load_dotenv
-load_dotenv()
 
 import streamlit as st
 import pandas as pd