From 1b3f546669db7f4ac7916e27ee41090c6d8858a5 Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Thu, 10 Apr 2025 21:18:05 +0800 Subject: [PATCH] =?UTF-8?q?refactor(frontend):=20=E9=87=8D=E6=9E=84?= =?UTF-8?q?=E5=89=8D=E7=AB=AF=E9=A1=B5=E9=9D=A2=E5=B9=B6=E6=B7=BB=E5=8A=A0?= =?UTF-8?q?=E7=8B=AC=E7=AB=8B=E8=BF=90=E8=A1=8C=E5=8A=9F=E8=83=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 在 chat_page 和 prompt_manage_page 中添加了独立运行的入口 - 引入 sys 和 pathlib 模块以支持路径操作 - 修改了模块导入方式,使其能够作为独立脚本运行 - 优化了代码结构,提高了可读性和可维护性 --- frontend/chat_page.py | 9 ++++++++- frontend/prompt_manage_page.py | 11 ++++++++++- 2 files changed, 18 insertions(+), 2 deletions(-) diff --git a/frontend/chat_page.py b/frontend/chat_page.py index dfe3e17..ebd8bbf 100644 --- a/frontend/chat_page.py +++ b/frontend/chat_page.py @@ -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 \ No newline at end of file + return demo + +if __name__ == "__main__": + chat_page().queue().launch() \ No newline at end of file diff --git a/frontend/prompt_manage_page.py b/frontend/prompt_manage_page.py index e52576d..a2426c7 100644 --- a/frontend/prompt_manage_page.py +++ b/frontend/prompt_manage_page.py @@ -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 \ No newline at end of file + return demo + + +if __name__ == "__main__": + demo = prompt_manage_page() + demo.queue() + demo.launch() \ No newline at end of file