Update app.py

app.py

@@ -98,19 +98,16 @@ def create_conversational_rag_chain(vectorstore):
     rag_chain = create_retrieval_chain(ha_retriever, question_answer_chain)
     msgs = StreamlitChatMessageHistory(key="special_app_key")
 
-    @st.cache_resource
-    def create_rag_chain():
-        return RunnableWithMessageHistory(
-            rag_chain,
-            lambda session_id: msgs,
-            input_messages_key="input",
-            history_messages_key="chat_history",
-            output_messages_key="answer",
-        )
-
-    return create_rag_chain()
-
-def main(conversational_rag_chain):
+    conversation_chain = RunnableWithMessageHistory(
+        rag_chain,
+        lambda session_id: msgs,
+        input_messages_key="input",
+        history_messages_key="chat_history",
+        output_messages_key="answer",
+    )
+    return conversation_chain
+
+def main():
     """Main function for the Streamlit app."""
     # Initialize chat history if not already present in session state
     msgs = st.session_state.get("chat_history", StreamlitChatMessageHistory(key="special_app_key"))