Jaykay73 commited on
Commit
b42b068
·
unverified ·
1 Parent(s): 4a7d3f4

Update recommender.py

Browse files
Files changed (1) hide show
  1. src/recommender.py +2 -2
src/recommender.py CHANGED
@@ -65,7 +65,7 @@ class MovieRecommender:
65
 
66
  # 1. HAPPY / COMEDY MODE -> Ban Dark Stuff
67
  if any(w in query_lower for w in ["happy", "uplifting", "comedy", "laugh", "cheerful", "funny"]):
68
- return ["Horror", "Thriller", "War", "Crime", "Tragedy"]
69
 
70
  # 2. FAMILY / KIDS MODE -> Ban Adult Stuff
71
  if any(w in query_lower for w in ["family", "kid", "child", "animation", "disney"]):
@@ -172,4 +172,4 @@ class MovieRecommender:
172
  'title': movie_data['title'],
173
  'score': float(distances[0][i])
174
  })
175
- return results
 
65
 
66
  # 1. HAPPY / COMEDY MODE -> Ban Dark Stuff
67
  if any(w in query_lower for w in ["happy", "uplifting", "comedy", "laugh", "cheerful", "funny"]):
68
+ return ["Horror", "Thriller", "Crime", "Tragedy"]
69
 
70
  # 2. FAMILY / KIDS MODE -> Ban Adult Stuff
71
  if any(w in query_lower for w in ["family", "kid", "child", "animation", "disney"]):
 
172
  'title': movie_data['title'],
173
  'score': float(distances[0][i])
174
  })
175
+ return results