From 2f9bdb122f626f18b50e084b5ee3988f4d89298d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sebastian=20Kr=C3=BCger?= <valknar@pivoine.art>
Date: Thu, 27 Nov 2025 15:29:16 +0100
Subject: [PATCH] fix: gpu header display

---
 src/ui/app.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/src/ui/app.py b/src/ui/app.py
index 3c573c2..ec1925f 100644
--- a/src/ui/app.py
+++ b/src/ui/app.py
@@ -283,6 +283,15 @@ class AudioCraftApp:
                 "settings": settings,
             }
 
+            # Update header GPU status on page load
+            def update_header_gpu():
+                status = self._get_gpu_status()
+                used = status.get('used_gb', 0)
+                total = status.get('total_gb', 24)
+                return f"GPU: {used:.1f}/{total:.1f} GB"
+
+            app.load(fn=update_header_gpu, outputs=[gpu_status])
+
         self.app = app
         return app