Update app.py

app.py

@@ -47,18 +47,19 @@ if "search_tool" not in st.session_state:
         serper_api_key=str(os.getenv('SERPER_API')))
 
 
-def get_answer(query):
+def get_answer(conversation, query):
     new_search_query = st.session_state.llm.invoke(
         f"Convert below query to english for Ahmedabad Municipal Corporation (AMC) You just need to give translated query. Don't add any additional details.\n Query: {query}").content
     search_result = st.session_state.search_tool.run(
         f"{new_search_query} site:https://ahmedabadcity.gov.in/")
 
-    system_prompt = """You are a helpful assistance for The Ahmedabad Municipal Corporation (AMC). which asnwer user query from given context only. Output language should be as same as `original_query_from_user`.
+    system_prompt = """You are a helpful assistance for The Ahmedabad Municipal Corporation (AMC). which asnwer user query from given context only. Output language should be as same as language of `original_query_from_user`.
   context: {context}
   original_query_from_user: {original_query}
+  history: {history}
   query: {query}"""
 
-    return st.session_state.llm.invoke(system_prompt.format(context=search_result, query=new_search_query, original_query=query)).content
+    return st.session_state.llm.invoke(system_prompt.format(context=search_result, history=conversation, query=new_search_query, original_query=query)).content
 
 
 session_manager.set_session_state(st.session_state)
@@ -96,13 +97,13 @@ if prompt := st.chat_input("Enter your query here... "):
             if message['role'] == 'assistant':
                 current_conversation += f"""ai: {message['content']}\n"""
 
-        current_conversation += f"""user: {prompt}\n"""
+        # current_conversation += f"""user: {prompt}\n"""
 
         print("****************************************** Messages ******************************************")
         print("messages", current_conversation)
         print()
         print()
-        response = get_answer(current_conversation)
+        response = get_answer(current_conversation,prompt)
         print("******************************************************** Response ********************************************************")
         print("MY RESPONSE IS:", response)