From 519a5f37736841e630218b44db0c8ebc961a6fc1 Mon Sep 17 00:00:00 2001 From: carry <2641257231@qq.com> Date: Sun, 6 Apr 2025 14:49:01 +0800 Subject: [PATCH] =?UTF-8?q?feat(frontend):=20=E6=B7=BB=E5=8A=A0=E5=89=8D?= =?UTF-8?q?=E7=AB=AF=E9=A1=B5=E9=9D=A2=E6=A8=A1=E5=9D=97=E5=B9=B6=E5=AE=9E?= =?UTF-8?q?=E7=8E=B0=E5=9F=BA=E6=9C=AC=E5=B8=83=E5=B1=80?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit - 新增 chat_page.py、setting_page.py 和 train_page.py 文件,分别实现聊天、设置和微调页面的基本布局 - 添加 main.py 文件,集成所有页面并创建主应用 - 在 requirements.txt 中添加 gradio 依赖 --- frontend/chat_page.py | 9 +++++++++ frontend/setting_page.py | 9 +++++++++ frontend/train_page.py | 9 +++++++++ main.py | 23 +++++++++++++++++++++++ requirements.txt | 3 ++- 5 files changed, 52 insertions(+), 1 deletion(-) create mode 100644 frontend/chat_page.py create mode 100644 frontend/setting_page.py create mode 100644 frontend/train_page.py create mode 100644 main.py diff --git a/frontend/chat_page.py b/frontend/chat_page.py new file mode 100644 index 0000000..b0fbf39 --- /dev/null +++ b/frontend/chat_page.py @@ -0,0 +1,9 @@ +import gradio as gr + +def chat_page(): + with gr.Blocks() as demo: + gr.Markdown("## 聊天") + with gr.Row(): + with gr.Column(): + pass + return demo \ No newline at end of file diff --git a/frontend/setting_page.py b/frontend/setting_page.py new file mode 100644 index 0000000..36c03bb --- /dev/null +++ b/frontend/setting_page.py @@ -0,0 +1,9 @@ +import gradio as gr + +def setting_page(): + with gr.Blocks() as demo: + gr.Markdown("## 设置") + with gr.Row(): + with gr.Column(): + pass + return demo \ No newline at end of file diff --git a/frontend/train_page.py b/frontend/train_page.py new file mode 100644 index 0000000..44e2d82 --- /dev/null +++ b/frontend/train_page.py @@ -0,0 +1,9 @@ +import gradio as gr + +def train_page(): + with gr.Blocks() as demo: + gr.Markdown("## 微调") + with gr.Row(): + with gr.Column(): + pass + return demo \ No newline at end of file diff --git a/main.py b/main.py new file mode 100644 index 0000000..0a915d2 --- /dev/null +++ b/main.py @@ -0,0 +1,23 @@ +import gradio as gr +from frontend.setting_page import setting_page +from frontend.chat_page import chat_page +from frontend.train_page import train_page +def main(): + setting_demo = setting_page() + chat_demo = chat_page() + train_demo = train_page() + + with gr.Blocks() as app: + gr.Markdown("# 基于文档驱动的自适应编码大模型微调框架") + with gr.Tabs(): + with gr.TabItem("微调"): + train_demo.render() + with gr.TabItem("聊天"): + chat_demo.render() + with gr.TabItem("设置"): + setting_demo.render() + + app.launch() + +if __name__ == "__main__": + main() \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 4e8ed94..5ec7739 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,4 @@ openai>=1.0.0 python-dotenv>=1.0.0 -pydantic>=2.0.0 \ No newline at end of file +pydantic>=2.0.0 +gradio>=3.0.0 \ No newline at end of file