Upload 3 files

app.py

@@ -56,10 +56,19 @@ total_tokens = {
 }
 
 # 计算点信息
-compute_points = []  # 初始化为列表，用于存储多个用户的计算点
+compute_points = {
+    "left": 0,          # 剩余计算点
+    "total": 0,         # 总计算点
+    "used": 0,          # 已使用计算点
+    "percentage": 0,    # 使用百分比
+    "last_update": None # 最后更新时间
+}
 
 # 计算点使用日志
-compute_points_log = []  # 初始化为列表，用于存储多个用户的日志
+compute_points_log = {
+    "columns": {},      # 列名
+    "log": []           # 日志数据
+}
 
 
 # 记录启动时间
@@ -812,71 +821,105 @@ def update_model_stats(model, prompt_tokens, completion_tokens):
 
 # 获取计算点信息
 def get_compute_points():
-    global compute_points, compute_points_log
+    global compute_points, USER_DATA
     
-    # 限制只获取前两个用户的数据
-    users_to_fetch = USER_DATA[:2]
+    if USER_NUM == 0:
+        return
     
-    new_compute_points = []
-    new_compute_points_log = []
-
-    for user_index, user_config in enumerate(users_to_fetch):
-        user_compute_points = {
-            "left": 0, "total": 0, "used": 0, "percentage": 0, "last_update": None, "error": None
-        }
-        user_compute_points_log = {
-            "columns": {}, "log": [], "error": None
+    try:
+        # 使用第一个用户的会话和凭据
+        session, cookies, session_token, _, _ = USER_DATA[0]
+        
+        # 检查token是否有效
+        if is_token_expired(session_token):
+            session_token = refresh_token(session, cookies)
+            if not session_token:
+                print("刷新token失败，无法获取计算点信息")
+                return
+            USER_DATA[0] = (session, cookies, session_token, USER_DATA[0][3], USER_DATA[0][4])
+        
+        headers = {
+            "accept": "application/json, text/plain, */*",
+            "accept-language": "zh-CN,zh;q=0.9",
+            "baggage": f"sentry-environment=production,sentry-release=93da8385541a6ce339b1f41b0c94428c70657e22,sentry-public_key=3476ea6df1585dd10e92cdae3a66ff49,sentry-trace_id={TRACE_ID}",
+            "reai-ui": "1",
+            "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
+            "sec-ch-ua-mobile": "?0",
+            "sec-ch-ua-platform": "\"Windows\"",
+            "sec-fetch-dest": "empty",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-site": "same-origin",
+            "sentry-trace": SENTRY_TRACE,
+            "session-token": session_token,
+            "x-abacus-org-host": "apps",
+            "cookie": cookies
         }
+        
+        response = session.get(
+            COMPUTE_POINTS_URL,
+            headers=headers
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            if result.get("success") and "result" in result:
+                data = result["result"]
+                compute_points["left"] = data.get("computePointsLeft", 0)
+                compute_points["total"] = data.get("totalComputePoints", 0)
+                compute_points["used"] = compute_points["total"] - compute_points["left"]
+                compute_points["percentage"] = round((compute_points["used"] / compute_points["total"]) * 100, 2) if compute_points["total"] > 0 else 0
+                compute_points["last_update"] = datetime.now()
+                print(f"计算点信息更新成功: 剩余 {compute_points['left']}, 总计 {compute_points['total']}")
+                
+                # 获取计算点使用日志
+                get_compute_points_log(session, cookies, session_token)
+            else:
+                print(f"获取计算点信息失败: {result.get('error', '未知错误')}")
+        else:
+            print(f"获取计算点信息失败，状态码: {response.status_code}")
+    except Exception as e:
+        print(f"获取计算点信息异常: {e}")
 
-        try:
-            # 从USER_DATA元组中正确获取cookies
-            session, cookies, session_token, conversation_id, model_map = user_config
-            
-            headers = {
-                "Cookie": cookies,
-                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/123.0.0.0 Safari/537.36",
-            }
-            
-            # 获取计算点信息
-            compute_url = "https://abacus.art/api/trpc/user.getComputePoints"
-            response = requests.get(compute_url, headers=headers)
-            response.raise_for_status()
-            data = response.json()
-            
-            points_data = data.get("result", {}).get("data", {})
-            user_compute_points["left"] = points_data.get("left", 0)
-            user_compute_points["total"] = points_data.get("total", 0)
-            user_compute_points["used"] = points_data.get("used", 0)
-            user_compute_points["percentage"] = points_data.get("percentage", 0)
-            user_compute_points["last_update"] = datetime.now()
-
-            # 获取计算点使用日志
-            log_url = "https://abacus.art/api/trpc/user.getComputePointsLog?batch=1&input=%7B%220%22%3A%7B%22json%22%3Anull%2C%22meta%22%3A%7B%22values%22%3A%5B%22undefined%22%5D%7D%7D%7D"
-            response = requests.get(log_url, headers=headers)
-            response.raise_for_status()
-            log_data = response.json()
-            
-            log_result = log_data[0].get("result", {}).get("data", {}).get("json", {})
-            user_compute_points_log["columns"] = log_result.get("columns", {})
-            user_compute_points_log["log"] = log_result.get("log", [])
-
-        except requests.exceptions.RequestException as e:
-            error_message = f"用户 {user_index + 1} 获取计算点信息异常: {e}"
-            print(error_message)
-            user_compute_points["error"] = str(e)
-            user_compute_points_log["error"] = str(e)
-        except Exception as e:
-            error_message = f"用户 {user_index + 1} 处理计算点信息时发生未知错误: {e}"
-            print(error_message)
-            user_compute_points["error"] = str(e)
-            user_compute_points_log["error"] = str(e)
-            
-        new_compute_points.append(user_compute_points)
-        new_compute_points_log.append(user_compute_points_log)
-
-    # 更新全局变量
-    compute_points = new_compute_points
-    compute_points_log = new_compute_points_log
+# 获取计算点使用日志
+def get_compute_points_log(session, cookies, session_token):
+    global compute_points_log
+    
+    try:
+        headers = {
+            "accept": "application/json, text/plain, */*",
+            "accept-language": "zh-CN,zh;q=0.9",
+            "content-type": "application/json",
+            "reai-ui": "1",
+            "sec-ch-ua": "\"Chromium\";v=\"116\", \"Not)A;Brand\";v=\"24\", \"Google Chrome\";v=\"116\"",
+            "sec-ch-ua-mobile": "?0",
+            "sec-ch-ua-platform": "\"Windows\"",
+            "sec-fetch-dest": "empty",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-site": "same-site",
+            "session-token": session_token,
+            "x-abacus-org-host": "apps",
+            "cookie": cookies
+        }
+        
+        response = session.post(
+            COMPUTE_POINTS_LOG_URL,
+            headers=headers,
+            json={"byLlm": True}
+        )
+        
+        if response.status_code == 200:
+            result = response.json()
+            if result.get("success") and "result" in result:
+                data = result["result"]
+                compute_points_log["columns"] = data.get("columns", {})
+                compute_points_log["log"] = data.get("log", [])
+                print(f"计算点使用日志更新成功，获取到 {len(compute_points_log['log'])} 条记录")
+            else:
+                print(f"获取计算点使用日志失败: {result.get('error', '未知错误')}")
+        else:
+            print(f"获取计算点使用日志失败，状态码: {response.status_code}")
+    except Exception as e:
+        print(f"获取计算点使用日志异常: {e}")
 
 
 # 添加登录相关路由

templates/dashboard.html

@@ -550,68 +550,43 @@
         </div>
         
         <div class="grid">
-            {% if compute_points %}
-                {% for user_points in compute_points %}
             <div class="card">
                 <div class="card-header">
                     <h2 class="card-title">
                         <span class="card-icon">💰</span>
-                        用户 {{ loop.index }} 计算点信息
+                        计算点信息
                     </h2>
                 </div>
-                {% if user_points.error %}
-                <div class="status-item">
-                    <span class="status-label">错误</span>
-                    <span class="status-value danger">获取失败: {{ user_points.error }}</span>
-                </div>
-                {% else %}
                 <div class="status-item">
                     <span class="status-label">总计算点</span>
-                    <span class="status-value compute-points">{{ user_points.total|int }}</span>
+                    <span class="status-value compute-points">{{ compute_points.total|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">已使用</span>
-                    <span class="status-value compute-points">{{ user_points.used|int }}</span>
+                    <span class="status-value compute-points">{{ compute_points.used|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">剩余</span>
-                    <span class="status-value compute-points">{{ user_points.left|int }}</span>
+                    <span class="status-value compute-points">{{ compute_points.left|int }}</span>
                 </div>
                 <div class="status-item">
                     <span class="status-label">使用比例</span>
                     <div style="width: 100%; text-align: right;">
-                        <span class="status-value compute-points {% if user_points.percentage > 80 %}danger{% elif user_points.percentage > 50 %}warning{% endif %}">
-                            {{ user_points.percentage }}%
+                        <span class="status-value compute-points {% if compute_points.percentage > 80 %}danger{% elif compute_points.percentage > 50 %}warning{% endif %}">
+                            {{ compute_points.percentage }}%
                         </span>
                         <div class="progress-container">
-                            <div class="progress-bar {% if user_points.percentage > 80 %}danger{% elif user_points.percentage > 50 %}warning{% endif %}" 
-                                 data-percentage="{{ user_points.percentage }}"></div>
+                            <div class="progress-bar {% if compute_points.percentage > 80 %}danger{% elif compute_points.percentage > 50 %}warning{% endif %}" style="width: {{ compute_points.percentage }}%"></div>
                         </div>
                     </div>
                 </div>
-                {% if user_points.last_update %}
+                {% if compute_points.last_update %}
                 <div class="status-item">
                     <span class="status-label">最后更新时间</span>
-                    <span class="status-value">{{ user_points.last_update.strftime('%Y-%m-%d %H:%M:%S') }}</span>
+                    <span class="status-value">{{ compute_points.last_update.strftime('%Y-%m-%d %H:%M:%S') }}</span>
                 </div>
                 {% endif %}
-                {% endif %} {# End if user_points.error #}
-            </div>
-                {% endfor %}
-            {% else %}
-            <div class="card">
-                <div class="card-header">
-                    <h2 class="card-title">
-                        <span class="card-icon">💰</span>
-                        计算点信息
-                    </h2>
-                </div>
-                <div class="status-item">
-                    <span class="status-label">状态</span>
-                    <span class="status-value warning">暂无计算点数据或未配置用户</span>
-                </div>
             </div>
-            {% endif %} {# End if compute_points #}
             
             <div class="card">
                 <div class="card-header">
@@ -660,34 +635,26 @@
             </div>
         </div>
         
-        {% if compute_points_log %}
-            {% for user_log in compute_points_log %}
         <div class="card">
             <div class="card-header">
                 <h2 class="card-title">
                     <span class="card-icon">📊</span>
-                    用户 {{ loop.index }} 计算点使用日志
+                    计算点使用日志
                 </h2>
             </div>
-            {% if user_log.error %}
-            <div class="status-item">
-                <span class="status-label">错误</span>
-                <span class="status-value danger">获取日志失败: {{ user_log.error }}</span>
-            </div>
-            {% elif user_log.log %}
             <div class="table-container">
                 <table class="data-table">
                     <thead>
                         <tr>
-                            {% for key, value in user_log.columns.items() %}
+                            {% for key, value in compute_points_log.columns.items() %}
                             <th>{{ value }}</th>
                             {% endfor %}
                         </tr>
                     </thead>
                     <tbody>
-                        {% for entry in user_log.log %}
+                        {% for entry in compute_points_log.log %}
                         <tr>
-                            {% for key, value in user_log.columns.items() %}
+                            {% for key, value in compute_points_log.columns.items() %}
                             <td class="compute-points">{{ entry.get(key, 0) }}</td>
                             {% endfor %}
                         </tr>
@@ -695,28 +662,7 @@
                     </tbody>
                 </table>
             </div>
-            {% else %}
-            <div class="status-item">
-                <span class="status-label">状态</span>
-                <span class="status-value">暂无日志记录</span>
-            </div>
-            {% endif %}
         </div>
-            {% endfor %}
-        {% else %}
-        <div class="card">
-            <div class="card-header">
-                <h2 class="card-title">
-                    <span class="card-icon">📊</span>
-                    计算点使用日志
-                </h2>
-            </div>
-            <div class="status-item">
-                <span class="status-label">状态</span>
-                <span class="status-value warning">暂无日志数据或未配置用户</span>
-            </div>
-        </div>
-        {% endif %}
         
         <div class="card">
             <div class="card-header">
@@ -823,12 +769,6 @@
         // 初始化隐藏回到顶部按钮
         document.querySelector('.float-btn').style.opacity = '0';
         
-        // 设置进度条宽度
-        document.querySelectorAll('.progress-bar[data-percentage]').forEach(bar => {
-            const percentage = bar.getAttribute('data-percentage');
-            bar.style.width = `${percentage}%`;
-        });
-        
         // 模型统计折叠功能
         const toggleBtn = document.getElementById('toggleModelStats');
         const hiddenModels = document.querySelectorAll('.hidden-model');