feat: Store vector for text in feedback, associate to code_vectors

lib/medical_transcription/code_feedback.ex

@@ -6,15 +6,16 @@ defmodule MedicalTranscription.CodeFeedback do
 
   schema "code_feedbacks" do
     field :text, :string
-    field :code, :string
-    field :long_description, :string
+    field :text_vector, Pgvector.Ecto.Vector
     field :response, :boolean
+
+    belongs_to :code_vector, MedicalTranscription.CodeVector
   end
 
   def changeset(code_feedback, params \\ %{}) do
     code_feedback
-    |> Ecto.Changeset.cast(params, [:text, :code, :long_description, :response])
-    |> Ecto.Changeset.validate_required([:text, :code, :long_description, :response])
+    |> Ecto.Changeset.cast(params, [:text, :text_vector, :response, :code_vector_id])
+    |> Ecto.Changeset.validate_required([:text, :text_vector, :response, :code_vector_id])
   end
 
   def track_response(params) do

lib/medical_transcription/code_searcher.ex

@@ -5,13 +5,9 @@ defmodule MedicalTranscription.CodeSearcher do
   """
 
   alias MedicalTranscription.CodeVector
-  alias AudioTagger.Structs.TagResult
 
   def process_chunk(model_tuple, text) do
     search_for_similar_codes(model_tuple, text)
-    |> Enum.map(fn %CodeVector{code: code, description: description} ->
-      %TagResult{code: code, label: description, score: 1.0}
-    end)
 
     # -- Remove matches that don't exceed a given threshold
     # TODO: This depends on receiving a similarity score from the Postgres query

lib/medical_transcription_web/components/components.ex

@@ -123,8 +123,8 @@ defmodule MedicalTranscriptionWeb.Components do
     ~H"""
     <div class="flex items-center gap-4 px-[14px] py-3 text-sm">
       <div class="flex gap-3">
-        <.feedback_button code={@code} text={@text} label={@label} response="true" />
-        <.feedback_button code={@code} text={@text} label={@label} response="false" class="" />
+        <.feedback_button code_vector_id={@code_vector_id} text={@text} response="true" />
+        <.feedback_button code_vector_id={@code_vector_id} text={@text} response="false" />
       </div>
 
       <div
@@ -142,9 +142,8 @@ defmodule MedicalTranscriptionWeb.Components do
     ~H"""
     <button
       phx-click="add_feedback"
-      phx-value-code={@code}
+      phx-value-code_vector_id={@code_vector_id}
       phx-value-text={@text}
-      phx-value-long-description={@label}
       phx-value-response={@response}
       type="button"
       class="p-2 border border-button-deactivated-background rounded-lg hover:bg-slate-200"

lib/medical_transcription_web/components/transcription_text_component.ex

@@ -6,7 +6,7 @@ defmodule MedicalTranscriptionWeb.Components.TranscriptionTextComponent do
   use MedicalTranscriptionWeb, :live_component
   import MedicalTranscriptionWeb.Components
   import MedicalTranscriptionWeb.Components.KeywordHighlighter
-  alias AudioTagger.Structs.TagResult
+  alias MedicalTranscription.CodeVector
 
   @impl Phoenix.LiveComponent
   def update(assigns, socket) do
@@ -53,8 +53,8 @@ defmodule MedicalTranscriptionWeb.Components.TranscriptionTextComponent do
       </div>
 
       <div class="flex-1 flex flex-col items-stretch gap-3">
-        <%= for %TagResult{code: code, label: label, score: score} <- @row.tags do %>
-          <.tag_result code={code} label={label} score={score} text={@row.text} />
+        <%= for %CodeVector{id: id, code: code, description: label} <- @row.tags do %>
+          <.tag_result code_vector_id={id} code={code} label={label} score={1.0} text={@row.text} />
         <% end %>
       </div>
     </div>

lib/medical_transcription_web/live/home_live/index.ex

@@ -81,12 +81,13 @@ defmodule MedicalTranscriptionWeb.HomeLive.Index do
 
   @impl true
   def handle_event("add_feedback", params, socket) do
-    fields =
-      params
-      |> Map.put("long_description", params["long-description"])
-      |> Map.drop(["long-description"])
+    text_vector =
+      AudioTagger.Vectors.embed([params["text"]]) |> Map.get(:pooled_state) |> Nx.to_flat_list()
 
-    result = MedicalTranscription.CodeFeedback.track_response(fields)
+    result =
+      params
+      |> Map.put("text_vector", text_vector)
+      |> MedicalTranscription.CodeFeedback.track_response()
 
     socket =
       case result do

priv/repo/migrations/20240125172837_migrate_to_vector_for_code_feedbacks.exs

@@ -0,0 +1,14 @@
+defmodule MedicalTranscription.Repo.Migrations.MigrateToVectorForCodeFeedbacks do
+  use Ecto.Migration
+
+  def change do
+    alter table("code_feedbacks") do
+      add :text_vector, :vector, size: 384
+      add :code_vector_id, references("code_vectors")
+
+      # These two columns can be found on the `code_vectors` table referenced above
+      remove :code, :string
+      remove :long_description, :string
+    end
+  end
+end