main/app/tensorboard.py

import os
import sys
import json
import logging
import webbrowser

from tensorboard import program

sys.path.append(os.getcwd())

from main.configs.config import Config
translations = Config().translations

with open(os.path.join("main", "configs", "config.json"), "r") as f:
    configs = json.load(f)

def launch_tensorboard():
    for l in ["root", "tensorboard"]:
        logging.getLogger(l).setLevel(logging.ERROR)

    tb = program.TensorBoard()
    tb.configure(argv=[None, "--logdir", "assets/logs", f"--port={configs['tensorboard_port']}"])
    url = tb.launch()

    print(f"{translations['tensorboard_url']}: {url}")
    if "--open" in sys.argv: webbrowser.open(url)

    return f"{translations['tensorboard_url']}: {url}"

if __name__ == "__main__": launch_tensorboard()