Update app.py

app.py

@@ -16,47 +16,31 @@ prompt=ChatPromptTemplate.from_messages([
      MessagesPlaceholder(variable_name="history"),
      ("human","{input}")])
 
-
-memory=ConversationBufferWindowMemory(k=3,return_messages=True)
-
-chain=(RunnablePassthrough.assign(history=RunnableLambda(memory.load_memory_variables)|
-       itemgetter("history"))|prompt|model)
-
-# Streamlit interface
-st.title("chat bot")
-st.write("Enter your input text:")
-
-
-def end_conv():
-    st.write("Conversation ended.")
-    
-
-    # Initialize session state for conversation history if not already done
-
-
-    # User input
-user_input = st.text_area("Input Text")
-
-    # Generate and display the response
-if st.button("Generate Response"):
-        # Load current conversation 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)
-
-        # Save the context in memory and session state
-    memory.save_context({"input": user_input}, {"output": response_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']}")
-
-    # End conversation button
-    if st.button("End Conversation"):
-        end_conv()
+# Initialize memory in session state
+if 'memory' not in st.session_state:
+    st.session_state.memory = ConversationBufferWindowMemory(k=10, return_messages=True)
+
+# Define the chain
+chain = (RunnablePassthrough.assign(history=RunnableLambda(st.session_state.memory.load_memory_variables) | itemgetter("history")) |
+         prompt | model)
+
+# Streamlit app
+st.title("Interactive Chatbot")
+
+# Initialize session state for user input
+if 'user_input' not in st.session_state:
+    st.session_state.user_input = ""
+
+# Input from user
+user_input = st.text_area("User: ", st.session_state.user_input, height=100)
+
+if st.button("Submit"):
+    response = chain.invoke({"input": user_input})
+    st.write(f"Assistant: {response.content}")
+    st.session_state.memory.save_context({"input": user_input}, {"output": response.content})
+    st.session_state.user_input = ""  # Clear the input box
+
+# Display chat history
+if st.checkbox("Show Chat History"):
+    chat_history = st.session_state.memory.load_memory_variables({})
+    st.write(chat_history)