FastPrompter/_translate.py at main · vacterro/FastPrompter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""Batch translate untranslated entries in all i18n language files."""
import json
import os
import sys
import time

sys.path.insert(0, os.path.join(os.path.dirname(__file__), "src"))
sys.stdout.reconfigure(encoding='utf-8')

from deep_translator import GoogleTranslator

from fastprompter.core.i18n import ensure_initialized
from fastprompter.core.i18n._engine import _registry as r

ensure_initialized()
en = r.get("EN", {})

with open("_untranslated.json", encoding="utf-8") as f:
    unt_data = json.load(f)

LANG_MAP = {
    "AR": "ar", "DA": "da", "DE": "de", "EST": "et", "FI": "fi",
    "FRA": "fr", "HE": "iw", "IT": "it", "JA": "ja", "KO": "ko",
    "NL": "nl", "NO": "no", "PL": "pl", "PT": "pt", "RU": "ru",
    "SPA": "es", "SV": "sv", "TH": "th", "UKR": "uk", "VI": "vi",
    "ZH": "zh-CN",
}

i18n_dir = os.path.join("src", "fastprompter", "core", "i18n")

for code in sorted(unt_data.keys()):
    target = LANG_MAP.get(code)
    if not target:
        continue
    keys = unt_data[code]
    if not keys:
        continue

    print(f"\n=== {code} ({target}) \u2014 {len(keys)} entries ===")

    filepath = os.path.join(i18n_dir, f"{code.lower()}.py")
    with open(filepath, encoding="utf-8") as f:
        content = f.read()

    changes = 0
    for key in keys:
        if key == "**__{text}__** ({time})" or key == "RGB":
            continue

        # Detect if key uses single or double quotes in file
        sq_key = key.replace("'", "\\'")
        dq_key = key.replace('"', '\\"')

        # Translate
        try:
            translator = GoogleTranslator(source="en", target=target)
            translated = translator.translate(key)
            if not translated or translated == key:
                print(f"  SKIP: {key[:50]}")
                continue
        except Exception as e:
            print(f"  FAIL: {key[:50]} - {e}")
            time.sleep(2)
            continue

        sq_val = translated.replace("'", "\'")
        dq_val = translated.replace('"', '\\"')

        # Try single-quoted pattern: 'key': 'key' -> 'key': 'translated'
        old_sq = f"'{sq_key}': '{sq_key}'"
        new_sq = f"'{sq_key}': '{sq_val}'"

        if old_sq in content:
            content = content.replace(old_sq, new_sq, 1)
            changes += 1
        else:
            # Try double-quoted pattern: "key": "key" -> "key": "translated"
            old_dq = f'"{dq_key}": "{dq_key}"'
            new_dq = f'"{dq_key}": "{dq_val}"'
            if old_dq in content:
                content = content.replace(old_dq, new_dq, 1)
                changes += 1
            else:
                print(f"  NOT FOUND: {key[:50]}")

        time.sleep(0.3)

    if changes > 0:
        with open(filepath, "w", encoding="utf-8") as f:
            f.write(content)
        print(f"  -> {changes} changes to {code.lower()}.py")
    else:
        print("  -> No changes")

print("\n=== DONE ===")