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