Rearrange for tidier structure (#2212)

2025-10-16 07:19:55 +08:00
parent 13065d7e5a
commit 8b493e091d
529 changed files with 165 additions and 138 deletions
@@ -0,0 +1,64 @@
+#!/usr/bin/env bash
+
+if [[ -z "$1" ]]; then
+    echo "Usage: $0 <image_path> [model] [prompt]"
+    echo "Tip: set GEMINI_WALLPAPER_MODEL and/or GEMINI_WALLPAPER_PROMPT to provide defaults."
+    exit 1
+fi
+
+# Variables
+SOURCE_IMG_PATH="$1"
+MODEL="${2:-${GEMINI_WALLPAPER_MODEL:-gemini-2.5-flash-lite}}" # We use the flash variant so it's fast
+WALLPAPER_NAME="$(basename "$SOURCE_IMG_PATH")"
+PROMPT="${3:-${GEMINI_WALLPAPER_PROMPT:-Categorize the wallpaper. Its file name is $WALLPAPER_NAME}}"
+RESIZED_IMG_PATH="/tmp/quickshell/ai/wallpaper.jpg"
+
+# Resize image for speed
+magick "$SOURCE_IMG_PATH" -resize 200x -quality 50 "$RESIZED_IMG_PATH"
+
+# Get API key
+API_KEY=$(secret-tool lookup 'application' 'illogical-impulse' | jq -r '.apiKeys.gemini')
+
+# Encode image to base64
+if [[ "$(base64 --version 2>&1)" = *"FreeBSD"* ]]; then
+    B64FLAGS="--input"
+else
+    B64FLAGS="-w0"
+fi
+B64DATA="$(base64 $B64FLAGS $RESIZED_IMG_PATH)"
+# echo $B64DATA
+
+# Prepare request data
+payload='{
+    "contents": [{
+        "parts":[
+            {
+                "inline_data": {
+                "mime_type":"image/jpeg",
+                "data": "'"$B64DATA"'"
+                }
+            },
+            {"text": "'"$PROMPT"'"}
+        ]
+    }],
+    "generationConfig": {
+        "responseMimeType": "text/x.enum",
+        "responseSchema": {
+            "type": "string",
+            "enum": [ "abstract", "anime", "city", "minimalist", "landscape", "plants", "person", "space" ]
+        },
+        "temperature": 0
+    }
+}'
+# echo "$payload" | jq
+
+# 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'
@@ -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"
@@ -0,0 +1,16 @@
+#!/usr/bin/env bash
+
+# Get the list, skip the header, and extract the first column (model names)
+model_names=$(ollama list | tail -n +2 | awk '{print $1}')
+
+# Build a JSON array
+json_array="["
+for name in $model_names; do
+    json_array+="\"$name\","
+done
+
+# Remove trailing comma and close the array
+json_array="${json_array%,}]"
+
+# Output the JSON array
+echo "$json_array"