Update app.py

app.py

@@ -50,19 +50,20 @@ api_key = os.getenv("OPENAI_API_KEY")
 
 
 
-# Updated caching mechanism using st.cache_data
-@st.cache_data(persist="disk")  # Using persist="disk" to save cache across sessions
+# Updated load_vector_store function with Streamlit text outputs and directory handling for Git
+@st.cache_data(persist="disk")
 def load_vector_store(file_path, store_name, force_reload=False):
-    vector_store_path = f"{store_name}.pkl"
+    local_repo_path = "Private_Book"
+    vector_store_path = os.path.join(local_repo_path, f"{store_name}.pkl")
 
     # Check if vector store already exists and force_reload is False
     if not force_reload and os.path.exists(vector_store_path):
         with open(vector_store_path, "rb") as f:
             VectorStore = pickle.load(f)
+        st.text(f"Loaded existing vector store from {vector_store_path}")
     else:
         # Load and process the PDF, then create the vector store
-        text_splitter = RecursiveCharacterTextSplitter(
-            chunk_size=1000, chunk_overlap=200, length_function=len)
+        text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200, length_function=len)
         text = load_pdf_text(file_path)
         chunks = text_splitter.split_text(text=text)
         embeddings = OpenAIEmbeddings()
@@ -71,11 +72,23 @@ def load_vector_store(file_path, store_name, force_reload=False):
         # Serialize the vector store
         with open(vector_store_path, "wb") as f:
             pickle.dump(VectorStore, f)
+        st.text(f"Created and saved vector store at {vector_store_path}")
 
-        # Commit and push changes to the repository
-        repo.git_add(vector_store_path)
-        repo.git_commit(f"Update vector store: {store_name}")
-        repo.git_push()
+        # Change working directory for Git operations
+        original_dir = os.getcwd()
+        os.chdir(local_repo_path)
+        
+        try:
+            # Commit and push changes to the repository
+            repo.git_add(vector_store_path)  # Stage the file for commit
+            repo.git_commit(f"Update vector store: {store_name}")  # Commit the file
+            repo.git_push()  # Push the commit to the remote repository
+            st.text("Committed and pushed vector store to repository.")
+        except Exception as e:
+            st.error(f"Error during Git operations: {e}")
+        finally:
+            # Change back to the original directory
+            os.chdir(original_dir)
 
     return VectorStore