run.py

# ./run.py

import os
import time
import argparse

import pyautogui
from dotenv import load_dotenv

from translate import translate


if __name__ == "__main__":

    load_dotenv()

    # Time (sec) to pause after each PyAuto Function Call
    pyautogui.PAUSE = float(os.getenv('PYAUTO_PAUSE', 0.1))

    parser = argparse.ArgumentParser(
        prog="Pose2Input-MK9",
        description="Translate the Video Input to Keystrokes"
    )
    parser.add_argument(
        '-d', '--debug_level', type=int, metavar='', default=0,
        help='<0, 1, 2, 3> set different levels of information for logs or live feed'
    )
    parser.add_argument(
        '-l', '--log_flag', action='store_true',
        help='stores the video_log in "logs" folder'
    )
    parser.add_argument(
        '-L', '--live_flag', action='store_true',
        help='displays the captured video'
    )
    args = parser.parse_args()

    # A Delay before Starting the Program
    time.sleep(int(os.getenv('DELAY_TIME', 0)))

    live_flag = args.live_flag
    log_flag = args.log_flag
    debug_level = args.debug_level

    # If Debug > 0 and no flag is selected
    # Then log_flag is automatically set to True
    if (debug_level > 0) and not (log_flag or live_flag):
        log_flag = True

    if not os.path.exists('logs'):
        os.mkdir('logs')

    translate(
        log_flag=log_flag,
        live_flag=live_flag,
        debug_level=debug_level,
        log_fps=int(os.getenv('LOG_FPS', 20)),
        camera_port=os.getenv('CAMERA_PORT', '0'),
        motion_threshold_factor=int(
            os.getenv('MOTION_THRESHOLD_FACTOR', 64)
        ),
    )