Update app/main.py

app/main.py

@@ -51,35 +51,45 @@ STATS_FILE = "stats.json"
 
 def load_stats():
     try:
-        # 使用绝对路径确保文件位置正确
         stats_path = os.path.abspath(STATS_FILE)
-        # 先尝试读取现有文件
+        # 确保目录存在
+        os.makedirs(os.path.dirname(stats_path), exist_ok=True)
+        
+        # 如果文件不存在则创建
+        if not os.path.exists(stats_path):
+            initial_stats = {
+                "total_calls": 0,
+                "today_calls": 0,
+                "total_tokens": 0,
+                "today_tokens": 0,
+                "last_reset": datetime.now().isoformat()
+            }
+            with open(stats_path, "w") as f:
+                json.dump(initial_stats, f, indent=2)
+            logger.info(f"已创建初始统计文件: {stats_path}")
+            return initial_stats
+            
+        # 读取现有文件
         with open(stats_path, "r") as f:
             return json.load(f)
-    except (FileNotFoundError, json.JSONDecodeError):
-        # 初始化并保存新文件
-        initial_stats = {
+    except Exception as e:
+        logger.error(f"加载统计文件失败: {str(e)}")
+        return {
             "total_calls": 0,
             "today_calls": 0,
             "total_tokens": 0,
             "today_tokens": 0,
             "last_reset": datetime.now().isoformat()
         }
-        try:
-            # 确保目录存在
-            os.makedirs(os.path.dirname(stats_path), exist_ok=True)
-            # 写入初始数据
-            with open(stats_path, "w") as f:
-                json.dump(initial_stats, f, indent=2)
-            logger.info(f"已创建初始统计文件: {stats_path}")
-            return initial_stats
-        except Exception as e:
-            logger.error(f"创建统计文件失败: {str(e)}")
-            return initial_stats
 
 def save_stats(stats):
-    with open(STATS_FILE, "w") as f:
-        json.dump(stats, f, indent=2)
+    try:
+        stats_path = os.path.abspath(STATS_FILE)
+        os.makedirs(os.path.dirname(stats_path), exist_ok=True)
+        with open(stats_path, "w") as f:
+            json.dump(stats, f, indent=2)
+    except Exception as e:
+        logger.error(f"保存统计文件失败: {str(e)}")
 
 def update_stats(calls=0, tokens=0):
     stats = load_stats()