enhancement: Automatically fetch captions when extracting

Author: Betree

apps/cf/lib/llms/statements_creator.ex

@@ -30,18 +30,13 @@ defmodule CF.LLMs.StatementsCreator do
   Create statements from a video that has captions using LLMs
   """
   def process_video!(video_id) do
-    DB.Schema.Video
-    |> join(:inner, [v], vc in DB.Schema.VideoCaption, on: v.id == vc.video_id)
-    |> where([v, vc], v.id == ^video_id)
-    |> order_by([v, vc], desc: vc.inserted_at)
-    |> limit(1)
-    |> select([v, vc], {v, vc})
-    |> DB.Repo.one()
-    |> case do
+    video = DB.Repo.get(DB.Schema.Video, video_id)
+
+    case fetch_or_download_captions(video) do
       nil ->
-        raise "Video or captions not found"
+        raise "Video captions not found"
 
-      {video, video_caption} ->
+      video_caption ->
         video_caption.parsed
         |> chunk_captions()
         |> Enum.map(fn captions ->
@@ -56,6 +51,23 @@ defmodule CF.LLMs.StatementsCreator do
     end
   end
 
+  defp fetch_or_download_captions(video) do
+    case DB.Schema.VideoCaption
+         |> where([vc], vc.video_id == ^video.id)
+         |> order_by(desc: :inserted_at)
+         |> limit(1)
+         |> DB.Repo.one() do
+      nil ->
+        case CF.Videos.download_captions(video) do
+          {:ok, video_caption} -> video_caption
+          _ -> nil
+        end
+
+      video_caption ->
+        video_caption
+    end
+  end
+
   @doc """
   Chunk captions everytime we reach the max caption length
   """

apps/cf/lib/videos/videos.ex

@@ -166,10 +166,7 @@ defmodule CF.Videos do
 
   @doc """
   Download and store captions for a video.
-  Returns captions if success or {:error, reason} if something bad happend.
-
-  Usage:
-  iex> download_captions(video)
+  Returns captions if success or {:error, reason} if something bad happened.
   """
   def download_captions(video = %Video{}) do
     # Try to fetch new captions
@@ -194,9 +191,6 @@ defmodule CF.Videos do
         end
 
         {:error, :not_found}
-
-      result ->
-        result
     end
   end