Update data.py

data.py

@@ -1,6 +1,14 @@
 import json
 import pandas as pd
 
+_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']))
@@ -38,7 +46,7 @@ for _, row in eligibility_df.iterrows():
     eligibility_lookup[card_name] = eligibility_info.strip()
 
 def get_all_features():
-    with driver.session() as session:
+    with get_driver().session() as session:
         result = session.run("MATCH (f:Feature) RETURN f.name AS feature")
         return [record["feature"] for record in result]