Update app.py

app.py

@@ -28,31 +28,36 @@ st.write("Enter your input text:")
 
 
 def end_conv():
-    st.write("Conversation ended.")
-    memory.clear()
+        st.write("Conversation ended.")
+        st.session_state.conversation_history = []
+
+    # Initialize session state for conversation history if not already done
+if 'conversation_history' not in st.session_state:
+    st.session_state.conversation_history = []
+
+    # User input
 user_input = st.text_area("Input Text")
-    
-# Generate and display the response
+
+    # Generate and display the response
 if st.button("Generate Response"):
-# Load current conversation history
+        # Load current conversation history
     history = memory.load_memory_variables({})['history']
 
-    # Invoke the chain to get the response
+        # Invoke the chain to get the response
     res = chain.invoke({"input": user_input})
-    response_content = res.content
+    response_content = res.get("content", "No content generated.")
     st.write("Generated Response:")
     st.write(response_content)
 
-        # Save the context in memory
+        # Save the context in memory and session state
     memory.save_context({"input": user_input}, {"output": response_content})
+    st.session_state.conversation_history.extend([{"role": "human", "content": user_input}, {"role": "assistant", "content": response_content}])
 
-        # Display the conversation history
-    #st.write("Conversation History:")
-    #for msg in memory.buffer:
-    #    st.write(f"{msg['role']}: {msg['content']}")
+        # Display the updated conversation history
+        #st.write("Conversation History:")
+        #for msg in st.session_state.conversation_history:
+         #   st.write(f"{msg['role']}: {msg['content']}")
 
-    # Optionally, you can also add a button to reset the conversation history
+    # End conversation button
     if st.button("End Conversation"):
-        end_conv()
-
-
+        end_conv()