Update app.py

app.py

@@ -11,9 +11,6 @@ import os
 from getpass import getpass
 from langchain import HuggingFaceHub
 from langchain_community.llms import HuggingFaceEndpoint
-from langchain_core.messages import AIMessage, HumanMessage
-from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-from langchain.memory import ChatMessageHistory
 
 
 
@@ -88,29 +85,19 @@ llm = HuggingFaceEndpoint(repo_id='google/gemma-7b',
 if 'entity_memory' not in st.session_state:
     st.session_state.entity_memory = ConversationEntityMemory(llm=llm, k=K )
     
-# Create the ConversationChain object with the specified configuration
-conversation = ConversationChain(llm=llm,,
+    # Create the ConversationChain object with the specified configuration
+Conversation = ConversationChain(llm=llm, 
+                                 prompt=ENTITY_MEMORY_CONVERSATION_TEMPLATE,
                                  memory=st.session_state.entity_memory
     )  
 
 
-#create prompt
-prompt = ChatPromptTemplate.from_messages([
-    ("system", "You are a helpful assistant. Answer all questions to the best of your ability."),
-    MessagesPlaceholder(variable_name="messages"),
-])
-
-#create the chain
-
-chain = prompt | conversation
-
-
 # Get the user input
 user_input = get_text()
 
 # Generate the output using the ConversationChain object and the user input, and add the input/output to the session
 if user_input:
-    output = chain.invoke({"messages": user_input}  
+    output = Conversation.run(input=user_input)  
     st.session_state.past.append(user_input)  
     st.session_state.generated.append(output)  
 
@@ -131,4 +118,4 @@ for i, sublist in enumerate(st.session_state.stored_session):
 # Allow the user to clear all stored conversation sessions
 if st.session_state.stored_session:   
     if st.sidebar.checkbox("Clear-all"):
-        del st.session_state.stored_session
+        del st.session_state.stored_session