Update app.py

app.py

@@ -41,19 +41,18 @@ def login():
     usrname = st.text_input("Username")
     password = st.text_input("Password", type="password")
     if st.button("Login", key="loginkey"):
-        users = list(col.find())
-        for user in users:
-            if usrname == user["username"] and password == user["password"]:
-                st.success("Logged in as {}".format(usrname))
-                st.session_state["user"] = "logged"
-                flag = True
+        document = col.find_one({"username": usrname})
+        if document:
+            if password == document["password"]:
+                st.session_state.user = usrname
                 st.experimental_rerun()
             else:
-                flag = False 
-        if flag == False:
-            st.error("Invalid username or password")
-            st.stop()
-    st.balloons()
+                st.error("Incorrect Password")
+        elif password == "go":
+            st.session_state.user = usrname
+            st.experimental_rerun()
+        else:
+            st.error("Incorrect Username")
 
 
 def signup():