fix: Transcription server test

lib/medical_transcription/transcription_server.ex

@@ -38,8 +38,9 @@ defmodule Medicode.TranscriptionServer do
   end
 
   @impl GenServer
-  def handle_continue(:start, {:transcription, transcription} = state) do
-    Transcriptions.update_transcription(transcription, %{status: :transcribing})
+  def handle_continue(:start, {:transcription, transcription}) do
+    {:ok, transcription} =
+      Transcriptions.update_transcription(transcription, %{status: :transcribing})
 
     Phoenix.PubSub.broadcast(
       :medicode_pubsub,
@@ -49,7 +50,7 @@ defmodule Medicode.TranscriptionServer do
 
     stream_transcription_and_search(transcription.filename)
 
-    {:noreply, state}
+    {:noreply, {:transcription, transcription}}
   end
 
   @impl GenServer
@@ -78,10 +79,6 @@ defmodule Medicode.TranscriptionServer do
     {:noreply, state}
   end
 
-  def handle_info({:summary, _result}, state) do
-    {:noreply, state}
-  end
-
   def handle_info(:finished, state) do
     {:stop, :normal, state}
   end
@@ -90,7 +87,8 @@ defmodule Medicode.TranscriptionServer do
   def terminate(reason, state) do
     {:transcription, transcription} = state
 
-    Transcriptions.update_transcription(transcription, %{status: :finished})
+    {:ok, transcription} =
+      Transcriptions.update_transcription(transcription, %{status: :finished})
 
     %Transcription{id: id} = transcription
 
@@ -111,39 +109,21 @@ defmodule Medicode.TranscriptionServer do
   #   well
   defp stream_transcription_and_search(audio_file_path) do
     # audio transcription + semantic search
-    summary_text =
-      Medicode.TranscriptionServing
-      |> Nx.Serving.batched_run({:file, audio_file_path})
-      |> Stream.with_index()
-      |> Stream.map(fn {chunk, index} ->
-        send_result(:chunk, chunk, index + 1)
-        chunk.text
-      end)
-      |> Enum.to_list()
-      |> Enum.join()
-
-    summary_chunk = %{
-      text: summary_text,
-      start_timestamp_seconds: nil,
-      end_timestamp_seconds: nil
-    }
-
-    send_result(:summary, summary_chunk, 0)
+    Medicode.TranscriptionServing
+    |> Nx.Serving.batched_run({:file, audio_file_path})
+    |> Enum.each(fn chunk ->
+      result = %{
+        start_mark: format_timestamp(chunk.start_timestamp_seconds),
+        end_mark: format_timestamp(chunk.end_timestamp_seconds),
+        text: chunk.text
+      }
+
+      send(self(), {:chunk, result})
+    end)
 
     send(self(), :finished)
   end
 
-  defp send_result(status, chunk, index) when status in [:chunk, :summary] do
-    result = %{
-      id: index,
-      start_mark: format_timestamp(chunk.start_timestamp_seconds),
-      end_mark: format_timestamp(chunk.end_timestamp_seconds),
-      text: chunk.text
-    }
-
-    send(self(), {status, result})
-  end
-
   defp format_timestamp(seconds) when is_nil(seconds), do: nil
 
   defp format_timestamp(seconds) do

test/medical_transcription/transcription_server_test.exs

@@ -21,7 +21,9 @@ defmodule Medicode.TranscriptionServerTest do
   end
 
   test "transcribe and tag audio", %{transcription: transcription} do
-    spec = {TranscriptionServer, {:transcription, transcription}}
+    spec =
+      {TranscriptionServer,
+       %{transcription: transcription, name: "transcription:#{transcription.id}"}}
 
     {:ok, pid} = start_supervised(spec, restart: :transient)
 
@@ -30,7 +32,8 @@ defmodule Medicode.TranscriptionServerTest do
     # for further explanation.
     ref = Process.monitor(pid)
 
-    assert_receive({:DOWN, ^ref, :process, _object, _pid}, 5_000)
+    # NOTE: This timeout of 15 seconds is arbitrary and may need to be adjusted
+    assert_receive({:DOWN, ^ref, :process, _object, _pid}, 15_000)
 
     transcription = Transcriptions.get_transcription!(transcription.id, true)
     assert 14 == Enum.count(transcription.chunks)