From bfbf3e9a3fe7a26f95c7f1209586a70167a2fb1e Mon Sep 17 00:00:00 2001 From: Marty Oehme Date: Thu, 5 Jun 2025 18:29:10 +0200 Subject: [PATCH] Add function to improve list of originals --- prophet/app.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/prophet/app.py b/prophet/app.py index 9f3fd62..df41e6b 100644 --- a/prophet/app.py +++ b/prophet/app.py @@ -77,6 +77,16 @@ def load_existing_improvements() -> list[Improvement]: print(f"Error loading file {fname}: {e}") return improvements + +def improve_originals(originals: list[Original]) -> list[Improvement]: + improvements: list[Improvement] = [] + for orig in originals: + better_title = improve_with_groq(orig.title) + + improvements.append(Improvement(original=orig, title=better_title, summary="")) + return improvements + + def improve_with_groq(original_content: str) -> str: client = Groq(api_key=os.getenv("GROQ_API_KEY", "NO_API_KEY_FOUND"))