Upload 5 files

ui/components/generation_panel.py

@@ -8,10 +8,12 @@ import gradio as gr
 from utils.constants import (
     MODEL_CHOICES,
     SAMPLER_CHOICES,
+    NOISE_SCHEDULE_CHOICES,
     DEFAULT_MODEL,
     DEFAULT_STEPS,
     DEFAULT_SCALE,
-    DEFAULT_SAMPLER
+    DEFAULT_SAMPLER,
+    DEFAULT_NOISE_SCHEDULE
 )
 
 
@@ -21,7 +23,7 @@ def create_generation_panel():
 
     Returns:
         Dict with control components:
-        - model, steps, scale, cfg_rescale, sampler, seed
+        - model, steps, scale, cfg_rescale, sampler, noise_schedule, variety_plus, seed
     Note: Resolution is now in the Prompt header, not here
     """
     with gr.Group():
@@ -71,6 +73,21 @@ def create_generation_panel():
                 elem_id="naia-sampler"
             )
 
+        # Noise Schedule and VAR+ row
+        with gr.Row():
+            noise_schedule = gr.Dropdown(
+                choices=NOISE_SCHEDULE_CHOICES,
+                value=DEFAULT_NOISE_SCHEDULE,
+                label="Noise Schedule",
+                elem_id="naia-noise-schedule"
+            )
+
+            variety_plus = gr.Checkbox(
+                value=False,
+                label="VAR+",
+                elem_id="naia-variety-plus"
+            )
+
         # Seed row
         seed = gr.Number(
             value=0,
@@ -87,5 +104,7 @@ def create_generation_panel():
         "scale": scale,
         "cfg_rescale": cfg_rescale,
         "sampler": sampler,
+        "noise_schedule": noise_schedule,
+        "variety_plus": variety_plus,
         "seed": seed
     }

ui/components/output_panel.py

@@ -71,6 +71,13 @@ def create_generation_info_panel():
             elem_id="naia-processed-negative"
         )
 
+        # Useful external links
+        gr.Markdown("""
+* If you need parameters inspection, visit: [novelai.net/inspect](https://novelai.net/inspect)
+* If you need tags wiki, visit: [danbooru.donmai.us/wiki_pages/tag_groups](https://danbooru.donmai.us/wiki_pages/tag_groups)
+* If you need image to danbooru tags, visit: [SmilingWolf/wd-tagger](https://huggingface.co/spaces/SmilingWolf/wd-tagger)
+""", elem_id="naia-gen-info-links")
+
     return {
         "info": output_info,
         "processed_prompt": processed_prompt,

ui/components/quick_search.py

@@ -87,7 +87,7 @@ def create_quick_search():
         recommended_tags = gr.Dataframe(
             headers=["Tag", "Count", "+", "-"],
             datatype=["str", "number", "str", "str"],
-            col_count=(4, "fixed"),
+            column_count=(4, "fixed"),
             row_count=(10, "dynamic"),
             interactive=False,
             wrap=True,

ui/components/settings_panel.py

@@ -178,7 +178,6 @@ LOCALSTORAGE_JS = """
         if (generateStartTime) {
             const elapsed = (Date.now() - generateStartTime) / 1000;
             addResponseTime(elapsed);
-            console.log('NAIA-WEB: Response time:', elapsed.toFixed(2), 's, Average:', getAverageResponseTime().toFixed(2), 's');
         }
 
         if (timerInterval) {
@@ -203,11 +202,13 @@ LOCALSTORAGE_JS = """
             const autoSaveContainer = document.querySelector('#naia-auto-save');
             const autoSaveCheckbox = autoSaveContainer ? autoSaveContainer.querySelector('input[type="checkbox"]') : null;
             if (autoSaveCheckbox && autoSaveCheckbox.checked) {
-                // Small delay to ensure image is fully loaded
-                setTimeout(() => {
+                // Mobile-friendly auto-save: click immediately to maintain user interaction context
+                const link = downloadBtn.querySelector('a[download]');
+                if (link) {
+                    link.click();
+                } else {
                     downloadBtn.click();
-                    console.log('NAIA-WEB: Auto Save triggered download');
-                }, 500);
+                }
             }
         }
     }
@@ -264,7 +265,6 @@ LOCALSTORAGE_JS = """
                 input.value = savedToken;
                 input.dispatchEvent(new Event('input', { bubbles: true }));
                 input.dispatchEvent(new Event('change', { bubbles: true }));
-                console.log('NAIA-WEB: Token loaded from localStorage');
             }
         } catch (e) {
             console.error('NAIA-WEB: Error loading token:', e);
@@ -387,6 +387,8 @@ LOCALSTORAGE_JS = """
             scale: toFloat(getInputValue('#naia-scale')),
             cfg_rescale: toFloat(getInputValue('#naia-cfg-rescale')),
             sampler: getInputValue('#naia-sampler'),
+            noise_schedule: getInputValue('#naia-noise-schedule'),
+            variety_plus: getInputValue('#naia-variety-plus', 'checkbox'),
             // Random resolution
             random_resolution: getInputValue('#naia-random-resolution', 'checkbox'),
             enabled_resolutions: getCheckboxGroupValues('#naia-res-checkboxes'),
@@ -417,10 +419,6 @@ LOCALSTORAGE_JS = """
 
         try {
             localStorage.setItem(SETTINGS_KEY, JSON.stringify(settings));
-            console.log('NAIA-WEB: Settings saved', settings.char_slots ? `(${settings.char_slots.length} char slots)` : '(no char slots)');
-            if (settings.char_slots && settings.char_slots.length > 0) {
-                console.log('NAIA-WEB: Saved char slots:', JSON.stringify(settings.char_slots));
-            }
         } catch (e) {
             console.error('NAIA-WEB: Error saving settings:', e);
         }
@@ -481,7 +479,6 @@ LOCALSTORAGE_JS = """
         renderFromCache() {
             const grid = document.getElementById('naia-qs-tag-grid');
             if (!grid) {
-                console.log('NAIA-WEB QuickSearch: Grid not ready, will render when accordion opens');
                 return;
             }
 
@@ -510,7 +507,6 @@ LOCALSTORAGE_JS = """
             }
 
             this.updatePageInfo();
-            console.log(`NAIA-WEB QuickSearch: Rendered page ${this.currentPage}/${this.totalPages} (${this.currentTags.length} tags)`);
         },
 
         // Show popup to choose Include or Exclude (positioned above grid)
@@ -582,7 +578,6 @@ LOCALSTORAGE_JS = """
                 actionInput.value = `${action}:${tag}`;
                 actionInput.dispatchEvent(new Event('input', { bubbles: true }));
                 actionInput.dispatchEvent(new Event('change', { bubbles: true }));
-                console.log(`NAIA-WEB QuickSearch: Tag action ${action}:${tag}`);
             }
         },
 
@@ -610,16 +605,11 @@ LOCALSTORAGE_JS = """
                 }
 
                 // Add click listener - just try to render after click
-                // renderWithRetry will handle checking if grid exists
                 button.addEventListener('click', () => {
-                    console.log('NAIA-WEB QuickSearch: Accordion clicked');
                     if (self.dataReady) {
-                        // Try to render with retries (accordion might need time to expand)
                         self.renderWithRetry(5);
                     }
                 });
-
-                console.log('NAIA-WEB QuickSearch: Click listener set up on accordion');
             };
 
             checkAccordion();
@@ -688,16 +678,10 @@ LOCALSTORAGE_JS = """
 
         // Initialize autocomplete on all target textareas
         init() {
-            console.log('NAIA-WEB Autocomplete: Starting initialization...');
             this.createDropdown();
-            console.log('NAIA-WEB Autocomplete: Dropdown created');
-            // Debug: show all textareas on page
-            this.debugListTextareas();
             this.attachToTargets();
-            console.log('NAIA-WEB Autocomplete: Attached to targets');
             // Re-attach when new elements appear (for dynamic char slots)
             this.setupMutationObserver();
-            console.log('NAIA-WEB Autocomplete: Initialized successfully');
         },
 
         // Create the dropdown element
@@ -736,9 +720,6 @@ LOCALSTORAGE_JS = """
             return container.querySelector('textarea');
         },
 
-        // Track which IDs we've already reported as not found (to reduce log spam)
-        _notFoundReported: new Set(),
-
         // Attach to all target textareas
         attachToTargets() {
             let attachedCount = 0;
@@ -748,11 +729,6 @@ LOCALSTORAGE_JS = """
                 if (textarea && !textarea.dataset.autocompleteAttached) {
                     this.attachToTextarea(textarea);
                     attachedCount++;
-                    console.log('NAIA-WEB Autocomplete: Attached to #' + containerId);
-                    this._notFoundReported.delete(containerId);
-                } else if (!textarea && !this._notFoundReported.has(containerId)) {
-                    // Only log "not found" once per ID
-                    this._notFoundReported.add(containerId);
                 }
             }
 
@@ -763,12 +739,10 @@ LOCALSTORAGE_JS = """
                 if (promptTextarea && !promptTextarea.dataset.autocompleteAttached) {
                     this.attachToTextarea(promptTextarea);
                     attachedCount++;
-                    console.log('NAIA-WEB Autocomplete: Attached to #naia-char-prompt-' + i);
                 }
                 if (negativeTextarea && !negativeTextarea.dataset.autocompleteAttached) {
                     this.attachToTextarea(negativeTextarea);
                     attachedCount++;
-                    console.log('NAIA-WEB Autocomplete: Attached to #naia-char-negative-' + i);
                 }
             }
 
@@ -777,11 +751,6 @@ LOCALSTORAGE_JS = """
             if (charSearchTextarea && !charSearchTextarea.dataset.autocompleteAttached) {
                 this.attachToTextarea(charSearchTextarea);
                 attachedCount++;
-                console.log('NAIA-WEB Autocomplete: Attached to character search input');
-            }
-
-            if (attachedCount > 0) {
-                console.log('NAIA-WEB Autocomplete: Total attached:', attachedCount);
             }
         },
 
@@ -840,25 +809,12 @@ LOCALSTORAGE_JS = """
             observer.observe(document.body, { childList: true, subtree: true });
         },
 
-        // Debug: List all textareas on page
-        debugListTextareas() {
-            const textareas = document.querySelectorAll('textarea');
-            console.log('NAIA-WEB Autocomplete DEBUG: Found', textareas.length, 'textareas on page:');
-            textareas.forEach((ta, i) => {
-                const parent = ta.closest('[id]');
-                const parentId = parent ? parent.id : 'no-id-parent';
-                console.log(`  [${i}] parent=#${parentId}, placeholder="${ta.placeholder || ''}"`.substring(0, 100));
-            });
-        },
-
         // Handle input event
         onInput(textarea) {
-            console.log('NAIA-WEB Autocomplete: Input event fired');
             this.activeTextarea = textarea;
 
             clearTimeout(this.debounceTimer);
             this.debounceTimer = setTimeout(() => {
-                console.log('NAIA-WEB Autocomplete: Debounce triggered, calling search');
                 this.search(textarea);
             }, this.DEBOUNCE_MS);
         },
@@ -951,12 +907,10 @@ LOCALSTORAGE_JS = """
             }
 
             const tokenInfo = this.getCurrentToken(textarea);
-            const { token, isWeightValue, weightPrefix, weightSuffix } = tokenInfo;
-            console.log('NAIA-WEB Autocomplete: Search called, token:', token, 'weightPrefix:', weightPrefix, 'weightSuffix:', weightSuffix);
+            const { token, isWeightValue } = tokenInfo;
 
             // Skip autocomplete if editing a weight value (e.g., "0.5" in "0.5::tag")
             if (isWeightValue) {
-                console.log('NAIA-WEB Autocomplete: Editing weight value, skipping');
                 this.close();
                 return;
             }
@@ -965,42 +919,33 @@ LOCALSTORAGE_JS = """
             this.currentTokenInfo = tokenInfo;
 
             if (token.length < this.MIN_CHARS) {
-                console.log('NAIA-WEB Autocomplete: Token too short, closing');
                 this.close();
                 return;
             }
 
             // Determine search mode based on textarea context
             this.currentSearchMode = this.getSearchMode(textarea);
-            console.log('NAIA-WEB Autocomplete: Search mode:', this.currentSearchMode);
 
             // Trigger Gradio search via hidden input
             // Format: "mode:query" (e.g., "no_character:blue" or "character_only:miku")
             const searchContainer = document.getElementById('naia-autocomplete-query');
             const searchInput = searchContainer ? searchContainer.querySelector('textarea, input') : null;
-            console.log('NAIA-WEB Autocomplete: Search input element:', searchInput);
             if (searchInput) {
                 const queryWithMode = this.currentSearchMode + ':' + token;
                 searchInput.value = queryWithMode;
                 searchInput.dispatchEvent(new Event('input', { bubbles: true }));
-                console.log('NAIA-WEB Autocomplete: Dispatched search event for:', queryWithMode);
-            } else {
-                console.log('NAIA-WEB Autocomplete: ERROR - Search input not found! Container:', searchContainer);
             }
         },
 
         // Receive results from Gradio (called from JS parameter)
         setResults(results) {
-            console.log('NAIA-WEB Autocomplete: setResults called with:', results);
             if (!results || results.length === 0) {
-                console.log('NAIA-WEB Autocomplete: No results, closing');
                 this.close();
                 return;
             }
 
             this.currentResults = results;
             this.selectedIndex = 0;
-            console.log('NAIA-WEB Autocomplete: Rendering', results.length, 'results');
             this.renderDropdown();
             this.show();
         },
@@ -1116,7 +1061,6 @@ LOCALSTORAGE_JS = """
                     const searchBtn = document.querySelector('#naia-char-search-btn');
                     if (searchBtn) {
                         searchBtn.click();
-                        console.log('NAIA-WEB Autocomplete: Clicked character search button');
                     }
                 }, 100);
             } else {
@@ -1259,7 +1203,6 @@ LOCALSTORAGE_JS = """
 
     // Initialize after DOM ready with retry
     function initAutocompleteWithRetry(attempt = 1) {
-        console.log('NAIA-WEB Autocomplete: Init attempt', attempt);
         Autocomplete.init();
 
         // If no textareas attached and still have retries, try again
@@ -1290,7 +1233,6 @@ LOCALSTORAGE_JS = """
         // Move download button into image container
         if (!imageContainer.contains(downloadBtn)) {
             imageContainer.appendChild(downloadBtn);
-            console.log('NAIA-WEB: Download button moved into image container');
         }
     }