Update data.py

data.py

@@ -1,8 +1,20 @@
 import json
 import pandas as pd
-from recommender.graph_retrieval_vectordb import NEO4J_PASS,NEO4J_URI,NEO4J_USER,get_driver
+from neo4j import GraphDatabase
+
+#neo4j credentials
+NEO4J_URI = os.environ.get("NEO4J_URI")
+NEO4J_USER = os.environ.get("NEO4J_USER")
+NEO4J_PASS = os.environ.get("NEO4J_PASS")
 
 _driver=None
+
+def get_driver():
+    global _driver
+    if _driver is None:
+        _driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USER, NEO4J_PASS))
+    return _driver
+
 #for adding bank name to the cards in the graph
 eligibility_df = pd.read_csv("cards_eligibility_updated.csv")
 card_to_bank = dict(zip(eligibility_df['Name'], eligibility_df['Bank']))