Update main.py

main.py

@@ -13,10 +13,10 @@ userdata = {
 
 GEMINI_KEY = userdata.get("GEMINI_API_KEY")
 
-#PERSIST_DIR = os.path.dirname(os.path.abspath(__file__))
-DATABASE_PATH = "/app/RAG/chroma.sqlite3"
+PERSIST_DIR = os.path.dirname(os.path.abspath(__file__))
+#DATABASE_PATH = "/app/RAG/chroma.sqlite3"
 #os.path.join(PROJECT_ROOT, 'RAG/') # Make this configurable if needed
-v_store = load_vector_store(GEMINI_KEY, DATABASE_PATH)
+v_store = load_vector_store(GEMINI_KEY, PERSIST_DIR)
 retriever = Retriever(v_store)
 gemini_llm = GeminiLLM(GEMINI_KEY)