Pick color extraction strategy from dropdown

This commit is contained in:
Claudio Ortolina
2025-06-08 20:01:49 +01:00
parent 2bbb91769d
commit 2cc1d7ca8f
10 changed files with 185 additions and 23 deletions
+4 -11
View File
@@ -6,7 +6,6 @@ defmodule MusicLibrary.Records do
import Ecto.Query, warn: false
alias MusicLibrary.Artists
alias MusicLibrary.Colors.EdgeWeightedExtractor
alias MusicLibrary.Records.{ArtistRecord, Cover, Record, SearchParser}
alias MusicLibrary.{BackgroundRepo, Repo, Worker}
@@ -255,18 +254,12 @@ defmodule MusicLibrary.Records do
|> BackgroundRepo.insert()
end
def generate_dominant_colors(record) do
with {:ok, colors} <- EdgeWeightedExtractor.extract_dominant_colors(record.cover_data) do
update_record(record, %{"dominant_colors" => colors})
end
end
def generate_dominant_colors_async(record) do
def extract_colors_async(record, method) do
meta = %{title: record.title, artists: Enum.map(record.artists, & &1.name)}
params = %{"id" => record.id}
params = %{"id" => record.id, "method" => method}
params
|> Worker.GenerateDominantColors.new(meta: meta)
|> Worker.ExtractColors.new(meta: meta)
|> BackgroundRepo.insert()
end
@@ -329,7 +322,7 @@ defmodule MusicLibrary.Records do
def create_record(attrs \\ %{}) do
with {:ok, record} <- do_create_record(attrs) do
generate_dominant_colors_async(record)
extract_colors_async(record, :fast)
record
|> Record.artist_ids()