Bypass login dialog if we already logged in by hf auth command

README.md

@@ -12,9 +12,47 @@ short_description: Test docker sdk streamlit template
 license: mit
 ---
 
-# Welcome to Streamlit!
+# Welcome to Chatbot!
 
-Edit `/src/streamlit_app.py` to customize this app to your heart's desire. :heart:
+The following command installs required packages of the chatbot.
 
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
+```
+pip install -r requirements.txt
+```
+
+The following commands provides API token to the chatbot. Choose from one of them.
+
+- Login Hugging Face Hub
+
+```
+hf auth login
+```
+
+- Set HF_TOKEN environment variable
+
+```
+export HF_TOKEN=<Your Hugging Face Token>
+```
+
+The following command runs the chatbot.
+
+```
+streamlit run src/streamlit_app.py
+```
+
+Use browser to access the chatbot after you see the following message.
+
+```
+  You can now view your Streamlit app in your browser.
+
+  Local URL: http://localhost:8501
+  Network URL: http://172.28.148.72:8501
+```
+
+Use Crtl+C to stop the chatbot.
+
+Logout Hugging Face Hub if you have logged in.
+
+```
+hf auth logout
+```

src/streamlit_app.py

@@ -1,7 +1,7 @@
 import os
 
 import streamlit as st
-from huggingface_hub import InferenceClient
+from huggingface_hub import HfFolder, InferenceClient
 
 chat_parameters = {
     "system_prompt": "You are a friendly Chatbot.",
@@ -62,8 +62,13 @@ def get_reponse(client, user_input):
 
 
 def login():
+    api_token = HfFolder.get_token()
     env_token = os.getenv("HF_TOKEN")
-    if env_token:
+
+    if api_token:
+        st.session_state.hf_token = api_token
+        switch_page()
+    elif env_token:
         st.session_state.hf_token = env_token
         switch_page()
     else: