Update app.py

app.py

@@ -36,19 +36,30 @@ pinecone_key = os.environ["PINECONE_API_KEY"]
 #     progress_bar=False,
 # )
 
-# Setup retriever, pulling from pinecone
-doc_file_name="cpv_full_southern_africa"
-document_store = PineconeDocumentStore(api_key=pinecone_key,
+
+@st.cache(allow_output_mutation=True)
+def get_document_store():
+    # Assuming you're using Elasticsearch as a document store
+    doc_file_name="cpv_full_southern_africa"
+    document_store = PineconeDocumentStore(api_key=pinecone_key,
                                        environment="asia-southeast1-gcp-free",
                                        index=doc_file_name)
+    return document_store
+
+@st.cache(allow_output_mutation=True)
+def get_retriever(document_store):
+    # Initialize the Pinecone retriever configuration here
+    retriever = EmbeddingRetriever(
+      document_store=document_store,
+      embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
+      model_format="sentence_transformers",
+      progress_bar=False,
+    )
+    return retriever
 
-retriever = EmbeddingRetriever(
-  document_store=document_store,
-  embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-  model_format="sentence_transformers",
-  progress_bar=False,
-)
-
+# Get (or initialize and get) the document store and retriever
+document_store = get_document_store()
+retriever = get_retriever(document_store)
 
 # template = PromptTemplate(
 #     prompt="""