Merge pull request #1 from headwayio/feature/summarize-all

lib/medical_transcription/transcriber.ex

@@ -6,6 +6,13 @@ defmodule MedicalTranscription.Transcriber do
 
   alias MedicalTranscription.CodeSearcher
 
+  defp get_tags_and_send_result(search_configuration, chunk, index, live_view_pid, is_summary) do
+    tags = CodeSearcher.process_chunk(search_configuration, chunk.text)
+    result = build_result(index, chunk, tags, is_summary)
+
+    send(live_view_pid, {:transcription_row, result})
+  end
+
   # Ideas for future exploration:
   # - Instead of storing the long description vectors in a binary file on disk, we could store them within a vector DB
   #   (such as pgvector or Pinecone.io)
@@ -15,13 +22,23 @@ defmodule MedicalTranscription.Transcriber do
   def stream_transcription_and_search(live_view_pid, audio_file_path) do
     search_configuration = CodeSearcher.prepare_search_configuration()
 
-    # Audio transcription + semantic search
-    for {chunk, index} <- stream_transcription(audio_file_path) do
-      tags = CodeSearcher.process_chunk(search_configuration, chunk.text)
-      result = build_result(index, chunk, tags)
+    # audio transcription + semantic search
+    all_chunk_text =
+      audio_file_path
+      |> stream_transcription()
+      |> Enum.reduce("", fn {chunk, index}, acc ->
+        get_tags_and_send_result(search_configuration, chunk, index, live_view_pid, false)
+
+        acc <> chunk.text
+      end)
 
-      send(live_view_pid, {:transcription_row, result})
-    end
+    get_tags_and_send_result(
+      search_configuration,
+      %{text: all_chunk_text, start_timestamp_seconds: 0, end_timestamp_seconds: 0},
+      0,
+      live_view_pid,
+      true
+    )
   end
 
   defp stream_transcription(audio_file_path) do
@@ -30,13 +47,14 @@ defmodule MedicalTranscription.Transcriber do
     |> Stream.with_index()
   end
 
-  defp build_result(index, chunk, tags) do
+  defp build_result(index, chunk, tags, is_summary) do
     %{
       id: index,
       start_mark: format_timestamp(chunk.start_timestamp_seconds),
       end_mark: format_timestamp(chunk.end_timestamp_seconds),
       text: chunk.text,
-      tags: tags
+      tags: tags,
+      is_summary: is_summary
     }
   end
 

lib/medical_transcription_web/components/transcription_text_component.ex

@@ -26,7 +26,9 @@ defmodule MedicalTranscriptionWeb.Components.TranscriptionTextComponent do
     <div class="flex gap-12 pb-10 border-b border-[#444444]/20">
       <div class="flex-1 flex flex-col gap-4">
         <p class="text-[32px] leading-normal font-semibold">
-          <%= @row.start_mark %> - <%= @row.end_mark %>
+          <%= if !@row.is_summary do %>
+            <%= @row.start_mark %> - <%= @row.end_mark %>
+          <% end %>
         </p>
         <p class="text-[28px] leading-normal text-type-black-tertiary">
           <.async_result :let={keywords} assign={@keywords}>