diff --git a/core/__init__.py b/core/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/core/__init__.py
@@ -0,0 +1 @@
+
diff --git a/predict.py b/core/predict.py
similarity index 100%
rename from predict.py
rename to core/predict.py
diff --git a/predict_onnx.py b/core/predict_onnx.py
similarity index 100%
rename from predict_onnx.py
rename to core/predict_onnx.py
diff --git a/gui/__init__.py b/gui/__init__.py
new file mode 100644
index 0000000..8b13789
--- /dev/null
+++ b/gui/__init__.py
@@ -0,0 +1 @@
+
diff --git a/dark_mode_style_fix.py b/gui/dark_mode_style_fix.py
similarity index 100%
rename from dark_mode_style_fix.py
rename to gui/dark_mode_style_fix.py
diff --git a/similar_history_match.py b/gui/similar_history_match.py
similarity index 98%
rename from similar_history_match.py
rename to gui/similar_history_match.py
index ea9f1ff..7e4df25 100644
--- a/similar_history_match.py
+++ b/gui/similar_history_match.py
@@ -1,5 +1,11 @@
 import numpy as np
 import pandas as pd
+# 从父目录找config
+import sys
+from pathlib import Path
+root_dir = Path(__file__).parent.parent
+if str(root_dir) not in sys.path:
+    sys.path.insert(0, str(root_dir))
 from config import MONSTER_COUNT
 from config import FIELD_FEATURE_COUNT
 
diff --git a/simular_history_match_ui.py b/gui/similar_history_match_ui.py
similarity index 98%
rename from simular_history_match_ui.py
rename to gui/similar_history_match_ui.py
index 6ddf18c..2b6a02f 100644
--- a/simular_history_match_ui.py
+++ b/gui/similar_history_match_ui.py
@@ -4,7 +4,13 @@
 import numpy as np
 import logging
 
-from similar_history_match import HistoryMatch
+from .similar_history_match import HistoryMatch
+# 从父目录找config
+import sys
+from pathlib import Path
+root_dir = Path(__file__).parent.parent
+if str(root_dir) not in sys.path:
+    sys.path.insert(0, str(root_dir))
 from config import MONSTER_COUNT, MONSTER_DATA
 
 logger = logging.getLogger(__name__)
diff --git a/main.py b/main.py
index 7721a90..5dff549 100644
--- a/main.py
+++ b/main.py
@@ -18,15 +18,15 @@
 import loadData
 import auto_fetch
 from maa_adb_connector import MaaAdbConnector, MaaFrameworkDetector
-from dark_mode_style_fix import DarkModeStyleFix
-import similar_history_match
+from gui.dark_mode_style_fix import DarkModeStyleFix
+from gui.similar_history_match import HistoryMatch
+from gui.similar_history_match_ui import HistoryMatchUI
 import recognize
 from recognize import MONSTER_COUNT
 from specialmonster import SpecialMonsterHandler
 import data_package
 import winrt_capture
 from config import FIELD_FEATURE_COUNT, MONSTER_DATA
-from simular_history_match_ui import HistoryMatchUI
 from input_panel_ui import InputPanelUI
 
 logging.getLogger().setLevel(logging.DEBUG)
@@ -42,12 +42,12 @@
 
 
 try:
-    from predict import CannotModel
+    from core.predict import CannotModel
     from train import UnitAwareTransformer
 
     logger.info("Using PyTorch model for predictions.")
 except:
-    from predict_onnx import CannotModel
+    from core.predict_onnx import CannotModel
 
     logger.info("Using ONNX model for predictions.")
 
@@ -123,7 +123,7 @@ def __init__(self):
         # 初始化UI后加载历史数据
         logger.info("尝试获取错题本")
         self.history_match = None
-        self.history_match = similar_history_match.HistoryMatch()
+        self.history_match = HistoryMatch()
         # Ensure feat_past and N_history are initialized
         try:
             self.history_match.feat_past = np.hstack([self.history_match.past_left, self.history_match.past_right])
diff --git a/main_old.py b/main_old.py
index 89d18ed..ab9f444 100644
--- a/main_old.py
+++ b/main_old.py
@@ -7,12 +7,12 @@
 import numpy as np
 import math
 from PIL import Image, ImageTk
-from predict import CannotModel
+from core.predict import CannotModel
 import loadData
 import recognize
 from train import UnitAwareTransformer
 from recognize import MONSTER_COUNT
-from similar_history_match import HistoryMatch
+from gui.similar_history_match import HistoryMatch
 from auto_fetch import AutoFetch
 
 logging.getLogger().setLevel(logging.DEBUG)
diff --git a/multi_instance.py b/multi_instance.py
index 2ed65b2..3ec7c1b 100644
--- a/multi_instance.py
+++ b/multi_instance.py
@@ -5,7 +5,7 @@
 import subprocess
 from pathlib import Path
 from PyQt6.QtWidgets import (
-    QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
+    QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout,
     QPushButton, QLabel, QPlainTextEdit, QSpinBox, QComboBox, QCheckBox,
     QMessageBox, QSplitter, QScrollArea, QFrame, QLineEdit
 )
@@ -109,10 +109,10 @@ def get_cannot_model():
     if _cannot_model is None:
         logger.info("首次初始化 CannotModel...")
         try:
-            from predict import CannotModel
+            from core.predict import CannotModel
             logger.info("Using PyTorch model for predictions.")
         except Exception:
-            from predict_onnx import CannotModel
+            from core.predict_onnx import CannotModel
             logger.info("Using ONNX model for predictions.")
         
         _cannot_model = CannotModel()
diff --git a/tools/convert_model.py b/tools/convert_model.py
index f55852b..c8856c2 100644
--- a/tools/convert_model.py
+++ b/tools/convert_model.py
@@ -1,9 +1,9 @@
 import sys
 sys.path.append(".")
 
-import predict
+import core.predict
 from train import UnitAwareTransformer
-import predict_onnx
+import core.predict_onnx
 import numpy as np
 from recognize import MONSTER_COUNT