show_choices.py

# ripped from homophones.py
from talon import app, clip, cron
from talon.voice import Context, Str, press
from talon.webview import Webview

from .utils import parse_word, string_capture, format_phrase_with_dictations
import os
pick_context = Context("pick_choice")

webview = Webview()
css_template = """
<style type="text/css">
body {
    padding: 0;
    margin: 0;
    font-size: 150%;
    min-width: 600px;
}

td {
    text-align: left;
    margin: 0;
    padding: 5px 10px;
}

h3 {
    padding: 5px 0px;
}

table {
    counter-reset: rowNumber;
}

table .count {
    counter-increment: rowNumber;
}

.count td:first-child::after {
    content: counter(rowNumber);
    min-with: 1em;
    margin-right: 0.5em;
}

.pick {
    font-weight: normal;
    font-style: italic;
}
td.value pre {
    max-width: 500px;
    white-space: nowrap;
    overflow: hidden;
    text-overflow: ellipsis;
}
.cancel {
    text-align: center;
}

</style>
"""

choices_template = (
    css_template
    + """
<div class="contents">
{% if title %}<h3>{{title}}</h3>{% endif %}
<table>
{% for word in choices %}
<tr class="{% if choice_labels == None %}count{% endif %}">
    <td class="pick">🔊 {% if choice_labels == None %} pick {% else %} {{choice_labels[loop.index-1]}} {% endif %} </td>
    <td class="value">{{ word }}</td>
</tr>
{% endfor %}
<tr><td colspan="2" class="pick cancel">🔊 cancel</td></tr>
</table>
</div>
"""
)

def close_choices():
    webview.hide()
    pick_context.unload()


def make_selection(m, choices, callback):
    cron.after("0s", close_choices)
    words = m._words
    d = None
    if len(words) == 1:
        d = int(parse_word(words[0]))
    else:
        d = int(parse_word(words[1]))
    
    choice = choices[d - 1]
    callback(choice)

def make_custom_selection(m, choice_labels, choices, callback):
    cron.after("0s", close_choices)
    choice = format_phrase_with_dictations(m)
    index = choice_labels.index(choice)
    callback(choice_labels[index])

def show_choices(choices, title, callback, choice_labels=None):
    global pick_context
    if len(choices) < 1:
        print('no choices to show')
        return
    webview.render(choices_template, choices=choices, title=title, choice_labels=choice_labels)
    webview.show()

    keymap = {"(cancel | 0)": lambda x: close_choices()}
    # assign number for choices
    if choice_labels == None:
        valid_indices = range(len(choices))
        keymap.update(
            {
                "[pick] %s" % (i + 1): lambda m: make_selection(m, choices, callback)
                for i in valid_indices
            }
        )
    # pre-assigned words for choices
    else:
        keymap.update(
            {
                (label): lambda m: make_custom_selection(m, choice_labels, choices, callback)
                for label in choice_labels
            }
        )

    pick_context.keymap(keymap)
    pick_context.load()