Update app.py

app.py

@@ -18,11 +18,11 @@ st.title("DialoGPT Chat")
 # Initialize chat history
 chat_history_ids = None
 
-# Conversation loop
-while st.button("Restart Conversation"):
-    # Get user input
-    user_input = st.text_input("You:", "")
+# User input text box
+user_input = st.text_input("You:", "")
 
+# Check if the user pressed Enter
+if st.button("Send"):
     if user_input:
         # Generate response
         chat_history_ids = generate_response(chat_history_ids, user_input)
@@ -31,6 +31,4 @@ while st.button("Restart Conversation"):
         st.text("DialoGPT: {}".format(tokenizer.decode(chat_history_ids[:, -1][0], skip_special_tokens=True)))
 
 # Inform the user that the conversation has ended
-st.text("Conversation Ended. Press the 'Restart Conversation' button to start a new conversation.")
-
-
+st.text("Press 'Send' to continue the conversation.")