Spaces:
Running
Running
“botayla” commited on
Commit ·
2d86cc8
1
Parent(s): f2d5b66
add youtube api
Browse files- src/recommendation/recommender.py +8 -2
- src/utils/config.py +1 -0
src/recommendation/recommender.py
CHANGED
|
@@ -9,6 +9,11 @@ from src.db.models import Note
|
|
| 9 |
from src.utils.logger import setup_logger
|
| 10 |
from src.utils.config import settings
|
| 11 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
logger = setup_logger(__name__)
|
| 13 |
|
| 14 |
|
|
@@ -19,9 +24,10 @@ class RecommendationService:
|
|
| 19 |
"""
|
| 20 |
|
| 21 |
def __init__(self, api_key: Optional[str] = None):
|
| 22 |
-
|
|
|
|
| 23 |
# Use the newer google-genai client
|
| 24 |
-
self.client = genai.Client(api_key=self.api_key)
|
| 25 |
self.youtube = build("youtube", "v3", developerKey=self.api_key)
|
| 26 |
|
| 27 |
async def get_recommendations_for_user(
|
|
|
|
| 9 |
from src.utils.logger import setup_logger
|
| 10 |
from src.utils.config import settings
|
| 11 |
|
| 12 |
+
import os
|
| 13 |
+
from dotenv import load_dotenv
|
| 14 |
+
load_dotenv()
|
| 15 |
+
|
| 16 |
+
|
| 17 |
logger = setup_logger(__name__)
|
| 18 |
|
| 19 |
|
|
|
|
| 24 |
"""
|
| 25 |
|
| 26 |
def __init__(self, api_key: Optional[str] = None):
|
| 27 |
+
|
| 28 |
+
self.api_key = api_key or os.getenv("YOUTUBE_API_KEY")
|
| 29 |
# Use the newer google-genai client
|
| 30 |
+
# self.client = genai.Client(api_key=self.api_key)
|
| 31 |
self.youtube = build("youtube", "v3", developerKey=self.api_key)
|
| 32 |
|
| 33 |
async def get_recommendations_for_user(
|
src/utils/config.py
CHANGED
|
@@ -30,6 +30,7 @@ class Settings(BaseSettings):
|
|
| 30 |
description="Whisper model size (larger = more accurate but slower)"
|
| 31 |
)
|
| 32 |
|
|
|
|
| 33 |
# Processing Limits
|
| 34 |
max_video_duration: int = Field(
|
| 35 |
default=7200,
|
|
|
|
| 30 |
description="Whisper model size (larger = more accurate but slower)"
|
| 31 |
)
|
| 32 |
|
| 33 |
+
|
| 34 |
# Processing Limits
|
| 35 |
max_video_duration: int = Field(
|
| 36 |
default=7200,
|