main.py

#!/usr/bin/python

from pypinyin import pinyin, Style
import subprocess
import sys

def show_error(message):
    subprocess.run([
        "notify-send",
        "-u", "critical",
        "Hanzi to Pinyin Error",
        message
    ])
    sys.exit(1)

try:
    # Get highlighted text
    highlighted_text = subprocess.check_output(
        ["xsel", "-p"], text=True, stderr=subprocess.DEVNULL
    ).strip()

    if not highlighted_text:
        sys.exit(0)  # Silent exit for no selection

    # Convert to Pinyin with tones
    pinyin_text = ' '.join([item[0] for item in pinyin(highlighted_text, style=Style.TONE)])

    # Copy to clipboard
    subprocess.run(
        ["xsel", "-ib"],
        input=pinyin_text,
        text=True,
        stderr=subprocess.DEVNULL
    )

except Exception as e:
    show_error(f"{str(e)}\n(Selected text: '{highlighted_text[:50]}')")