Spaces:
Running
Running
“botayla” commited on
Commit ·
e92f08b
1
Parent(s): 0b64e1c
update
Browse files
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 |
-
|
|
|
|
| 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}")
|