Update app.py

app.py

@@ -38,12 +38,16 @@ previous_response = ""
 provided_docs = ""
 def conversational_chat(query):
     global previous_response, provided_docs
-    for i in st.session_state['history']:
+    for i in st.session_state['history'][-5:]:
         if i is not None:
             previous_response += f"Human: {i[0]}\n Chatbot: {i[1]}\n"
     provided_docs = "".join(st.session_state["docs"])
     result = llm_chain.predict(chat_history=previous_response, human_input=query, provided_docs=provided_docs)
     st.session_state['history'].append((query, result))
+    
+    # Keep only the last 5 history entries
+    st.session_state['history'] = st.session_state['history'][-5:]
+    
     return result
 
 st.title("CRETA 🤖")