Update app.py

app.py

@@ -26,6 +26,7 @@ prompt = PromptTemplate(
 
 # Initialize memory
 memory = ConversationBufferMemory()
+memory.save_context({"input": "I need some help"}, {"output": "ok! how can i help you."})
 
 # Create the LLM chain
 llm_chain = LLMChain(
@@ -34,6 +35,12 @@ llm_chain = LLMChain(
     memory=memory
 )
 
+
+def generate_response(user_input):
+    response = llm_chain.invoke({"history":memory.chat_memory.messages, 'user_input':user_input})
+    return response
+
+
 # Streamlit app
 st.title("AI Chatbot")
 st.write("Welcome to the AI Chatbot! Ask anything you like.")
@@ -44,11 +51,9 @@ user_input = st.text_input("You:", key="input")
 if st.button("Send"):
     if user_input:
         # Generate response
-        response = llm_chain.invoke({"history": memory.chat_memory.messages, 'user_input': user_input})
+        response = generate_response(user_input)
         response_text = response['text']
         # Display the response
         st.text_area("ChatBot:", response_text, height=100)
-
-        st.write('----------------------------------------------------------------------------------------------')
         st.write('History:')
         st.write(response['history'])