Update data.py

data.py

@@ -1,15 +1,8 @@
 import json
 import pandas as pd
-from neo4j import GraphDatabase
+from recommender.graph_retrieval_vectordb import NEO4J_PASS,NEO4J_URI,NEO4J_USER,get_driver
 
-_driver = None
-
-def get_driver():
-    global _driver
-    if _driver is None:
-        _driver = GraphDatabase.driver(NEO4J_URI, auth=(NEO4J_USER, NEO4J_PASS))
-    return _driver
-    
+_driver=None
 #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']))
@@ -46,10 +39,10 @@ for _, row in eligibility_df.iterrows():
     """
     eligibility_lookup[card_name] = eligibility_info.strip()
 
+   
 def get_all_features():
     with get_driver().session() as session:
         result = session.run("MATCH (f:Feature) RETURN f.name AS feature")
         return [record["feature"] for record in result]
 
 features = get_all_features()
-