Jaykay73 commited on
Update recommender.py
Browse files- 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", "
|
| 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
|