From 994836cc05d30a6b9506424f3a3aada356d5b4a1 Mon Sep 17 00:00:00 2001 From: Tangent Date: Sun, 23 Aug 2026 02:40:19 -0600 Subject: [PATCH] fix loops, improve time estimator --- make_cover_briefs.lua | 63 +++++++++++++++++++++++++------------------ 1 file changed, 37 insertions(+), 26 deletions(-) diff --git a/make_cover_briefs.lua b/make_cover_briefs.lua index 40803f5..6369d5a 100755 --- a/make_cover_briefs.lua +++ b/make_cover_briefs.lua @@ -156,17 +156,20 @@ end local convert_ebooks = function(books) for name, data in pairs(books) do - if data.extracted_file then return end - if not data.ebook_file then return end + local function loop() + if data.extracted_file then return end + if not data.ebook_file then return end - local extracted_file = "extracted_texts/" .. name .. ".txt" - os.execute(config.llm_covers.calibre_location .. "/Contents/MacOS/ebook-convert " - .. data.ebook_file:enquote() .. " " .. extracted_file:enquote()) + local extracted_file = "extracted_texts/" .. name .. ".txt" + os.execute(config.llm_covers.calibre_location .. "/Contents/MacOS/ebook-convert " + .. data.ebook_file:enquote() .. " " .. extracted_file:enquote()) - if utility.path_exists(extracted_file) then - data.extracted_file = extracted_file - data.extracted_file_size = utility.file_size(extracted_file) + if utility.path_exists(extracted_file) then + data.extracted_file = extracted_file + data.extracted_file_size = utility.file_size(extracted_file) + end end + loop() end end @@ -248,32 +251,38 @@ end local generate_cover_briefs = function(books) for name, data in pairs(books) do - if data.cover_brief_file then return end - print(name) + local function loop() + if data.cover_brief_file then return end + print(name) - local text - if data.extracted_file_size > config.llm_covers.maximum_bytes then - text = generate_summary(name, data) - else - text = read_all(extracted_file) + local text + if data.extracted_file_size > config.llm_covers.maximum_bytes then + text = generate_summary(name, data) + else + text = read_all(data.extracted_file) + end + + print("Generating cover brief.") + text = send_prompt(prompts.cover_brief .. text) + + local cover_brief_file = "cover_briefs/" .. name .. ".txt" + write_all(cover_brief_file, text) + write_all("for_gemini/" .. name .. ".txt", "Title: " .. name .. prompts.for_gemini .. text) + data.cover_brief_file = cover_brief_file end - - print("Generating cover brief.") - text = send_prompt(prompts.cover_brief .. text) - - local cover_brief_file = "cover_briefs/" .. name .. ".txt" - write_all(cover_brief_file, text) - write_all("for_gemini/" .. name .. ".txt", "Title: " .. name .. prompts.for_gemini .. text) - data.cover_brief_file = cover_brief_file + loop() end end local estimate_time_required = function(books) local sum, count = 0, 0 for name, data in pairs(books) do - if data.cover_brief_file then return end - sum = sum + 60 * (2 + data.extracted_file_size/(config.llm_covers.maximum_bytes/7.5)) - count = count + 1 + local function loop() + if data.cover_brief_file then return end + sum = sum + 60 * (1.5 + data.extracted_file_size/(config.llm_covers.maximum_bytes/7.5)) + count = count + 1 + end + loop() end print(count .. " cover briefs to generate. Estimated time: " @@ -293,5 +302,7 @@ generate_cover_briefs(books) print("") timing.print_report() +-- utility.print_table(books) + -- ebook_file, extracted_file, extracted_file_size, -- intermediate_file, condensed_intermediate_file, cover_brief_file