main-old.py

"""
@Author：Huterox
@Description：Go For It
@Time：2024/4/16 8:58
@Copyright：©2018-2024 awesome!
"""
import os

import gradio as gr

from utils import getConfig
from webui.gradio.chat import ChatBotComponent
from handler.helperChat import ChatBotHandler
from webui.gradio.novel import NovelComponent
from webui.gradio.setting import SettingComponent

my_theme = gr.Theme.load("./theme/miku.json")

config = getConfig()

"""
*************注意，当前所有的功能都将封装到组件当中，换一句话说所有的base功能实现都在handler里面***********
"""
if __name__ == '__main__':
    os.environ["no_proxy"] = "localhost,127.0.0.1,::1"
    with gr.Blocks(
        title='Novel tweet generator',
        theme=my_theme,
        fill_height = False,
        css="""
        footer {visibility: hidden}
        """,

    ) as demo:
        title = gr.Markdown("#### 创作生成器v0.1-beta（￣︶￣）↗　")
        # 顶部tag导航
        with gr.Tab(label="Novel助手"):
            # chat 对话机器人
            chat = ChatBotComponent(gr)
            # chat.set_response(ChatBotHandler().chat)
            chat.set_streamBot(ChatBotHandler())
            chat.creat()

        with gr.Tab(label="Novel合成"):
            # novel合成
            novel = NovelComponent(gr)
            novel.create()

        with gr.Tab(label="key设置"):
            # 设置页面
            setting = SettingComponent(gr)
            setting.create()

    demo.launch(share=config.get("share",True))