Update app.py

app.py

@@ -15,11 +15,8 @@ import sys
 import pandas as pd
 
 def conversational_chat(query):    
-    result = chain({"question": query, 
-    "chat_history": st.session_state['history']})
-    st.session_state['history'].append((query, result["answer"]))
-
-    return result["answer"]
+    output = llm_chain.predict(human_input=query)
+    return output
 
 
 user_api_key = st.sidebar.text_input(
@@ -32,10 +29,16 @@ if user_api_key is not None and user_api_key.strip() != "":
 
     #setting up the LLM 
     repo_id = "tiiuae/falcon-7b-instruct"
-    chain = ConversationalRetrievalChain.from_llm(
-    llm = HuggingFaceHub(huggingfacehub_api_token=user_api_key, 
-    repo_id=repo_id, 
-    model_kwargs={"temperature":0.6, "max_new_tokens":2000}))
+    prompt = PromptTemplate(
+        input_variables=["history", "human_input"],
+        template=template
+    )
+    llm_chain = LLMChain(
+        llm=HuggingFaceHub(huggingfacehub_api_token=huggingfacehub_api_token, repo_id="tiiuae/falcon-7b-instruct", model_kwargs={"temperature": 0.2}),
+        prompt=prompt,
+        verbose=True,
+        memory=ConversationBufferWindowMemory(k=2)
+    )
     
 
     if 'history' not in st.session_state: