subtitler.py

import signal
import sys
import textwrap
import threading
import tkinter as tk
from os import getenv
from queue import Queue

from dotenv import load_dotenv

from modules.audio_translate import translate_audio

load_dotenv()

OFFSET_X = int(getenv('OFFSET_X'))
OFFSET_Y = int(getenv('OFFSET_Y'))
SUBTITLE_FONT_SIZE = int(getenv('SUBTITLE_FONT_SIZE'))
SUBTITLE_COLOR = getenv('SUBTITLE_COLOR')
SUBTITLE_BG_COLOR = getenv('SUBTITLE_BG_COLOR')
SACRIFICIAL_COLOR = getenv('SACRIFICIAL_COLOR')


def subtitle_updater(root, queue, label):
    # Check if there is something new in the queue to display.
    while not queue.empty():
        # destroy old label since new message inbound
        label.destroy()
        if root.wm_state() == 'withdrawn':
            # show root window
            root.deiconify()

        # create subtitle based on message in queue
        msg = queue.get()
        label = tk.Label(
            text=textwrap.fill(msg, 64),
            font=('Comic Sans MS', SUBTITLE_FONT_SIZE, 'bold italic'),
            fg=SUBTITLE_COLOR,
            bg=SUBTITLE_BG_COLOR
        )

        # hide root and destroy label after 3s
        label.after(3000, root.withdraw)
        label.after(3000, label.destroy)

        # place subtitle at bottom middle of screen
        label.pack(side='bottom', anchor='s')
        root.update_idletasks()

    # run every 0.5s
    root.after(50, lambda: subtitle_updater(root, queue, label))


def setup_overlay():
    # set tkinter gui to be topmost without window
    root = tk.Tk()
    root.overrideredirect(True)
    root.geometry(f'{root.winfo_screenwidth()}x{root.winfo_screenheight()}+{OFFSET_X}+{OFFSET_Y}')
    root.lift()
    root.wm_attributes('-topmost', True)
    root.wm_attributes('-disabled', True)

    # Sacrifice random color for transparency
    root.wm_attributes('-transparentcolor', SACRIFICIAL_COLOR)
    root.config(bg=SACRIFICIAL_COLOR)

    # hide initial window
    root.withdraw()

    return root


def close_app(*_):
    print('Closing subtitler.')
    sys.exit(0)


def start_app():
    # catch keyboard interrupt to stop main thread
    signal.signal(signal.SIGINT, close_app)

    overlay = setup_overlay()
    subtitle = tk.Label()
    subtitle_queue = Queue()

    # thread to listen and translate audio
    threading.Thread(target=translate_audio, args=[subtitle_queue], daemon=True).start()

    # updates subtitles every 0.5s by checking queue
    subtitle_updater(overlay, subtitle_queue, subtitle)

    # set full-screen applications to borderless window for subtitles to appear over it
    overlay.mainloop()


if __name__ == '__main__':
    start_app()