nai-aclab 1.2.0 → 1.2.1

package/README.md

@@ -72,6 +72,8 @@ API 토큰을 아직 넣고 싶지 않다면 `API 없이 체험하기`를 켜둔
 
 ![API 설정 메뉴](docs/screenshots/07-api-settings.png)
 
+왼쪽 아래의 `NAI V5 할당량` 패널에서는 현재 V5 사용 한도와 Image Anlas를 확인할 수 있습니다. 약 1분마다 자동으로 갱신되며, `↻` 버튼을 누르면 바로 새로고침됩니다. API 토큰을 바꾸거나 이미지 생성이 끝난 뒤에도 자동으로 최신 정보가 반영됩니다.
+
 ## 기본 사용 순서
 
 ### 1. 작가 태그 등록

package/app.py

@@ -850,6 +850,69 @@ class NovelAIClient:
             else:
                 raise RuntimeError(f"알 수 없는 API 응답 형식: {maybe_json[:300]}")
 
+    def subscription_quota(self) -> dict:
+        """Return a safe, UI-ready subset of the NovelAI subscription response."""
+        if self.settings.mock_mode:
+            return {"available": False, "reason": "체험 모드에서는 할당량을 확인할 수 없습니다."}
+        token = self.settings.token.strip()
+        if not token:
+            return {"available": False, "reason": "API 토큰을 입력하면 할당량을 확인합니다."}
+
+        headers = {
+            "Authorization": f"Bearer {token}",
+            "User-Agent": self.settings.user_agent.strip() or DEFAULT_USER_AGENT,
+            "Accept": "application/json",
+            "Origin": "https://novelai.net",
+            "Referer": "https://novelai.net/",
+        }
+        errors: list[str] = []
+        # Subscription requests never follow a user-configured image endpoint.
+        # The token is sent only to NovelAI's official hosts.
+        for endpoint in (
+            "https://image.novelai.net/user/subscription",
+            "https://api.novelai.net/user/subscription",
+        ):
+            request = urllib.request.Request(endpoint, headers=headers, method="GET")
+            try:
+                with urllib.request.urlopen(request, timeout=20) as response:
+                    data = json.loads(response.read().decode("utf-8"))
+                break
+            except urllib.error.HTTPError as exc:
+                errors.append(str(exc.code))
+            except (urllib.error.URLError, TimeoutError, json.JSONDecodeError) as exc:
+                errors.append(type(exc).__name__)
+        else:
+            status = errors[-1] if errors else "unknown"
+            return {"available": False, "reason": f"할당량 정보를 불러오지 못했습니다. ({status})"}
+
+        steps = data.get("trainingStepsLeft") or {}
+        fixed_anlas = int(steps.get("fixedTrainingStepsLeft") or 0)
+        purchased_anlas = int(steps.get("purchasedTrainingSteps") or 0)
+        usage = data.get("usage") or data.get("imageGenerationUsage") or {}
+        if not isinstance(usage, dict):
+            usage = {}
+        percent = usage.get("percent")
+        try:
+            percent = float(percent) if percent is not None else None
+        except (TypeError, ValueError):
+            percent = None
+        next_percent = usage.get("timeUntilNextPercent")
+        try:
+            next_percent = float(next_percent) if next_percent is not None else None
+        except (TypeError, ValueError):
+            next_percent = None
+        return {
+            "available": True,
+            "tier": int(data.get("tier") or 0),
+            "active": bool(data.get("active")),
+            "subscription_anlas": fixed_anlas,
+            "paid_anlas": purchased_anlas,
+            "total_anlas": fixed_anlas + purchased_anlas,
+            "v5_percent": percent,
+            "v5_is_negative": bool(usage.get("isNegative")),
+            "v5_next_percent_seconds": next_percent,
+        }
+
 
 class ScrollFrame(ttk.Frame if ttk else object):
     def __init__(self, parent):

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "nai-aclab",
-  "version": "1.2.0",
+  "version": "1.2.1",
   "description": "Local web UI for experimenting with NovelAI artist tag weight combinations.",
   "license": "MIT",
   "type": "commonjs",

package/web/index.html

@@ -26,6 +26,13 @@
           <button class="nav-item" data-tab="history">◈ 생성 기록</button>
         </nav>
         <div class="sidebar-tools">
+          <section class="quota-widget" id="quotaWidget" aria-live="polite">
+            <div class="quota-widget-header">
+              <span>NAI V5 할당량</span>
+              <button class="quota-refresh" id="quotaRefreshButton" type="button" title="할당량 새로고침" aria-label="할당량 새로고침">↻</button>
+            </div>
+            <div class="quota-widget-content" id="quotaWidgetContent">할당량을 확인하는 중입니다.</div>
+          </section>
           <button class="nav-item settings-nav-item" data-tab="settings">⚙ API 설정</button>
         </div>
         <div class="sidebar-footer">

package/web/main.js

@@ -28,9 +28,12 @@ let modalZoom = 1;
 let modalPanX = 0;
 let modalPanY = 0;
 let modalDragging = false;
-let modalDragStart = { x: 0, y: 0, panX: 0, panY: 0 };
+let modalDragStart = { x: 0, y: 0, panX: 0, panY: 0 };
 let presetPickerKind = null;
 let activeJobs = { generate: null, batting: null };
+let quotaData = null;
+let quotaRefreshInFlight = false;
+let quotaPollingStarted = false;
 const pendingJobId = "__pending__";
 
 const pageCopy = {
@@ -146,6 +149,96 @@ function setSaveState(text, tone = "success") {
   root.dataset.tone = tone;
 }
 
+function formatQuotaDuration(seconds) {
+  const value = Number(seconds);
+  if (!Number.isFinite(value) || value < 0) return "";
+  // Older responses used milliseconds while the current API uses seconds.
+  const normalized = value > 100000 ? value / 1000 : value;
+  const minutes = Math.max(1, Math.ceil(normalized / 60));
+  if (minutes < 60) return `${minutes}분`;
+  const hours = Math.floor(minutes / 60);
+  const remainder = minutes % 60;
+  return remainder ? `${hours}시간 ${remainder}분` : `${hours}시간`;
+}
+
+function renderQuota() {
+  const root = $("#quotaWidgetContent");
+  if (!root) return;
+  root.innerHTML = "";
+  root.classList.remove("exhausted");
+  if (!quotaData?.available) {
+    root.textContent = quotaData?.reason || "API 토큰을 입력하면 할당량을 확인합니다.";
+    return;
+  }
+  const tierNames = ["Paper", "Tablet", "Scroll", "Opus"];
+  const percent = Number(quotaData.v5_percent);
+  const hasV5Quota = Number.isFinite(percent);
+  const isExhausted = Boolean(quotaData.v5_is_negative) || (hasV5Quota && percent <= 0);
+  const displayedPercent = hasV5Quota ? Math.max(0, percent) : null;
+  const barPercent = displayedPercent === null ? 0 : Math.min(100, displayedPercent);
+
+  const summary = document.createElement("div");
+  summary.className = "quota-summary";
+  const title = document.createElement("span");
+  title.textContent = hasV5Quota ? "V5 사용 한도" : "이미지 Anlas";
+  const value = document.createElement("strong");
+  value.textContent = hasV5Quota
+    ? (isExhausted ? "소진" : `${displayedPercent.toFixed(displayedPercent % 1 ? 1 : 0)}%`)
+    : `${Number(quotaData.total_anlas || 0).toLocaleString()}`;
+  summary.append(title, value);
+  root.appendChild(summary);
+
+  if (hasV5Quota) {
+    const bar = document.createElement("div");
+    bar.className = "quota-bar";
+    const fill = document.createElement("span");
+    fill.style.width = `${barPercent}%`;
+    bar.appendChild(fill);
+    root.appendChild(bar);
+  }
+
+  const details = document.createElement("div");
+  details.className = "quota-detail";
+  const tier = tierNames[Number(quotaData.tier)] || "NovelAI";
+  const anlas = `Anlas ${Number(quotaData.total_anlas || 0).toLocaleString()}`;
+  const recovery = formatQuotaDuration(quotaData.v5_next_percent_seconds);
+  details.textContent = hasV5Quota
+    ? `${tier} · ${anlas}${recovery ? ` · 다음 +1% ${recovery}` : ""}`
+    : `${tier} · ${anlas}`;
+  root.appendChild(details);
+  const updated = document.createElement("div");
+  updated.className = "quota-updated";
+  updated.textContent = "방금 갱신됨";
+  root.appendChild(updated);
+  root.classList.toggle("exhausted", isExhausted);
+}
+
+async function refreshQuota({ silent = false } = {}) {
+  if (quotaRefreshInFlight) return;
+  quotaRefreshInFlight = true;
+  const button = $("#quotaRefreshButton");
+  if (button) button.disabled = true;
+  try {
+    const data = await request("/api/quota");
+    quotaData = data.quota || null;
+    renderQuota();
+    if (!silent && quotaData?.available) showToast("NAI 할당량을 갱신했습니다.", "success");
+  } catch (error) {
+    quotaData = { available: false, reason: "할당량 정보를 불러오지 못했습니다." };
+    renderQuota();
+    if (!silent) showToast(`할당량을 갱신하지 못했습니다: ${error.message}`, "error");
+  } finally {
+    quotaRefreshInFlight = false;
+    if (button) button.disabled = false;
+  }
+}
+
+function startQuotaPolling() {
+  if (quotaPollingStarted) return;
+  quotaPollingStarted = true;
+  setInterval(() => refreshQuota({ silent: true }), 60_000);
+}
+
 function showToast(message, tone = "info") {
   const region = $("#toastRegion");
   if (!region) return;
@@ -208,7 +301,10 @@ async function saveNow() {
       renderGenerate();
     }
     setSaveState(hadPendingToken ? "API 토큰 변경 저장됨" : "자동 저장됨", "success");
-    if (hadPendingToken) showToast("새 API 토큰으로 교체했습니다.", "success");
+    if (hadPendingToken) {
+      showToast("새 API 토큰으로 교체했습니다.", "success");
+      await refreshQuota({ silent: true });
+    }
   } catch (error) {
     setSaveState("저장 실패", "error");
     showToast(`저장하지 못했습니다: ${error.message}`, "error");
@@ -2632,6 +2728,8 @@ async function loadState() {
   switchTab(currentTab);
   updateJobActionButtons();
   await previewPrompt();
+  await refreshQuota({ silent: true });
+  startQuotaPolling();
 }
 
 function bindEvents() {
@@ -2641,6 +2739,7 @@ function bindEvents() {
     await loadState();
     showToast("저장된 내용을 다시 불러왔습니다.", "success");
   };
+  $("#quotaRefreshButton").onclick = () => refreshQuota();
   $("#previewButton").onclick = previewPrompt;
   $("#generateButtonInline").onclick = startGeneration;
   $("#stopGenerateButton").onclick = stopGeneration;

package/web/styles.css

@@ -117,6 +117,96 @@ button {
   gap: 8px;
 }
 
+.quota-widget {
+  border: 1px solid rgba(140, 182, 255, 0.28);
+  border-radius: 8px;
+  background: rgba(24, 30, 38, 0.84);
+  overflow: hidden;
+}
+
+.quota-widget-header {
+  display: flex;
+  align-items: center;
+  justify-content: space-between;
+  gap: 8px;
+  padding: 10px 10px 7px 12px;
+  color: #dce8ff;
+  font-size: 12px;
+  font-weight: 750;
+}
+
+.quota-refresh {
+  width: 24px;
+  height: 24px;
+  padding: 0;
+  border: 1px solid var(--line);
+  border-radius: 6px;
+  background: var(--surface-2);
+  color: var(--muted);
+  line-height: 1;
+}
+
+.quota-refresh:hover:not(:disabled) {
+  color: var(--text);
+  border-color: rgba(140, 182, 255, 0.58);
+}
+
+.quota-refresh:disabled {
+  cursor: wait;
+  opacity: 0.62;
+}
+
+.quota-widget-content {
+  display: grid;
+  gap: 8px;
+  padding: 0 12px 11px;
+  color: var(--muted);
+  font-size: 12px;
+  line-height: 1.45;
+}
+
+.quota-summary {
+  display: flex;
+  justify-content: space-between;
+  gap: 8px;
+  color: var(--text);
+}
+
+.quota-summary strong {
+  color: var(--accent-2);
+  font-size: 15px;
+}
+
+.quota-bar {
+  height: 6px;
+  overflow: hidden;
+  border-radius: 999px;
+  background: rgba(140, 182, 255, 0.15);
+}
+
+.quota-bar > span {
+  display: block;
+  width: 0;
+  height: 100%;
+  border-radius: inherit;
+  background: linear-gradient(90deg, var(--accent), var(--accent-2));
+  transition: width 420ms ease;
+}
+
+.quota-widget-content.exhausted .quota-bar > span {
+  background: var(--danger);
+}
+
+.quota-detail,
+.quota-updated {
+  color: var(--muted);
+  font-size: 11px;
+}
+
+.quota-updated {
+  color: #718092;
+}
+
 .settings-nav-item {
   border-color: rgba(94, 224, 184, 0.22);
   background: rgba(94, 224, 184, 0.06);

package/web_app.py

@@ -565,6 +565,13 @@ class Handler(BaseHTTPRequestHandler):
             state = load_state()
             prompt, negative, artists = build_prompt(state)
             self.send_json({"prompt": prompt, "negative": negative, "artists": artists})
+        elif route == "/api/quota":
+            if not self.authorized_api_request():
+                self.send_error(403)
+                return
+            state = load_state()
+            state.api.token = load_api_token()
+            self.send_json({"quota": NovelAIClient(state.api).subscription_quota()})
         elif route == "/api/job":
             if not self.authorized_api_request():
                 self.send_error(403)