feat: 重新加入将usage显示为进度条功能

原功能显示赠金进度条，
本在 ec10bf7faa610cac1d8ff35a420426fd9f60d3b6 引入，
在 94f0d7d23587384474abc2472c96689144208bba 移除

assets/custom.css

@@ -111,7 +111,7 @@ footer {
 }
 .progress-bar {
     background-color: var(--input-background-fill);;
-    margin: 0 1em;
+    margin: .5em 0 !important;
     height: 20px;
     border-radius: 10px;
     overflow: hidden;

config_example.json

@@ -2,6 +2,7 @@
     // 你的OpenAI API Key，一般必填，
     // 若缺省填为 "openai_api_key": "" 则必须再在图形界面中填入API Key
     "openai_api_key": "",
+    "usage_limit": 120, // API Key的当月限额，单位：美元
     // 你的xmchat API Key，与OpenAI API Key不同
     "xmchat_api_key": "",
     "language": "auto",

locale/en_US.json

@@ -50,6 +50,7 @@
     "，本次对话累计消耗了 ": "，Total cost for this dialogue is ",
     "**获取API使用情况失败**": "**Failed to get API usage**",
     "**本月使用金额** ": "**Monthly usage** ",
+    "本月使用金额": "Monthly usage",
     "获取API使用情况失败:": "Failed to get API usage:",
     "API密钥更改为了": "The API key is changed to",
     "JSON解析错误,收到的内容: ": "JSON parsing error, received content: ",

locale/ja_JP.json

@@ -50,6 +50,7 @@
     "，本次对话累计消耗了 ": ", 今の会話で消費合計 ",
     "**获取API使用情况失败**": "**API使用状況の取得に失敗しました**",
     "**本月使用金额** ": "**今月の使用料金** ",
+    "本月使用金额": "今月の使用料金",
     "获取API使用情况失败:": "API使用状況の取得に失敗しました:",
     "API密钥更改为了": "APIキーが変更されました",
     "JSON解析错误,收到的内容: ": "JSON解析エラー、受信内容: ",

modules/config.py

@@ -19,6 +19,7 @@ __all__ = [
     "advance_docs",
     "update_doc_config",
     "render_latex",
+    "usage_limit",
     "multi_api_key",
     "server_name",
     "server_port",
@@ -82,6 +83,8 @@ if render_latex:
 else:
     os.environ["RENDER_LATEX"] = "no"
 
+usage_limit = os.environ.get("USAGE_LIMIT", config.get("usage_limit", 120))
+
 ## 多账户机制
 multi_api_key = config.get("multi_api_key", False) # 是否开启多账户机制
 if multi_api_key:

modules/models/models.py

@@ -25,7 +25,7 @@ from ..presets import *
 from ..llama_func import *
 from ..utils import *
 from .. import shared
-from ..config import retrieve_proxy
+from ..config import retrieve_proxy, usage_limit
 from modules import config
 from .base_model import BaseLLMModel, ModelType
 
@@ -90,8 +90,19 @@ class OpenAIClient(BaseLLMModel):
             except Exception as e:
                 logging.error(f"获取API使用情况失败:" + str(e))
                 return i18n("**获取API使用情况失败**")
-            rounded_usage = "{:.5f}".format(usage_data["total_usage"] / 100)
-            return i18n("**本月使用金额** ") + f"\u3000 ${rounded_usage}"
+            # rounded_usage = "{:.5f}".format(usage_data["total_usage"] / 100)
+            rounded_usage = round(usage_data["total_usage"] / 100, 5)
+            usage_percent = round(usage_data["total_usage"] / usage_limit, 2)
+            # return i18n("**本月使用金额** ") + f"\u3000 ${rounded_usage}"
+            return """\
+                <b>""" + i18n("本月使用金额") + f"""</b>
+                <div class="progress-bar">
+                    <div class="progress" style="width: {usage_percent}%;">
+                        <span class="progress-text">{usage_percent}%</span>
+                    </div>
+                </div>
+                <div style="display: flex; justify-content: space-between;"><span>${rounded_usage}</span><span>${usage_limit}</span></div>
+                """
         except requests.exceptions.ConnectTimeout:
             status_text = (
                 STANDARD_ERROR_MSG + CONNECTION_TIMEOUT_MSG + ERROR_RETRIEVE_MSG