Update app.py

app.py

@@ -239,7 +239,8 @@ qa = RetrievalQA.from_chain_type(
         "prompt": PromptTemplate(
             template=prompt_template,
             input_variables=["context", "question", "chat_history"]
-        )
+        ),
+        "memory": st.session_state.conversation_memory,
     }
 )
 
@@ -334,17 +335,17 @@ if submit_button and query:
     # Show user message
     with st.chat_message("user", avatar="👤"):
         st.markdown(query)
-    
+        
     # Show typing indicator while generating response "🌿"
     with st.chat_message("assistant", avatar=robot_image):  
         with st.status("Analyzing your query...", expanded=True):
             st.write("🔍 Retrieving relevant information...")
             st.write("📝 Generating personalized response...")
-            
+            chat_history = st.session_state.conversation_memory.load_memory_variables({}).get("chat_history", "")
             # Generate response
             result = qa({
             "query": query,
-            "chat_history": st.session_state.conversation_memory.load_memory_variables({})["chat_history"]
+            "chat_history": chat_history if chat_history else ""
             })
             
             if result['result'].strip() == "":