diff --git a/heartbeat.py b/heartbeat.py
index 2fc27e5..4600b46 100644
--- a/heartbeat.py
+++ b/heartbeat.py
@@ -14,6 +14,8 @@ import platform
 
 # 获取当前脚本所在的目录
 current_dir = os.path.dirname(os.path.realpath(__file__))
+if getattr(sys, 'frozen', False):
+    current_dir = sys._MEIPASS  # 如果是frozen的（即打包后的exe），使用这个目录
 
 # 创建日志文件
 log_file_path = os.path.join(current_dir, 'heartbeat.log')