Update app.py

app.py

@@ -5,18 +5,12 @@ from freegpt import Client
 def get_response(prompt):
     try:
         response = client.create_completion("gpt3", prompt)
-        return response.text.strip()
+        st.chat_message("You", user_input, avatar=None)
+        st.chat_message("KviGPT", response, avatar=None)
     except Exception as e:
         return "Error: {}".format(str(e))
 
 st.sidebar.title("KviGPT", help="By KVI Kontent")
 st.sidebar.subheader("Chat based on Gpt-3")
-user_input = st.chat_input("Say something")
+user_input = st.chat_input("Say something", on_submit=get_response)
 generate_btn = st.button("Generate Response")
-
-if generate_btn:
-    response = get_response(user_input)
-    st.chat_message("You", user_input, avatar=None)
-    st.chat_message("KviGPT", response, avatar=None)
-else:
-    st.write("Please provide a prompt to generate a response.")