main.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
import argparse
import logging
import platform

import gradio as gr

import log
from project_settings import environment, project_path, log_directory, time_zone_info
from tabs.vad_tab import get_vad_tab
from tabs.shell_tab import get_shell_tab

log.setup_size_rotating(log_directory=log_directory, tz_info=time_zone_info)

logger = logging.getLogger("main")


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--examples_dir",
        # default=(project_path / "data").as_posix(),
        default=(project_path / "data/examples").as_posix(),
        type=str
    )
    parser.add_argument(
        "--models_repo_id",
        default="qgyd2021/cc_vad",
        type=str
    )
    parser.add_argument(
        "--trained_model_dir",
        default=(project_path / "trained_models").as_posix(),
        type=str
    )
    parser.add_argument(
        "--hf_token",
        default=environment.get("hf_token"),
        type=str,
    )
    parser.add_argument(
        "--server_port",
        default=environment.get("server_port", 7860),
        type=int
    )

    args = parser.parse_args()
    return args


def main():
    args = get_args()

    # ui
    with gr.Blocks() as blocks:
        gr.Markdown(value="vad.")
        with gr.Tabs():
            _ = get_vad_tab(
                trained_model_dir=args.trained_model_dir,
                examples_dir=args.examples_dir,
                models_repo_id=args.models_repo_id,
                hf_token=args.hf_token,
            )
            _ = get_shell_tab()

    # http://127.0.0.1:7866/
    # http://10.75.27.247:7866/
    blocks.queue().launch(
        # share=True,
        share=False if platform.system() == "Windows" else False,
        server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
        server_port=args.server_port,
        show_error=True
    )
    return


if __name__ == "__main__":
    main()