ai-generated ui translations

2025-10-14 18:50:01 +02:00
parent a228c54dd5
commit 1dd4c4a109
6 changed files with 254 additions and 74 deletions
@@ -0,0 +1,65 @@
+#!/usr/bin/env bash
+
+if [[ -z "$1" ]]; then
+    echo "Usage: $0 <target_locale> [model]"
+    exit 1
+fi
+
+# Variables
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+XDG_CONFIG_HOME="${XDG_CONFIG_HOME:-$HOME/.config}"
+SHELL_CONFIG_DIR="$XDG_CONFIG_HOME/illogical-impulse"
+SHELL_CONFIG_FILE="${SHELL_CONFIG_DIR}/config.json"
+TRANSLATIONS_DIR="${SCRIPT_DIR}/../../translations"
+TRANSLATIONS_TARGET_DIR="${SHELL_CONFIG_DIR}/translations"
+SOURCE_LOCALE="en_US"
+NOTIFICATION_APP_NAME="Shell"
+TARGET_LOCALE="$1"
+MODEL="${2:-${GEMINI_MODEL:-gemini-2.5-flash}}"
+
+# Update the source keys for translation
+"${TRANSLATIONS_DIR}/tools/manage-translations.sh" update -l "$SOURCE_LOCALE" --yes
+mkdir -p "$TRANSLATIONS_TARGET_DIR"
+
+# Construct the prompt string
+instruction='You are to translate the user interface of a **desktop shell**. Given a JSON object of key-value pairs, return a JSON with the same structure, with keys unchanged and values translated to '"$TARGET_LOCALE"'. Be as **concise** as possible to save screen space, and make sure terminology is relevant (e.g. "discharging" refers to the battery status).'
+content=$(cat "${TRANSLATIONS_DIR}/en_US.json")
+prompt_json=$(jq -n --arg prompt_text "$instruction" --arg content "$content" '$prompt_text + "\n```\n" + $content + "\n```\n"')
+
+# Prepare request data using jq
+payload=$(jq -n \
+    --arg prompt "$prompt_json" \
+    --arg temperature "0" \
+    --arg model "$MODEL" \
+    '{
+        contents: [{
+            parts: [
+                {text: $prompt}
+            ]
+        }],
+        generationConfig: {
+            temperature: ($temperature | tonumber),
+            "responseMimeType": "application/json",
+        }
+    }'
+)
+# echo "$payload" | jq
+
+# Get API key
+API_KEY=$(secret-tool lookup 'application' 'illogical-impulse' | jq -r '.apiKeys.gemini')
+
+# Notify start
+notify-send "Translation started" "Will take 2 minutes, and you'll be notified when it's done, so feel free to do something else in the meantime." -a "$NOTIFICATION_APP_NAME"
+
+# Make the request
+response=$(curl "https://generativelanguage.googleapis.com/v1beta/models/${MODEL}:generateContent" \
+-H "x-goog-api-key: $API_KEY" \
+-H 'Content-Type: application/json' \
+-X POST \
+-d "$payload" 2> /dev/null)
+# echo "$response" | jq
+
+# Write the result
+echo "$response" | jq -r '.candidates[0].content.parts[0].text' > "${TRANSLATIONS_TARGET_DIR}/${TARGET_LOCALE}.json"
+jq --arg locale "$TARGET_LOCALE" '.language.ui = $locale' "$SHELL_CONFIG_FILE" > "${SHELL_CONFIG_FILE}.tmp" && mv "${SHELL_CONFIG_FILE}.tmp" "$SHELL_CONFIG_FILE"
+notify-send "Translation complete" "Enjoy! In case you wanna refine it, the file is in ${TRANSLATIONS_TARGET_DIR}/${TARGET_LOCALE}.json" -a "$NOTIFICATION_APP_NAME"