Update app.py

app.py

@@ -48,39 +48,43 @@ pinecone_key = os.environ["PINECONE_API_KEY"]
 
 #___________________________________________________________________________________________________________
 
-@st.cache_resource 
-def get_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_resource 
+# def get_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
 
-# Get (or initialize and get) the document store
-document_store = get_document_store()
 
+# # Get (or initialize and get) the document store
+# document_store = get_document_store()
 
-# @st.cache(allow_output_mutation=True)
-# def get_retriever(document_store):
-#     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 = get_retriever(document_store)
-
-
-# Instantiate retriever
-retriever = EmbeddingRetriever(
-  document_store=document_store,
-  embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
-  model_format="sentence_transformers",
-  progress_bar=False,
-)
+@st.cache_resource
+def get_retriever():
+    doc_file_name="cpv_full_southern_africa"
+    document_store = PineconeDocumentStore(api_key=pinecone_key,
+                                       environment="asia-southeast1-gcp-free",
+                                       index=doc_file_name)
+    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 = get_retriever()
+
+
+# # Instantiate retriever
+# retriever = EmbeddingRetriever(
+#   document_store=document_store,
+#   embedding_model="sentence-transformers/multi-qa-mpnet-base-dot-v1",
+#   model_format="sentence_transformers",
+#   progress_bar=False,
+# )
 
 prompt_template="Answer the given question using the following documents. \
 Formulate your answer in the style of an academic report. \