main3.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
docker build -t llm_eval_system:v20250724_1442 .

docker stop llm_eval_system_7862 && docker rm llm_eval_system_7862

docker run -itd \
--name llm_eval_system_7862 \
--restart=always \
--network host \
-e port=7862 \
llm_eval_system:v20250724_1442 \
/bin/bash
"""
import argparse
import logging
import platform

import gradio as gr

from project_settings import environment, project_path, log_directory
from toolbox.os.command import Command
import log

log.setup_size_rotating(log_directory=log_directory)

logger = logging.getLogger("main")


def get_args():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--eval_data_dir",
        default=(project_path / "data/eval_data").as_posix(),
        type=str,
    )
    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:
        with gr.Tabs():
            with gr.TabItem("shell"):
                shell_text = gr.Textbox(label="cmd")
                shell_button = gr.Button("run")
                shell_output = gr.Textbox(label="output", max_lines=100)

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

    # http://127.0.0.1:7861/
    # http://10.75.27.247:7861/
    blocks.queue().launch(
        share=False if platform.system() == "Windows" else False,
        server_name="127.0.0.1" if platform.system() == "Windows" else "0.0.0.0",
        # server_name="0.0.0.0",
        server_port=environment.get("port", 7860, dtype=int),
    )
    return


if __name__ == "__main__":
    main()