refactor(frontend): 重构前端页面并添加独立运行功能

- 在 chat_page 和 prompt_manage_page 中添加了独立运行的入口
- 引入 sys 和 pathlib 模块以支持路径操作
- 修改了模块导入方式,使其能够作为独立脚本运行
- 优化了代码结构,提高了可读性和可维护性
This commit is contained in:
carry 2025-04-10 21:18:05 +08:00
parent 402bc73dce
commit 1b3f546669
2 changed files with 18 additions and 2 deletions

View File

@ -1,4 +1,7 @@
import gradio as gr
import sys
from pathlib import Path
sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import model,tokenizer
def chat_page():
@ -23,6 +26,10 @@ def chat_page():
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
bot, chatbot, chatbot
)
clear.click(lambda: None, None, chatbot, queue=False)
return demo
return demo
if __name__ == "__main__":
chat_page().queue().launch()

View File

@ -1,5 +1,8 @@
import gradio as gr
import sys
from pathlib import Path
from typing import List
sys.path.append(str(Path(__file__).resolve().parent.parent))
from global_var import prompt_store
from schema.prompt import promptTempleta
def prompt_manage_page():
@ -113,4 +116,10 @@ def prompt_manage_page():
outputs=[prompt_table]
)
return demo
return demo
if __name__ == "__main__":
demo = prompt_manage_page()
demo.queue()
demo.launch()