Update app.py

app.py

@@ -29,19 +29,26 @@ st.write("Enter your input text:")
 # User input
 user_input = st.text_area("Input Text")
 
+def end_conv():
+    if st.button("End Conversation"):
+        st.write("Conversation ended.")
+        memory.clear()
+
+while st.button("End Conversation")!=True:
+    user_input = st.text_area("Input Text")
 # Generate and display the response
-if st.button("Generate Response"):
+    if st.button("Generate Response"):
     # Load current conversation history
-    history = memory.load_memory_variables({})['history']
+        history = memory.load_memory_variables({})['history']
 
     # Invoke the chain to get the response
-    res = chain.invoke({"input": user_input})
-    response_content = res.content
-    st.write("Generated Response:")
-    st.write(response_content)
+        res = chain.invoke({"input": user_input})
+        response_content = res.content
+        st.write("Generated Response:")
+        st.write(response_content)
 
         # Save the context in memory
-    memory.save_context({"input": user_input}, {"output": response_content})
+        memory.save_context({"input": user_input}, {"output": response_content})
 
         # Display the conversation history
     #st.write("Conversation History:")
@@ -49,8 +56,7 @@ if st.button("Generate Response"):
     #    st.write(f"{msg['role']}: {msg['content']}")
 
     # Optionally, you can also add a button to reset the conversation history
-if st.button("End Conversation"):
-    st.write("Conversation ended.")
-    memory.clear()
+    if st.button("End Conversation"):
+        end_conv()