refactor(frontend): 重构前端页面并添加独立运行功能
- 在 chat_page 和 prompt_manage_page 中添加了独立运行的入口 - 引入 sys 和 pathlib 模块以支持路径操作 - 修改了模块导入方式,使其能够作为独立脚本运行 - 优化了代码结构,提高了可读性和可维护性
This commit is contained in:
parent
402bc73dce
commit
1b3f546669
@ -1,4 +1,7 @@
|
|||||||
import gradio as gr
|
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
|
from global_var import model,tokenizer
|
||||||
|
|
||||||
def chat_page():
|
def chat_page():
|
||||||
@ -23,6 +26,10 @@ def chat_page():
|
|||||||
|
|
||||||
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
|
msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
|
||||||
bot, chatbot, chatbot
|
bot, chatbot, chatbot
|
||||||
|
|
||||||
)
|
)
|
||||||
clear.click(lambda: None, None, chatbot, queue=False)
|
clear.click(lambda: None, None, chatbot, queue=False)
|
||||||
return demo
|
return demo
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
chat_page().queue().launch()
|
@ -1,5 +1,8 @@
|
|||||||
import gradio as gr
|
import gradio as gr
|
||||||
|
import sys
|
||||||
|
from pathlib import Path
|
||||||
from typing import List
|
from typing import List
|
||||||
|
sys.path.append(str(Path(__file__).resolve().parent.parent))
|
||||||
from global_var import prompt_store
|
from global_var import prompt_store
|
||||||
from schema.prompt import promptTempleta
|
from schema.prompt import promptTempleta
|
||||||
def prompt_manage_page():
|
def prompt_manage_page():
|
||||||
@ -113,4 +116,10 @@ def prompt_manage_page():
|
|||||||
outputs=[prompt_table]
|
outputs=[prompt_table]
|
||||||
)
|
)
|
||||||
|
|
||||||
return demo
|
return demo
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
demo = prompt_manage_page()
|
||||||
|
demo.queue()
|
||||||
|
demo.launch()
|
Loading…
x
Reference in New Issue
Block a user