Update app.py

app.py

@@ -50,30 +50,52 @@ 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):
+    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 = load_pdf_text(file_path)
+        chunks = text_splitter.split_text(text=text)
+        embeddings = OpenAIEmbeddings()
+        VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
+
+        # 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}")
+
+        # Change working directory for Git operations
+        original_dir = os.getcwd()
+        os.chdir(local_repo_path)
+        
+        try:
+            # Check current working directory and list files for debugging
+            st.text(f"Current working directory: {os.getcwd()}")
+            st.text(f"Files in current directory: {os.listdir()}")
+        
+            # Adjusted file path for Git command
+            repo.git_add(f"{store_name}.pkl")  # Use just the file name
+            repo.git_commit(f"Update vector store: {store_name}")
+            repo.git_push()
+            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
 
-        # Check if we need to force reload the vector store (e.g., when the PDF changes)
-        if force_reload or not os.path.exists(f"{store_name}.pkl"):
-            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()
-            VectorStore = FAISS.from_texts(chunks, embedding=embeddings)
-            with open(f"{store_name}.pkl", "wb") as f:
-                pickle.dump(VectorStore, f)
-        else:
-            with open(f"{store_name}.pkl", "rb") as f:
-                VectorStore = pickle.load(f)
-    
-        return VectorStore
 
 # Utility function to load text from a PDF
 def load_pdf_text(file_path):