Update app.py

app.py

@@ -40,7 +40,7 @@ def get_text():
 if 'key' not in st.session_state:
     st.session_state['key'] = st.text_input("key: ", key="input1")
 
-if st.session_state['key'] == st.secrets["login_key"]:
+elif st.session_state['key'] == st.secrets["login_key"]:
     user_input = get_text()
     if user_input:
         prompt = 'The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.'
@@ -59,5 +59,5 @@ if st.session_state['key'] == st.secrets["login_key"]:
         for i in range(len(st.session_state['generated'])-1, -1, -1):
             message(st.session_state["generated"][i], key=str(i))
             message(st.session_state['past'][i], is_user=True, key=str(i) + '_user')
-else:
+elif st.session_state['key'] != st.secrets["login_key"]:
     st.warning('wrong key!')