diff --git a/frontend/__init__.py b/frontend/__init__.py
new file mode 100644
index 0000000..7313f97
--- /dev/null
+++ b/frontend/__init__.py
@@ -0,0 +1,3 @@
+from .chat_page import chat_page
+from .setting_page import setting_page
+from .train_page import train_page
\ No newline at end of file
diff --git a/main.py b/main.py
index 5a93227..d0eabe0 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,6 @@
 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
+from frontend import chat_page,setting_page,train_page
 from db import initialize_db as init_db,get_engine
 
 if __name__ == "__main__":