51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
|
"""Programmatische Icon-Generierung via Pillow."""
|
||
|
|
|
||
|
|
from PIL import Image, ImageDraw
|
||
|
|
|
||
|
|
from whisper_local.tray._tray import AppState
|
||
|
|
|
||
|
|
_STATE_COLORS: dict[AppState, tuple[int, int, int]] = {
|
||
|
|
AppState.WAITING: (150, 150, 150),
|
||
|
|
AppState.RECORDING: (220, 50, 50),
|
||
|
|
AppState.TRANSCRIBING: (220, 180, 0),
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
def create_icon(state: AppState, size: int = 64) -> Image.Image:
|
||
|
|
"""Erzeugt ein Mikrofon-Icon in der Farbe des übergebenen Zustands."""
|
||
|
|
color = _STATE_COLORS[state]
|
||
|
|
img = Image.new("RGBA", (size, size), (0, 0, 0, 0))
|
||
|
|
draw = ImageDraw.Draw(img)
|
||
|
|
|
||
|
|
cx = size // 2
|
||
|
|
bw = max(4, size // 6) # Hälfte der Körperbreite
|
||
|
|
top = size // 8
|
||
|
|
mid = size // 2
|
||
|
|
|
||
|
|
# Mikrofon-Körper (abgerundetes Rechteck)
|
||
|
|
draw.rounded_rectangle(
|
||
|
|
[cx - bw, top, cx + bw, mid + bw],
|
||
|
|
radius=bw,
|
||
|
|
fill=color,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Bogen (Stativ-Bogen)
|
||
|
|
arc_r = size // 4
|
||
|
|
arc_top = mid
|
||
|
|
lw = max(2, size // 20)
|
||
|
|
draw.arc(
|
||
|
|
[cx - arc_r, arc_top, cx + arc_r, arc_top + arc_r],
|
||
|
|
start=0, end=180, fill=color, width=lw,
|
||
|
|
)
|
||
|
|
|
||
|
|
# Stiel
|
||
|
|
pole_top = arc_top + arc_r // 2
|
||
|
|
pole_bot = size - size // 8
|
||
|
|
draw.line([cx, pole_top, cx, pole_bot], fill=color, width=lw)
|
||
|
|
|
||
|
|
# Sockel
|
||
|
|
base = size // 5
|
||
|
|
draw.line([cx - base, pole_bot, cx + base, pole_bot], fill=color, width=lw)
|
||
|
|
|
||
|
|
return img
|