Sulaiman8 commited on
Commit
8ccdfc2
·
verified ·
1 Parent(s): 6976103

Update recommender/recommender.py

Browse files
Files changed (1) hide show
  1. recommender/recommender.py +4 -3
recommender/recommender.py CHANGED
@@ -13,7 +13,7 @@ from recommender.graph_retrieval_vectordb import generate_cypher,run_cypher_quer
13
  def recommend_cards_gradio(user_query, preferences, income, cibil, age,
14
  min_joining_fee, max_joining_fee,
15
  min_annual_fee, max_annual_fee,
16
- use_eligibility=True,include_cobranded=True):
17
  try:
18
  # print(user_query)
19
  if(user_query):
@@ -59,6 +59,7 @@ def recommend_cards_gradio(user_query, preferences, income, cibil, age,
59
  {},
60
  "Card not found"
61
  )
 
62
  direct_query,excluded_cards = convert_to_direct_query_gradio(user_query, preferences, all_card_names=all_card_names,feature_list=features)
63
  queries = generate_multi_queries(direct_query)
64
 
@@ -67,7 +68,7 @@ def recommend_cards_gradio(user_query, preferences, income, cibil, age,
67
  else:
68
  query_intent = find_intent(user_query)
69
  print(query_intent)
70
- cypher_query = generate_cypher(direct_query, query_intent,include_cobranded)
71
  print("Generated Cypher:\n", cypher_query)
72
 
73
  try:
@@ -75,7 +76,7 @@ def recommend_cards_gradio(user_query, preferences, income, cibil, age,
75
  user_query, cypher_query, use_eligibility,
76
  income, cibil, age,
77
  min_joining_fee, max_joining_fee,
78
- min_annual_fee, max_annual_fee, excluded_cards
79
  )
80
  except Neo4jConnectionError as graph_err:
81
  return (
 
13
  def recommend_cards_gradio(user_query, preferences, income, cibil, age,
14
  min_joining_fee, max_joining_fee,
15
  min_annual_fee, max_annual_fee,
16
+ use_eligibility=True,):
17
  try:
18
  # print(user_query)
19
  if(user_query):
 
59
  {},
60
  "Card not found"
61
  )
62
+
63
  direct_query,excluded_cards = convert_to_direct_query_gradio(user_query, preferences, all_card_names=all_card_names,feature_list=features)
64
  queries = generate_multi_queries(direct_query)
65
 
 
68
  else:
69
  query_intent = find_intent(user_query)
70
  print(query_intent)
71
+ cypher_query = generate_cypher(direct_query, query_intent)
72
  print("Generated Cypher:\n", cypher_query)
73
 
74
  try:
 
76
  user_query, cypher_query, use_eligibility,
77
  income, cibil, age,
78
  min_joining_fee, max_joining_fee,
79
+ min_annual_fee, max_annual_fee,excluded_cards
80
  )
81
  except Neo4jConnectionError as graph_err:
82
  return (