frontend/chat_page.py

@@ -1,7 +1,4 @@
 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():
@@ -26,10 +23,6 @@ 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
-
-if __name__ == "__main__":
-    chat_page().queue().launch()

frontend/model_manage_page.py

@@ -18,7 +18,7 @@ def model_manage_page():
         output_text = gr.Textbox(label="当前状态", interactive=False)
         with gr.Row():
             with gr.Column(scale=3):
-                model_select_dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True)  # 将子文件夹列表添加到Dropdown组件中，并设置为可选
+                dropdown = gr.Dropdown(choices=model_folders, label="选择模型", interactive=True)  # 将子文件夹列表添加到Dropdown组件中，并设置为可选
                 max_seq_length_input = gr.Number(label="最大序列长度", value=4096, precision=0)
                 load_in_4bit_input = gr.Checkbox(label="使用4位量化", value=True)
             with gr.Column(scale=1):
@@ -49,7 +49,7 @@ def model_manage_page():
             except Exception as e:
                 return f"加载模型时出错: {str(e)}"
 
-        load_button.click(fn=load_model, inputs=[model_select_dropdown, max_seq_length_input, load_in_4bit_input], outputs=output_text)
+        load_button.click(fn=load_model, inputs=[dropdown, max_seq_length_input, load_in_4bit_input], outputs=output_text)
 
         def unload_model():
             try:
@@ -98,7 +98,7 @@ def model_manage_page():
             except Exception as e:
                 return f"刷新模型列表时出错: {str(e)}"
         
-        refresh_button.click(fn=refresh_model_list, inputs=None, outputs=model_select_dropdown)
+        refresh_button.click(fn=refresh_model_list, inputs=None, outputs=dropdown)
 
     return demo
 

frontend/prompt_manage_page.py

@@ -1,8 +1,5 @@
 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():
@@ -117,9 +114,3 @@ def prompt_manage_page():
         )
 
         return demo
-    
-
-if __name__ == "__main__":
-    demo = prompt_manage_page()
-    demo.queue()
-    demo.launch()