42 lines
1014 B
Python
42 lines
1014 B
Python
from datetime import datetime
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class RecommendationRequest(BaseModel):
|
|
playlist_id: int | None = None
|
|
query: str | None = None # Manual search/request
|
|
bandcamp_mode: bool = False # Prioritize Bandcamp/indie artists
|
|
|
|
|
|
class RecommendationItem(BaseModel):
|
|
id: int
|
|
title: str
|
|
artist: str
|
|
album: str | None = None
|
|
spotify_id: str | None = None
|
|
preview_url: str | None = None
|
|
image_url: str | None = None
|
|
bandcamp_url: str | None = None
|
|
reason: str
|
|
score: float | None = None
|
|
saved: bool = False
|
|
created_at: datetime
|
|
|
|
model_config = {"from_attributes": True}
|
|
|
|
|
|
class RecommendationResponse(BaseModel):
|
|
recommendations: list[RecommendationItem]
|
|
remaining_today: int | None = None # None = unlimited (pro)
|
|
|
|
|
|
class TasteProfile(BaseModel):
|
|
top_genres: list[dict]
|
|
avg_energy: float
|
|
avg_danceability: float
|
|
avg_valence: float
|
|
avg_tempo: float
|
|
era_preferences: list[str]
|
|
mood_summary: str
|