Can fetch genres from OpenAI

lib/music_library/records.ex

@@ -185,6 +185,27 @@ defmodule MusicLibrary.Records do
     end
   end
 
+  def populate_genres(record) do
+    artists =
+      record.artists
+      |> Enum.map(fn a -> a.name end)
+      |> Enum.join(",")
+
+    prompt = """
+    Provide a list of music genres applicable to the album "#{record.title}" by #{artists}.
+
+    Limit the list to 5 genres, ordered by decreasing specificity, all lowercase.
+
+    Return a valid JSON list.
+    """
+
+    {:ok, response} = OpenAI.gpt(prompt)
+
+    record
+    |> Record.add_genres(response["genres"])
+    |> Repo.update()
+  end
+
   defp get_cover_art_or_default(musicbrainz_id) do
     case music_brainz_config().api.get_cover_art(
            {:musicbrainz_id, musicbrainz_id},

lib/music_library/records/record.ex

@@ -84,6 +84,10 @@ defmodule MusicLibrary.Records.Record do
     |> put_embed(:artists, artists_attrs)
   end
 
+  def add_genres(record, genres) do
+    change(record, genres: genres)
+  end
+
   def add_cover_data(record, cover_data) do
     record
     |> change(cover_data: cover_data)

lib/music_library_web/live/collection_live/show.ex

@@ -63,6 +63,26 @@ defmodule MusicLibraryWeb.CollectionLive.Show do
     end
   end
 
+  def handle_event("populate_genres", %{"id" => id}, socket) do
+    record = Records.get_record!(id)
+
+    case Records.populate_genres(record) do
+      {:ok, updated_record} ->
+        {:noreply,
+         socket
+         |> put_flash(:info, gettext("Genres populated successfully"))
+         |> assign(:record, updated_record)}
+
+      {:error, reason} ->
+        {:noreply,
+         socket
+         |> put_flash(
+           :error,
+           gettext("Error populating genres") <> "," <> inspect(reason)
+         )}
+    end
+  end
+
   def handle_event("refresh_cover", %{"id" => id}, socket) do
     record = Records.get_record!(id)
 

lib/music_library_web/live/collection_live/show.html.heex

@@ -56,6 +56,18 @@
           {gettext("MB Data")}
         </.button>
       </.link>
+      <.link phx-click={JS.push("populate_genres", value: %{id: @record.id})}>
+        <.button type="button" class="relative -ml-px inline-flex items-center rounded-none">
+          <span class="sr-only">{gettext("Populate")}</span>
+          <.icon
+            name="hero-sparkles"
+            class="h-4 w-4 mr-1 phx-click-loading:animate-shake"
+            aria-hidden="true"
+            data-slot="icon"
+          />
+          {gettext("Genres")}
+        </.button>
+      </.link>
       <.link
         phx-click={JS.push("delete", value: %{id: @record.id})}
         data-confirm={gettext("Are you sure?")}

lib/music_library_web/live/wishlist_live/show.ex

@@ -83,6 +83,26 @@ defmodule MusicLibraryWeb.WishlistLive.Show do
     end
   end
 
+  def handle_event("populate_genres", %{"id" => id}, socket) do
+    record = Records.get_record!(id)
+
+    case Records.populate_genres(record) do
+      {:ok, updated_record} ->
+        {:noreply,
+         socket
+         |> put_flash(:info, gettext("Genres populated successfully"))
+         |> assign(:record, updated_record)}
+
+      {:error, reason} ->
+        {:noreply,
+         socket
+         |> put_flash(
+           :error,
+           gettext("Error populating genres") <> "," <> inspect(reason)
+         )}
+    end
+  end
+
   @impl true
   def handle_info({MusicLibraryWeb.RecordLive.FormComponent, {:saved, record}}, socket) do
     {:noreply, assign(socket, :record, record)}

lib/music_library_web/live/wishlist_live/show.html.heex

@@ -56,6 +56,18 @@
           {gettext("MB Data")}
         </.button>
       </.link>
+      <.link phx-click={JS.push("populate_genres", value: %{id: @record.id})}>
+        <.button type="button" class="relative -ml-px inline-flex items-center rounded-none">
+          <span class="sr-only">{gettext("Populate")}</span>
+          <.icon
+            name="hero-sparkles"
+            class="h-4 w-4 mr-1 phx-click-loading:animate-shake"
+            aria-hidden="true"
+            data-slot="icon"
+          />
+          {gettext("Genres")}
+        </.button>
+      </.link>
       <.link
         phx-click={JS.push("delete", value: %{id: @record.id})}
         data-confirm={gettext("Are you sure?")}

lib/open_ai.ex

@@ -0,0 +1,64 @@
+defmodule OpenAI do
+  def gpt(prompt) do
+    {:ok, collector} = Agent.start_link(fn -> "" end)
+
+    gpt_stream(prompt, fn data ->
+      case get_in(data, ["choices", Access.at(0), "delta", "content"]) do
+        nil -> :ok
+        data -> Agent.update(collector, fn current -> current <> data end)
+      end
+    end)
+
+    result = Agent.get(collector, & &1) |> Jason.decode!()
+    Agent.stop(collector)
+    {:ok, result}
+  end
+
+  defp gpt_stream(prompt, cb) do
+    fun = fn request, finch_request, finch_name, finch_options ->
+      fun = fn
+        {:status, status}, response ->
+          %{response | status: status}
+
+        {:headers, headers}, response ->
+          %{response | headers: headers}
+
+        {:data, data}, response ->
+          body =
+            data
+            |> String.split("data: ")
+            |> Enum.map(fn str ->
+              str
+              |> String.trim()
+              |> decode_body(cb)
+            end)
+            |> Enum.filter(fn d -> d != :ok end)
+
+          old_body = if response.body == "", do: [], else: response.body
+
+          %{response | body: old_body ++ body}
+      end
+
+      case Finch.stream(finch_request, finch_name, Req.Response.new(), fun, finch_options) do
+        {:ok, response} -> {request, response}
+        {:error, exception} -> {request, exception}
+      end
+    end
+
+    Req.post!("https://api.openai.com/v1/chat/completions",
+      json: %{
+        model: "gpt-4o-mini",
+        messages: [%{role: "user", content: prompt}],
+        response_format: %{type: "json_object"},
+        stream: true,
+        temperature: 0.2
+      },
+      auth: {:bearer, System.fetch_env!("OPENAI_KEY")},
+      finch_request: fun
+    )
+  end
+
+  defp decode_body("", _), do: :ok
+  defp decode_body("[DONE]", _), do: :ok
+  defp decode_body(json, cb), do: cb.(Jason.decode!(json))
+end