Update app.py

app.py

@@ -10,12 +10,9 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 import os
 
 # Set up the directories for data and vector DB
-DATA_DIR = "/content/MyData"
-DB_DIR = "/content/VectorDB"
+DATA_DIR = "MyData"
+DB_DIR = "VectorDB"
 
-# Create directories if they don't exist
-os.makedirs(DATA_DIR, exist_ok=True)
-os.makedirs(DB_DIR, exist_ok=True)
 
 # Initialize the embeddings model
 embeddings_model = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
@@ -34,7 +31,7 @@ def load_data():
     return vectorstore
 
 # Set up the generative AI model
-llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro", google_api_key="YOUR_GOOGLE_API_KEY")
+llm = ChatGoogleGenerativeAI(model="gemini-1.5-pro", google_api_key="AIzaSyAnsIVS4x_7lJLe9AYXGLV8FRwUTQkB-1w")
 
 # Load vector store
 vectorstore = load_data()