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, log_directory, time_zone_info
from toolbox.os.command import Command

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(
        "--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 shell(cmd: str):
    return Command.popen(cmd)


def main():
    args = get_args()

    # ui
    with gr.Blocks() as blocks:
        gr.Markdown(value="vad.")
        with gr.Tabs():
            with gr.TabItem("shell"):
                shell_text = gr.Textbox(label="cmd")
                shell_button = gr.Button("run")
                shell_output = gr.Textbox(label="output")

                shell_button.click(
                    shell,
                    inputs=[shell_text,],
                    outputs=[shell_output],
                )

    # 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
    )
    return


if __name__ == "__main__":
    main()