import gradio as gr
from modules.tiktok import tiktok_parser
from modules.bili import bili_parser
from modules.bvid2acid import bv2acid
from utils import LANG

ZH2EN = {
    "本站不提供任何视频存储服务，仅提供最基本的解析服务，请勿滥用": "This site does not provide any audio storage services, only provide the most basic parsing services, please DO NOT abuse",
    "抖音": "Tiktok",
    "B站": "Bilibili",
}


def _L(zh_txt: str):
    return ZH2EN[zh_txt] if LANG else zh_txt


if __name__ == "__main__":
    with gr.Blocks() as demo:
        gr.Markdown(_L("本站不提供任何视频存储服务，仅提供最基本的解析服务，请勿滥用"))
        with gr.Tab(_L("抖音")):
            tiktok_parser()

        with gr.Tab(_L("B站")):
            with gr.Column():
                bv2acid()
                bili_parser()

    demo.launch()