Upload app.py with huggingface_hub

app.py

@@ -95,9 +95,10 @@ IDENTITY_NEG = os.getenv(
 # -limb artifacts. Applied on person edits (whenever the identity lock is on).
 ANATOMY_POS = os.getenv(
     "ANATOMY_POS",
-    " Keep the body anatomy natural and correct: exactly two hands each with exactly "
-    "five fingers, two arms and two legs, no extra, missing, duplicated or fused "
-    "fingers, hands or limbs.")
+    " Render the hands perfectly: each hand has exactly five fingers — one thumb and "
+    "four fingers — in correct natural proportions, cleanly separated, no extra or "
+    "missing or fused fingers. Keep body anatomy correct: two hands, two arms, two "
+    "legs, no extra or duplicated limbs.")
 # Extra anatomy negatives (active when Guidance > 1).
 ANATOMY_NEG = os.getenv(
     "ANATOMY_NEG",
@@ -380,6 +381,7 @@ def infer(
     preserve_identity=True,
     identity_strength=65,
     output_size=1280,
+    fix_hands=False,
     progress=gr.Progress(track_tqdm=True),
 ):
     gc.collect()
@@ -421,6 +423,14 @@ def infer(
 
     pipe.set_adapters([adapter_name], adapter_weights=[1.0])
 
+    # "Fix hands": the anti-extra-finger terms live in the NEGATIVE prompt, which is
+    # ignored at Guidance=1 (CFG off on this distilled model). Raising Guidance turns
+    # true-CFG on so those negatives actually suppress extra/fused fingers; a couple
+    # more steps give CFG room to work. Costs ~2x time — hence it's opt-in.
+    if bool(fix_hands):
+        guidance_scale = max(float(guidance_scale), float(os.getenv("FIXHANDS_CFG", "2.5")))
+        steps = max(int(steps), int(os.getenv("FIXHANDS_STEPS", "6")))
+
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
 
@@ -1219,6 +1229,16 @@ function init() {
         });
     }
 
+    const fixHandsCheck = document.getElementById('custom-fixhands');
+    if (fixHandsCheck) {
+        fixHandsCheck.addEventListener('change', () => {
+            const container = document.getElementById('gradio-fixhands');
+            if (!container) return;
+            const cb = container.querySelector('input[type="checkbox"]');
+            if (cb && cb.checked !== fixHandsCheck.checked) cb.click();
+        });
+    }
+
     function showLoader() {
         const l = document.getElementById('output-loader');
         if (l) l.classList.add('active');
@@ -1394,6 +1414,7 @@ with gr.Blocks() as demo:
     preserve_identity = gr.Checkbox(value=True, elem_id="gradio-preserve",      elem_classes="hidden-input", container=False)
     identity_strength = gr.Slider(minimum=0, maximum=100, step=5, value=65,     elem_id="gradio-idstrength", elem_classes="hidden-input", container=False)
     output_size       = gr.Slider(minimum=1024, maximum=2048, step=128, value=1280, elem_id="gradio-outsize", elem_classes="hidden-input", container=False)
+    fix_hands         = gr.Checkbox(value=False, elem_id="gradio-fixhands",      elem_classes="hidden-input", container=False)
     result            = gr.Image(elem_id="gradio-result",                       elem_classes="hidden-input", container=False, format="png")
 
     example_idx      = gr.Textbox(value="", elem_id="example-idx-input",   elem_classes="hidden-input", container=False)
@@ -1561,6 +1582,10 @@ with gr.Blocks() as demo:
                 <input type="checkbox" id="custom-preserve" checked>
                 <label for="custom-preserve">Keep her face / identity (pose &amp; head can still change) &mdash; uncheck for anime/3D styles</label>
               </div>
+              <div class="checkbox-row">
+                <input type="checkbox" id="custom-fixhands">
+                <label for="custom-fixhands">Fix hands (activates anti-extra-finger negatives &mdash; slower, ~2&times;)</label>
+              </div>
               <div class="slider-row">
                 <label>Identity lock</label>
                 <input type="range" id="custom-idstrength" min="0" max="100" step="5" value="65">
@@ -1616,9 +1641,9 @@ with gr.Blocks() as demo:
 
     run_btn.click(
         fn=infer,
-        inputs=[hidden_images_b64, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, preserve_identity, identity_strength, output_size],
+        inputs=[hidden_images_b64, prompt, lora_adapter, seed, randomize_seed, guidance_scale, steps, preserve_identity, identity_strength, output_size, fix_hands],
         outputs=[result, seed],
-        js=r"""(imgs, p, la, s, rs, gs, st, pi, idst, osz) => {
+        js=r"""(imgs, p, la, s, rs, gs, st, pi, idst, osz, fh) => {
             const images    = window.__uploadedImages || [];
             const b64Array  = images.map(img => img.b64);
             const imgsJson  = JSON.stringify(b64Array);
@@ -1627,12 +1652,14 @@ with gr.Blocks() as demo:
             const presEl    = document.getElementById('custom-preserve');
             const idstEl    = document.getElementById('custom-idstrength');
             const oszEl     = document.getElementById('custom-outsize');
+            const fhEl      = document.getElementById('custom-fixhands');
             const promptVal = promptEl ? promptEl.value : p;
             const loraVal   = loraEl   ? loraEl.value   : la;
             const presVal   = presEl   ? presEl.checked : pi;
             const idstVal   = idstEl   ? parseInt(idstEl.value) : idst;
             const oszVal    = oszEl    ? parseInt(oszEl.value)  : osz;
-            return [imgsJson, promptVal, loraVal, s, rs, gs, st, presVal, idstVal, oszVal];
+            const fhVal     = fhEl     ? fhEl.checked : fh;
+            return [imgsJson, promptVal, loraVal, s, rs, gs, st, presVal, idstVal, oszVal, fhVal];
         }""",
     )