From afa96b1d449c4acf0fb2e31f3abd0dacbd3fc963 Mon Sep 17 00:00:00 2001
From: duffyduck <info@hacker-net.de>
Date: Sat, 16 May 2026 23:25:55 +0200
Subject: [PATCH] feat(flux): HF-Token in Diagnostic statt .env
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Passwort-Feld in der FLUX-Section, mit Show/Hide-Toggle und kurzem
Hinweis-Link zu den HuggingFace-Schritten (Lizenz-Agree + Token-Erzeugung).
Wert wird in voice_config.json persistiert und per config-Broadcast an
die flux-bridge gepusht; dort vor jedem from_pretrained als HF_TOKEN +
HUGGING_FACE_HUB_TOKEN env gesetzt.

HF_TOKEN aus .env.example + docker-compose.yml entfernt. Auch FLUX_MODEL
aus compose raus — Default-Modell kommt jetzt komplett aus Diagnostic.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
---
 bridge/aria_bridge.py   |  4 ++--
 diagnostic/index.html   | 19 ++++++++++++++++++-
 diagnostic/server.js    |  8 ++++++++
 flux/.env.example       | 21 +++++++--------------
 flux/bridge.py          | 15 +++++++++++++--
 flux/docker-compose.yml | 15 ++++++---------
 6 files changed, 54 insertions(+), 28 deletions(-)

diff --git a/bridge/aria_bridge.py b/bridge/aria_bridge.py
index f43e53b..e08833b 100644
--- a/bridge/aria_bridge.py
+++ b/bridge/aria_bridge.py
@@ -505,7 +505,7 @@ class ARIABridge:
                     if k in vc:
                         self._f5tts_config[k] = vc[k]
                 # FLUX-Felder (Default-Modell + Keywords) gleicher Mechanismus
-                for k in ("fluxDefaultModel", "fluxKeywordRaw", "fluxKeywordSwitch"):
+                for k in ("fluxDefaultModel", "fluxKeywordRaw", "fluxKeywordSwitch", "huggingfaceToken"):
                     if k in vc:
                         self._flux_config[k] = vc[k]
                 logger.info("Voice-Config geladen: tts=%s voice=%s f5tts=%s flux=%s",
@@ -1786,7 +1786,7 @@ class ARIABridge:
             # FLUX-Felder: gleiche Logik wie F5-TTS. flux-bridge applied
             # fluxDefaultModel selbst (Pipeline-Swap). Keywords nutzt Brain
             # via /shared/config/voice_config.json.
-            for k in ("fluxDefaultModel", "fluxKeywordRaw", "fluxKeywordSwitch"):
+            for k in ("fluxDefaultModel", "fluxKeywordRaw", "fluxKeywordSwitch", "huggingfaceToken"):
                 if k in payload:
                     if not hasattr(self, "_flux_config"):
                         self._flux_config = {}
diff --git a/diagnostic/index.html b/diagnostic/index.html
index 5078fc0..29580c5 100644
--- a/diagnostic/index.html
+++ b/diagnostic/index.html
@@ -642,6 +642,21 @@
             placeholder="fix"
             style="background:#1E1E2E;color:#fff;border:1px solid #2A2A3E;border-radius:6px;padding:6px 10px;font-size:13px;">
 
+          <label style="color:#8888AA;font-size:12px;margin-top:4px;">
+            HuggingFace-Token (nur fuer FLUX.1-dev — gated Modell, Lizenz-Bestaetigung).
+            Wird per RVS an die flux-bridge gepusht. Leer = kein Token (Schnell-Modell laeuft auch ohne).
+          </label>
+          <div style="display:flex;gap:4px;">
+            <input type="password" id="diag-flux-hf-token"
+              placeholder="hf_..."
+              style="flex:1;min-width:0;box-sizing:border-box;background:#1E1E2E;color:#fff;border:1px solid #2A2A3E;border-radius:6px;padding:6px 10px;font-size:13px;font-family:monospace;">
+            <button type="button" class="btn secondary" onclick="toggleSecret('diag-flux-hf-token', this)" style="padding:4px 10px;flex-shrink:0;" title="Anzeigen/Verbergen">&#128065;</button>
+          </div>
+          <div style="color:#666680;font-size:10px;">
+            Erst auf <a href="https://huggingface.co/black-forest-labs/FLUX.1-dev" target="_blank" style="color:#0096FF;">huggingface.co/.../FLUX.1-dev</a> "Agree" klicken,
+            dann unter <a href="https://huggingface.co/settings/tokens" target="_blank" style="color:#0096FF;">Settings → Tokens</a> einen Read-Token erzeugen.
+          </div>
+
           <div style="display:flex;gap:8px;align-items:center;margin-top:6px;">
             <button class="btn primary" onclick="sendVoiceConfig()" style="padding:6px 14px;font-size:12px;">
               Anwenden
@@ -1388,6 +1403,7 @@
           setIfPresent('diag-flux-default-model', msg.fluxDefaultModel);
           setIfPresent('diag-flux-keyword-raw', msg.fluxKeywordRaw);
           setIfPresent('diag-flux-keyword-switch', msg.fluxKeywordSwitch);
+          setIfPresent('diag-flux-hf-token', msg.huggingfaceToken);
           return;
         }
 
@@ -2706,12 +2722,13 @@
       const fluxDefaultModel = document.getElementById('diag-flux-default-model')?.value || undefined;
       const fluxKeywordRaw = document.getElementById('diag-flux-keyword-raw')?.value;
       const fluxKeywordSwitch = document.getElementById('diag-flux-keyword-switch')?.value;
+      const huggingfaceToken = document.getElementById('diag-flux-hf-token')?.value;
       send({
         action: 'send_voice_config',
         ttsEnabled, xttsVoice, whisperModel,
         f5ttsModel, f5ttsCkptFile, f5ttsVocabFile,
         f5ttsCfgStrength, f5ttsNfeStep,
-        fluxDefaultModel, fluxKeywordRaw, fluxKeywordSwitch,
+        fluxDefaultModel, fluxKeywordRaw, fluxKeywordSwitch, huggingfaceToken,
       });
       const statusEl = document.getElementById('voice-status');
       if (statusEl && xttsVoice) {
diff --git a/diagnostic/server.js b/diagnostic/server.js
index 24d998b..dafdb45 100644
--- a/diagnostic/server.js
+++ b/diagnostic/server.js
@@ -1957,6 +1957,14 @@ wss.on("connection", (ws) => {
         if (msg.fluxKeywordSwitch !== undefined) {
           voiceConfig.fluxKeywordSwitch = String(msg.fluxKeywordSwitch || "").trim().toLowerCase() || "fix";
         }
+        // HuggingFace-Token fuer gated FLUX.1-dev. Wird per RVS an die
+        // flux-bridge gepusht, dort als HF_TOKEN env gesetzt vor dem
+        // naechsten from_pretrained. Leerer String = "kein Token" (statt
+        // 'behalt was du hattest'), damit Stefan ihn auch wieder loeschen
+        // kann.
+        if (msg.huggingfaceToken !== undefined) {
+          voiceConfig.huggingfaceToken = String(msg.huggingfaceToken || "").trim();
+        }
         try {
           fs.mkdirSync("/shared/config", { recursive: true });
           fs.writeFileSync("/shared/config/voice_config.json", JSON.stringify(voiceConfig, null, 2));
diff --git a/flux/.env.example b/flux/.env.example
index 5ad2e63..17bb674 100644
--- a/flux/.env.example
+++ b/flux/.env.example
@@ -10,23 +10,16 @@ RVS_TLS=true
 RVS_TLS_FALLBACK=true
 RVS_TOKEN=dein_token_hier
 
-# HuggingFace-Token — NUR noetig wenn Du FLUX.1-dev nutzen willst.
-# FLUX.1-dev ist auf HuggingFace 'gated' (= Du musst dort auf der
-# Modell-Seite einmalig "Agree" klicken und beim Download wird Dein
-# Login geprueft). Reine Lizenz-Sache — die Inference laeuft danach
-# komplett lokal, kein Netz, keine Telemetrie.
+# HuggingFace-Token + Default-Modell werden in ARIA Diagnostic verwaltet
+# (Section "FLUX Bildgenerierung") und per RVS an die flux-bridge gepusht.
+# Hier nichts noetig.
 #
-# Wenn Du nur FLUX.1-schnell nutzt (Apache-2.0, kein Gate), kannst Du
-# das Feld leer lassen. Falls Du spaeter dev aktivierst:
+# Token-Pflicht NUR fuer FLUX.1-dev (gated). Workflow falls Du dev nutzen
+# willst:
 #   1) https://huggingface.co/black-forest-labs/FLUX.1-dev → "Agree"
 #   2) https://huggingface.co/settings/tokens → "Read"-Token erzeugen
-#   3) Hier eintragen + docker compose restart flux-bridge
-HF_TOKEN=
-
-# Welches Modell beim Startup geladen wird, steuert ARIA selbst:
-# Diagnostic > FLUX > Default-Modell. Solange dort nichts gewaehlt ist,
-# laedt die flux-bridge erstmal nichts — sie wartet auf den ersten
-# Render-Request oder einen Diagnostic-config-Broadcast.
+#   3) Token in Diagnostic > FLUX Bildgenerierung > HuggingFace-Token
+# FLUX.1-schnell (Apache-2.0) laeuft ohne Token.
 
 # Offloading-Strategie (VRAM-Steuerung):
 #   model       — Default. Komponentenweise CPU-Offload, gut fuer 12 GB Karten.
diff --git a/flux/bridge.py b/flux/bridge.py
index 36d2b83..9b2ef94 100644
--- a/flux/bridge.py
+++ b/flux/bridge.py
@@ -509,8 +509,19 @@ async def run_loop(runner: FluxRunner) -> None:
                             await _flux_queue.put(payload)
                         elif mtype == "config":
                             # Diagnostic-Broadcast (oder aria-bridge nach Reconnect).
-                            # Wir interessieren uns nur fuer fluxDefaultModel — die
-                            # Keywords nutzt das Brain, nicht wir.
+                            # HuggingFace-Token MUSS vor dem Modell-Swap gesetzt sein,
+                            # weil FluxPipeline.from_pretrained den Token aus der env
+                            # liest. Reihenfolge im selben Tick gewaehrleistet das.
+                            if "huggingfaceToken" in payload:
+                                tok = (payload.get("huggingfaceToken") or "").strip()
+                                if tok:
+                                    os.environ["HF_TOKEN"] = tok
+                                    os.environ["HUGGING_FACE_HUB_TOKEN"] = tok
+                                    logger.info("[config] HF-Token gesetzt (len=%d)", len(tok))
+                                else:
+                                    os.environ.pop("HF_TOKEN", None)
+                                    os.environ.pop("HUGGING_FACE_HUB_TOKEN", None)
+                                    logger.info("[config] HF-Token entfernt (leerer Wert)")
                             tag = (payload.get("fluxDefaultModel") or "").strip()
                             if tag:
                                 asyncio.create_task(_apply_default_change(tag))
diff --git a/flux/docker-compose.yml b/flux/docker-compose.yml
index c516cfe..690b935 100644
--- a/flux/docker-compose.yml
+++ b/flux/docker-compose.yml
@@ -19,16 +19,13 @@
 
 services:
 
-  # ─── FLUX.1-dev Bildgenerierung (GPU) ─────────
-  # Empfaengt flux_request via RVS, rendert PNG mit FLUX.1-dev (12B Params)
+  # ─── FLUX Bildgenerierung (GPU) ─────────
+  # Empfaengt flux_request via RVS, rendert PNG mit FLUX (12B Params)
   # und broadcastet flux_response mit base64-PNG zurueck. aria-bridge speichert
   # die Datei nach /shared/uploads/ und ARIA referenziert sie via [FILE:]-Marker.
   #
-  # Modell-Wahl per FLUX_MODEL:
-  #   - black-forest-labs/FLUX.1-dev      (Default, 28 Steps, non-commercial)
-  #   - black-forest-labs/FLUX.1-schnell  (4 Steps, Apache-2.0, schneller)
-  # HuggingFace-Token noetig fuer FLUX.1-dev — vorher `huggingface-cli login`
-  # oder HF_TOKEN in .env setzen, sonst 403 beim ersten Download.
+  # Modell-Wahl + HuggingFace-Token werden in ARIA Diagnostic eingestellt
+  # ("FLUX Bildgenerierung") und per RVS gepusht — hier nichts noetig.
   flux-bridge:
     build: .
     container_name: aria-flux-bridge
@@ -45,13 +42,13 @@ services:
       - RVS_TLS=${RVS_TLS:-true}
       - RVS_TLS_FALLBACK=${RVS_TLS_FALLBACK:-true}
       - RVS_TOKEN=${RVS_TOKEN}
-      - FLUX_MODEL=${FLUX_MODEL:-black-forest-labs/FLUX.1-dev}
+      # Hardware-Bootstrap (Diagnostic-Settings uebersteuern alles andere
+      # zur Laufzeit — diese envs sind nur Edge-Case-Fallbacks).
       - FLUX_DEVICE=${FLUX_DEVICE:-cuda}
       - FLUX_DTYPE=${FLUX_DTYPE:-bfloat16}
       - FLUX_OFFLOAD=${FLUX_OFFLOAD:-model}
       - FLUX_MAX_STEPS=${FLUX_MAX_STEPS:-50}
       - FLUX_MAX_DIM=${FLUX_MAX_DIM:-1536}
-      - HF_TOKEN=${HF_TOKEN:-}        # FLUX.1-dev braucht Login-Token
     volumes:
       - ./hf-cache:/root/.cache/huggingface      # Bind-Mount. FLUX.1-dev ~24 GB on disk!
                                                  # Wenn flux auf der gleichen Maschine