Update app/main.py

app/main.py

@@ -51,16 +51,31 @@ STATS_FILE = "stats.json"
 
 def load_stats():
     try:
-        with open(STATS_FILE, "r") as f:
+        # 使用绝对路径确保文件位置正确
+        stats_path = os.path.abspath(STATS_FILE)
+        # 先尝试读取现有文件
+        with open(stats_path, "r") as f:
             return json.load(f)
     except (FileNotFoundError, json.JSONDecodeError):
-        return {
+        # 初始化并保存新文件
+        initial_stats = {
             "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:
@@ -367,6 +382,17 @@ async def global_exception_handler(request: Request, exc: Exception):
     return JSONResponse(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, content=ErrorResponse(message=str(exc), type="internal_error").dict())
 
 
+@app.get("/api/stats")
+async def get_stats():
+    stats = load_stats()
+    return {
+        "today_calls": stats["today_calls"],
+        "total_calls": stats["total_calls"],
+        "today_tokens": stats["today_tokens"],
+        "total_tokens": stats["total_tokens"],
+        "last_reset": stats["last_reset"]
+    }
+
 @app.get("/", response_class=HTMLResponse)
 async def root():
     html_content = f"""