From d3d5fa52569d1e425d27935b7d8184c8ba195d10 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebasti=C3=A1n=20Ram=C3=ADrez?= <tiangolo@gmail.com>
Date: Sat, 26 Jul 2025 23:16:26 +0200
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A8=20Refactor=20translate=20script=20?=
 =?UTF-8?q?with=20extra=20feedback=20(prints)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 requirements-translations.txt |  1 +
 scripts/translate.py          | 20 +++++++++++++-------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/requirements-translations.txt b/requirements-translations.txt
index 7a2a8004e..90f718032 100644
--- a/requirements-translations.txt
+++ b/requirements-translations.txt
@@ -1 +1,2 @@
 pydantic-ai==0.0.30
+GitPython==3.1.45
diff --git a/scripts/translate.py b/scripts/translate.py
index 09533747b..6af12dc98 100644
--- a/scripts/translate.py
+++ b/scripts/translate.py
@@ -106,7 +106,9 @@ def translate_page(*, lang: str, path: Path) -> None:
     original_content = path.read_text()
     old_translation: str | None = None
     if out_path.exists():
+        print(f"Found existing translation: {out_path}")
         old_translation = out_path.read_text()
+    print(f"Translating {path} to {lang} ({language})")
     agent = Agent("openai:gpt-4o")
 
     prompt_segments = [
@@ -131,13 +133,14 @@ def translate_page(*, lang: str, path: Path) -> None:
         ]
     )
     prompt = "\n\n".join(prompt_segments)
-
+    print(f"Running agent for {out_path}")
     result = agent.run_sync(prompt)
     out_content = f"{result.data.strip()}\n"
+    print(f"Saving translation to {out_path}")
     out_path.write_text(out_content)
 
 
-def iter_paths_to_translate() -> Iterable[Path]:
+def iter_all_en_paths() -> Iterable[Path]:
     """
     Iterate on the markdown files to translate in order of priority.
     """
@@ -161,13 +164,16 @@ def iter_paths_to_translate() -> Iterable[Path]:
         yield path
 
 
-@app.command()
-def translate_all(lang: str) -> None:
-    paths_to_process: list[Path] = []
-    for path in iter_paths_to_translate():
+def iter_en_paths_to_translate() -> Iterable[Path]:
+    for path in iter_all_en_paths():
         if str(path).replace("docs/en/docs/", "").startswith(non_translated_sections):
             continue
-        paths_to_process.append(path)
+        yield path
+
+
+@app.command()
+def translate_all(lang: str) -> None:
+    paths_to_process = list(iter_en_paths_to_translate())
     print("Original paths:")
     for p in paths_to_process:
         print(f"  - {p}")