Update app.py

app.py

@@ -6,7 +6,7 @@ import uuid
 import streamlit as st
 
 sys.path.append(os.path.abspath("src"))
-from src.chatbot import generate_response, COUNTRY_HELPLINES, get_helpline_for_country
+from src.chatbot import generate_response_with_history, COUNTRY_HELPLINES, get_helpline_for_country
 
 # Set up logging for safeguard triggers
 logging.basicConfig(filename="safeguard_logs.txt", level=logging.INFO)
@@ -61,7 +61,7 @@ user_input = st.text_input("Ask me anything:")
 if st.button("Send"):
     if user_input.strip():
         # Generate chatbot response using country selection
-        response = generate_response(user_input, st.session_state.selected_country)
+        response = generate_response_with_history(user_input, st.session_state.selected_country)
 
         # Store conversation
         st.session_state.chat_history.append(f"You: {user_input}")