Update app.py

app.py

@@ -42,23 +42,22 @@ if 'key' not in st.session_state:
 
 if st.session_state['key'] == st.secrets["login_key"]:
     user_input = get_text()
-else:
-    st.warning('wrong key!')
-
-if user_input:
-    prompt = 'The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.'
+    if user_input:
+        prompt = 'The following is a conversation with an AI assistant. The assistant is helpful, creative, clever, and very friendly.'
+        if st.session_state['generated']:
+            for past, generated in zip(st.session_state['past'], st.session_state['generated']):
+                if len(generated)>0:
+                    prompt += f'\nHuman:{past}\nAI:{generated}'
+        prompt += f'\nHuman:{user_input}\nAI:'
+        print(prompt)
+        output = openai_create(prompt)
+        print(f'{user_input} -> {output}')
+        st.session_state.past.append(user_input)
+        st.session_state.generated.append(output)
+    
     if st.session_state['generated']:
-        for past, generated in zip(st.session_state['past'], st.session_state['generated']):
-            if len(generated)>0:
-                prompt += f'\nHuman:{past}\nAI:{generated}'
-    prompt += f'\nHuman:{user_input}\nAI:'
-    print(prompt)
-    output = openai_create(prompt)
-    print(f'{user_input} -> {output}')
-    st.session_state.past.append(user_input)
-    st.session_state.generated.append(output)
-
-if st.session_state['generated']:
-    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')
+        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:
+    st.warning('wrong key!')