Update app.py

app.py

@@ -31,7 +31,7 @@ def end_conv():
     st.write("Conversation ended.")
     memory.clear()
 
-while st.button("End Conversation")!=True:
+for i in range(1,5):
     user_input = st.text_area("Input Text")
 # Generate and display the response
     if st.button("Generate Response"):
@@ -53,7 +53,7 @@ while st.button("End Conversation")!=True:
     #    st.write(f"{msg['role']}: {msg['content']}")
 
     # Optionally, you can also add a button to reset the conversation history
-    #if st.button("End Conversation"):
-      #  end_conv()
+    if st.button("End Conversation"):
+        end_conv()