Update app.py

app.py

@@ -149,8 +149,9 @@ def display_documents(docs, on_click=None):
 
 def main(conversational_rag_chain):
     """Main function for the Streamlit app."""
-    msgs = st.session_state.get("chat_history", StreamlitChatMessageHistory())  # Initialize chat history
-    chain_with_history =conversational_rag_chain
+    # Initialize chat history if not already present in session state
+    msgs = st.session_state.get("chat_history", StreamlitChatMessageHistory(key="special_app_key"))
+    chain_with_history = conversational_rag_chain
 
     st.title("Conversational RAG Chatbot")
 
@@ -160,10 +161,13 @@ def main(conversational_rag_chain):
     if prompt := st.chat_input():
         st.chat_message("human").write(prompt)
 
-        # Process user input
+        # Prepare the input dictionary with the correct keys
+        input_dict = {"input": prompt, "chat_history": msgs.messages}
         config = {"configurable": {"session_id": "any"}}
-        response = chain_with_history.invoke({"question": prompt}, config)
-        st.chat_message("ai").write(response.content)
+
+        # Process user input and handle response
+        response = chain_with_history.invoke(input_dict, config)
+        st.chat_message("ai").write(response["answer"])
 
         # Display retrieved documents (if any and present in response)
         if "docs" in response and response["documents"]:
@@ -173,7 +177,8 @@ def main(conversational_rag_chain):
                 st.write(f"Expanding document {index+1}...")
             display_documents(docs, expand_document)  # Pass click function
 
-    st.session_state["chat_history"] = msgs  # Update chat history in session state
+    # Update chat history in session state
+    st.session_state["chat_history"] = msgs
 
 if __name__ == "__main__":
     main(conversational_rag_chain)