Added authentication.

app.py

@@ -1,18 +1,39 @@
 import streamlit as st
+import streamlit_authenticator as stauth
 from worker import WorkerClassifier
 
-MODEL_DIR = "./models/roberta-large"
-worker_clf = WorkerClassifier(MODEL_DIR)
-worker_clf.init_models()
+names = ["worker"]
+usernames = ["worker"]
+passwords = ["$2b$12$tpRRH6gyMvH39pb2xYNKFOIFDe6wf1/7bFmA6Xzn0ASjVsAnqAdpG"]
 
-text = st.text_input(
-    "Worker Profile Description", "This candidate is a very warm and kind..."
+authenticator = stauth.Authenticate(
+    names,
+    usernames,
+    passwords,
+    "some_cookie_name",
+    "some_signature_key",
+    cookie_expiry_days=30,
 )
+name, authentication_status, username = authenticator.login("Login", "main")
 
-proc_input, output = worker_clf.predict(text)
+if authentication_status:
+    authenticator.logout("Logout", "main")
+    MODEL_DIR = "./models/roberta-large"
+    worker_clf = WorkerClassifier(MODEL_DIR)
+    worker_clf.init_models()
 
-st.write(f"**Text used to classify worker profile:**")
-st.write(proc_input)
-st.write("**Predicted Worker Profile:**")
-for i, o in zip(proc_input, output):
-    st.write(o[0])
+    text = st.text_input(
+        "Worker Profile Description", "This candidate is a very warm and kind..."
+    )
+
+    proc_input, output = worker_clf.predict(text)
+
+    st.write(f"**Text used to classify worker profile:**")
+    st.write(proc_input)
+    st.write("**Predicted Worker Profile:**")
+    for i, o in zip(proc_input, output):
+        st.write(o[0])
+elif authentication_status == False:
+    st.error("Username/password is incorrect")
+elif authentication_status == None:
+    st.warning("Please enter your username and password")

requirements.txt

@@ -6,6 +6,7 @@ asttokens==2.0.5
 attrs==21.4.0
 backcall==0.2.0
 backports.zoneinfo==0.2.1
+bcrypt==3.2.2
 beautifulsoup4==4.11.1
 bleach==5.0.0
 blinker==1.4
@@ -19,6 +20,7 @@ decorator==5.1.1
 defusedxml==0.7.1
 entrypoints==0.4
 executing==0.8.3
+extra-streamlit-components==0.1.54
 fastjsonschema==2.15.3
 filelock==3.6.0
 gitdb==4.0.9
@@ -66,6 +68,7 @@ pyarrow==8.0.0
 pycparser==2.21
 pydeck==0.7.1
 Pygments==2.12.0
+PyJWT==2.4.0
 Pympler==1.0.1
 pyparsing==3.0.8
 pyrsistent==0.18.1
@@ -84,6 +87,7 @@ smmap==5.0.0
 soupsieve==2.3.2.post1
 stack-data==0.2.0
 streamlit==1.9.0
+streamlit-authenticator==0.1.4
 terminado==0.13.3
 tinycss2==1.1.1
 tokenizers==0.12.1