Upload 4 files

app.py

@@ -21,6 +21,8 @@ API_ENDPOINT_URL = "https://abacus.ai/api/v0/describeDeployment"
 MODEL_LIST_URL = "https://abacus.ai/api/v0/listExternalApplications"
 CHAT_URL = "https://apps.abacus.ai/api/_chatLLMSendMessageSSE"
 USER_INFO_URL = "https://abacus.ai/api/v0/_getUserInfo"
+COMPUTE_POINTS_URL = "https://apps.abacus.ai/api/_getOrganizationComputePoints"
+COMPUTE_POINTS_LOG_URL = "https://abacus.ai/api/v0/_getOrganizationComputePointLog"
 
 
 USER_AGENTS = [
@@ -51,6 +53,21 @@ total_tokens = {
     "total": 0         # 总token统计
 }
 
+# 计算点信息
+compute_points = {
+    "left": 0,          # 剩余计算点
+    "total": 0,         # 总计算点
+    "used": 0,          # 已使用计算点
+    "percentage": 0,    # 使用百分比
+    "last_update": None # 最后更新时间
+}
+
+# 计算点使用日志
+compute_points_log = {
+    "columns": {},      # 列名
+    "log": []           # 日志数据
+}
+
 
 # HTML模板
 INDEX_HTML = """
@@ -123,6 +140,9 @@ INDEX_HTML = """
         .status-value.warning {
             color: #ffc107;
         }
+        .status-value.danger {
+            color: #dc3545;
+        }
         .footer {
             margin-top: 2rem;
             text-align: center;
@@ -183,6 +203,30 @@ INDEX_HTML = """
             font-family: monospace;
             color: #28a745;
         }
+        .compute-points {
+            font-family: monospace;
+            color: #6f42c1;
+            font-weight: bold;
+        }
+        .progress-container {
+            width: 100%;
+            height: 10px;
+            background-color: #e9ecef;
+            border-radius: 5px;
+            margin-top: 0.5rem;
+            overflow: hidden;
+        }
+        .progress-bar {
+            height: 100%;
+            border-radius: 5px;
+            background-color: #28a745;
+        }
+        .progress-bar.warning {
+            background-color: #ffc107;
+        }
+        .progress-bar.danger {
+            background-color: #dc3545;
+        }
         @media (max-width: 768px) {
             .container {
                 padding: 1rem;
@@ -224,6 +268,61 @@ INDEX_HTML = """
             </div>
         </div>
 
+        <h2>💰 计算点信息</h2>
+        <div class="status-card">
+            <div class="status-item">
+                <span class="status-label">总计算点</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">{{ compute_points.used|int }}</span>
+            </div>
+            <div class="status-item">
+                <span class="status-label">剩余</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 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 compute_points.percentage > 80 %}danger{% elif compute_points.percentage > 50 %}warning{% endif %}" style="width: {{ compute_points.percentage }}%"></div>
+                    </div>
+                </div>
+            </div>
+            {% if compute_points.last_update %}
+            <div class="status-item">
+                <span class="status-label">最后更新时间</span>
+                <span class="status-value">{{ compute_points.last_update.strftime('%Y-%m-%d %H:%M:%S') }}</span>
+            </div>
+            {% endif %}
+        </div>
+
+        <h2>📊 计算点使用日志</h2>
+        <div class="status-card">
+            <table class="usage-table">
+                <thead>
+                    <tr>
+                        {% for key, value in compute_points_log.columns.items() %}
+                        <th>{{ value }}</th>
+                        {% endfor %}
+                    </tr>
+                </thead>
+                <tbody>
+                    {% for entry in compute_points_log.log %}
+                    <tr>
+                        {% for key, value in compute_points_log.columns.items() %}
+                        <td class="compute-points">{{ entry.get(key, 0) }}</td>
+                        {% endfor %}
+                    </tr>
+                    {% endfor %}
+                </tbody>
+            </table>
+        </div>
+
         <h2>🔍 模型使用统计</h2>
         <div class="status-card">
             <div class="status-item">
@@ -983,6 +1082,9 @@ def keep_alive():
 
 @app.route("/", methods=["GET"])
 def index():
+    # 在每次访问首页时更新计算点信息
+    get_compute_points()
+    
     uptime = datetime.now() - START_TIME
     days = uptime.days
     hours, remainder = divmod(uptime.seconds, 3600)
@@ -1003,7 +1105,9 @@ def index():
         models=sorted(list(MODELS)),
         year=datetime.now().year,
         model_stats=model_usage_stats,
-        total_tokens=total_tokens
+        total_tokens=total_tokens,
+        compute_points=compute_points,
+        compute_points_log=compute_points_log
     )
 
 
@@ -1042,8 +1146,115 @@ def update_model_stats(model, prompt_tokens, completion_tokens):
     total_tokens["total"] += (prompt_tokens + completion_tokens)
 
 
+# 获取计算点信息
+def get_compute_points():
+    global compute_points, USER_DATA
+    
+    if USER_NUM == 0:
+        return
+    
+    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}")
+
+# 获取计算点使用日志
+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}")
+
+
 if __name__ == "__main__":
     # 启动保活线程
     threading.Thread(target=keep_alive, daemon=True).start()
+    
+    # 获取初始计算点信息
+    get_compute_points()
+    
     port = int(os.environ.get("PORT", 9876))
     app.run(port=port, host="0.0.0.0")