“botayla” commited on
Commit
e92f08b
·
1 Parent(s): 0b64e1c
Files changed (1) hide show
  1. src/recommendation/recommender.py +4 -3
src/recommendation/recommender.py CHANGED
@@ -3,7 +3,7 @@ from typing import List, Dict, Optional
3
  from googleapiclient.discovery import build
4
  from src import db
5
  from src.utils.logger import setup_logger
6
-
7
  import os
8
  from dotenv import load_dotenv
9
  load_dotenv()
@@ -90,7 +90,7 @@ class RecommendationService:
90
  .list(
91
  q=enhanced_query,
92
  part="snippet",
93
- maxResults=limit,
94
  type="video",
95
  relevanceLanguage="en",
96
  videoEmbeddable="true",
@@ -115,7 +115,8 @@ class RecommendationService:
115
  logger.info(f"✅ Found video: {snippet['title']}")
116
 
117
  logger.info(f"🚀 Total videos fetched: {len(videos)}")
118
- return videos
 
119
 
120
  except Exception as e:
121
  logger.error(f"❌ YouTube search failed: {e}")
 
3
  from googleapiclient.discovery import build
4
  from src import db
5
  from src.utils.logger import setup_logger
6
+ import random
7
  import os
8
  from dotenv import load_dotenv
9
  load_dotenv()
 
90
  .list(
91
  q=enhanced_query,
92
  part="snippet",
93
+ maxResults=limit*3, # fetch more to filter later
94
  type="video",
95
  relevanceLanguage="en",
96
  videoEmbeddable="true",
 
115
  logger.info(f"✅ Found video: {snippet['title']}")
116
 
117
  logger.info(f"🚀 Total videos fetched: {len(videos)}")
118
+ random.shuffle(videos)
119
+ return videos[:limit]
120
 
121
  except Exception as e:
122
  logger.error(f"❌ YouTube search failed: {e}")