Update main.py

main.py

@@ -14,22 +14,20 @@ userdata = {
 GEMINI_KEY = userdata.get("GEMINI_API_KEY")
 
 
-import sqlite3
+# import sqlite3
 
-DATABASE_PATH = "/app/RAG/chroma.sqlite3"
+# DATABASE_PATH = "/app/RAG/chroma.sqlite3"
 
-try:
-    conn = sqlite3.connect(DATABASE_PATH, check_same_thread=False)
-    print("Database connection successful!")
-except sqlite3.OperationalError as e:
-    print(f"Database connection failed: {e}")
+# try:
+#     conn = sqlite3.connect(DATABASE_PATH, check_same_thread=False)
+#     print("Database connection successful!")
+# except sqlite3.OperationalError as e:
+#     print(f"Database connection failed: {e}")
 
 
 
 
-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
+PERSIST_DIR = "/app/RAG/chroma_db"
 v_store = load_vector_store(GEMINI_KEY, PERSIST_DIR)
 retriever = Retriever(v_store)
 gemini_llm = GeminiLLM(GEMINI_KEY)