Create Login Dialog

src/streamlit_app.py

@@ -55,12 +55,27 @@ def get_reponse(client, user_input):
             {"role": "assistant", "content": response_text})
 
 
+def login():
+    env_token = os.getenv("HF_TOKEN")
+    if env_token:
+        st.session_state.hf_token = env_token
+        st.rerun()
+    else:
+        st.title("🔐 Please input Hugging Face Token")
+        user_token = st.text_input("HF_TOKEN", type="password")
+        if st.button("✅ Login"):
+            if user_token:
+                st.session_state.hf_token = user_token
+                st.rerun()
+            else:
+                st.warning("Please enter a valid token.")
+
+
 def chatbot():
     model_id = "openai/gpt-oss-20b"
-    hf_token = os.getenv("HF_TOKEN")
 
     # Initialize InferenceClient
-    client = InferenceClient(token=hf_token, model=model_id)
+    client = InferenceClient(token=st.session_state.hf_token, model=model_id)
 
     # Initialize chat history
     if "chat_history" not in st.session_state:
@@ -91,5 +106,15 @@ def chatbot():
             st.markdown(f"**{role}:** {msg['content']}")
 
 
+def main():
+    if "hf_token" not in st.session_state:
+        st.session_state.hf_token = None
+
+    if st.session_state.hf_token:
+        chatbot()
+    else:
+        login()
+
+
 if __name__ == "__main__":
-    chatbot()
+    main()