From 6259fbe09d5a977f760ea78b916ec2ca4735a265 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 18 Feb 2024 03:57:36 -0500
Subject: [PATCH] image gen stats

---
 packages/components/src/api/stats/queues/GetQueueStats.ts  | 1 +
 .../fakeyou.com/src/components/layout/SideNav/SideNav.tsx  | 7 +++++++
 .../src/components/layout/SideNav/SideNavJobs.tsx          | 6 ++++++
 3 files changed, 14 insertions(+)

diff --git a/packages/components/src/api/stats/queues/GetQueueStats.ts b/packages/components/src/api/stats/queues/GetQueueStats.ts
index 839dde3a36..dac2474ddc 100644
--- a/packages/components/src/api/stats/queues/GetQueueStats.ts
+++ b/packages/components/src/api/stats/queues/GetQueueStats.ts
@@ -20,6 +20,7 @@ export interface ByQueueStats {
   pending_svc_jobs: number,
   pending_tacotron2_jobs: number,
   pending_voice_designer: number,
+  pending_stable_diffusion: number,
 }
 
 export interface LegacyTtsInfo {
diff --git a/packages/fakeyou.com/src/components/layout/SideNav/SideNav.tsx b/packages/fakeyou.com/src/components/layout/SideNav/SideNav.tsx
index 6072312522..3a6994acfd 100644
--- a/packages/fakeyou.com/src/components/layout/SideNav/SideNav.tsx
+++ b/packages/fakeyou.com/src/components/layout/SideNav/SideNav.tsx
@@ -146,6 +146,7 @@ export default function SideNav({
         pending_svc_jobs: 0,
         pending_tacotron2_jobs: 0,
         pending_voice_designer: 0,
+        pending_stable_diffusion: 0,
       },
     },
     legacy_tts: {
@@ -566,6 +567,12 @@ export default function SideNav({
                   {queueStats.inference.by_queue.pending_svc_jobs}
                 </span>
               </div>
+              <div>
+                Image Generation:{" "}
+                <span className="text-red">
+                  {queueStats.inference.by_queue.pending_stable_diffusion}
+                </span>
+              </div>
               <div>
                 {t("queueFaceAnimator")}:{" "}
                 <span className="text-red">
diff --git a/packages/fakeyou.com/src/components/layout/SideNav/SideNavJobs.tsx b/packages/fakeyou.com/src/components/layout/SideNav/SideNavJobs.tsx
index f8654c1a76..be454db528 100644
--- a/packages/fakeyou.com/src/components/layout/SideNav/SideNavJobs.tsx
+++ b/packages/fakeyou.com/src/components/layout/SideNav/SideNavJobs.tsx
@@ -26,6 +26,12 @@ export default function SideNavJobs({ byQueue, t, ttsQueuedCount }: Props) {
           { byQueue.pending_svc_jobs }
         </span>
       </div>
+      <div>
+        Image Generation:{" "}
+        <span className="text-red">
+          { byQueue.pending_stable_diffusion }
+        </span>
+      </div>
       <div>
         {t("queueFaceAnimator")}:{" "}
         <span className="text-red">