Merge branch 'feature-refresh-temp-files' into main

Added a "fake" settings option that refreshes some internal temporary files without needing to restart

README.md

@@ -15,6 +15,7 @@ You can install it using the inbuilt available extensions list, clone the files
 
 ## Common Problems & Known Issues:
 - Depending on your browser settings, sometimes an old version of the script can get cached. Try `CTRL+F5` to force-reload the site without cache if e.g. a new feature doesn't appear for you after an update.
+- If the prompt popup has broken styling for you or doesn't appear at all (like [this](https://github.com/DominikDoom/a1111-sd-webui-tagcomplete/assets/34448969/7bbfdd54-fc23-4bfc-85af-24704b139b3a)), make sure to update your openpose-editor extension if you have it installed. It is known to cause issues with other extensions in older versions.
 
 ## Screenshots & Demo videos
 <details>

javascript/_textAreas.js

@@ -8,7 +8,8 @@ const core = [
     "#txt2img_prompt > label > textarea",
     "#img2img_prompt > label > textarea",
     "#txt2img_neg_prompt > label > textarea",
-    "#img2img_neg_prompt > label > textarea"
+    "#img2img_neg_prompt > label > textarea",
+    ".prompt > label > textarea"
 ];
 
 // Third party text area selectors

javascript/tagAutocomplete.js

@@ -958,6 +958,21 @@ function navigateInList(textArea, event) {
     event.stopPropagation();
 }
 
+async function refreshTacTempFiles() {
+    setTimeout(async () => {
+        wildcardFiles = [];
+        wildcardExtFiles = [];
+        yamlWildcards = [];
+        embeddings = [];
+        hypernetworks = [];
+        loras = [];
+        lycos = [];
+        await processQueue(QUEUE_FILE_LOAD, null);
+
+        console.log("TAC: Refreshed temp files");
+    }, 2000);
+}
+
 function addAutocompleteToArea(area) {
     // Return if autocomplete is disabled for the current area type in config
     let textAreaId = getTextAreaIdentifier(area);
@@ -1033,6 +1048,8 @@ async function setup() {
             }, 500);
         })
     });
+    // Listener for internal temp files refresh button
+    gradioApp().querySelector("#refresh_tac_refreshTempFiles")?.addEventListener("click", refreshTacTempFiles);
 
     // Add mutation observer for the model hash text to also allow hash-based blacklist again
     let modelHashText = gradioApp().querySelector("#sd_checkpoint_hash");

scripts/tag_autocomplete_helper.py

@@ -241,41 +241,49 @@ write_to_temp_file('lyco.txt', [])
 if not TEMP_PATH.joinpath("emb.txt").exists():
     write_to_temp_file('emb.txt', [])
 
-# Write wildcards to wc.txt if found
-if WILDCARD_PATH.exists():
-    wildcards = [WILDCARD_PATH.relative_to(FILE_DIR).as_posix()] + get_wildcards()
-    if wildcards:
-        write_to_temp_file('wc.txt', wildcards)
-
-# Write extension wildcards to wce.txt if found
-if WILDCARD_EXT_PATHS is not None:
-    wildcards_ext = get_ext_wildcards()
-    if wildcards_ext:
-        write_to_temp_file('wce.txt', wildcards_ext)
-    # Write yaml extension wildcards to wcet.txt if found
-    wildcards_yaml_ext = get_ext_wildcard_tags()
-    if wildcards_yaml_ext:
-        write_to_temp_file('wcet.txt', wildcards_yaml_ext)
-
 # Write embeddings to emb.txt if found
 if EMB_PATH.exists():
     # Get embeddings after the model loaded callback
     script_callbacks.on_model_loaded(get_embeddings)
 
-if HYP_PATH.exists():
-    hypernets = get_hypernetworks()
-    if hypernets:
-        write_to_temp_file('hyp.txt', hypernets)
+def refresh_temp_files():
+    write_temp_files()
+    get_embeddings(shared.sd_model)
 
-if LORA_PATH is not None and LORA_PATH.exists():
-    lora = get_lora()
-    if lora:
-        write_to_temp_file('lora.txt', lora)
+def write_temp_files():
+    # Write wildcards to wc.txt if found
+    if WILDCARD_PATH.exists():
+        wildcards = [WILDCARD_PATH.relative_to(FILE_DIR).as_posix()] + get_wildcards()
+        if wildcards:
+            write_to_temp_file('wc.txt', wildcards)
 
-if LYCO_PATH is not None and LYCO_PATH.exists():
-    lyco = get_lyco()
-    if lyco:
-        write_to_temp_file('lyco.txt', lyco)
+    # Write extension wildcards to wce.txt if found
+    if WILDCARD_EXT_PATHS is not None:
+        wildcards_ext = get_ext_wildcards()
+        if wildcards_ext:
+            write_to_temp_file('wce.txt', wildcards_ext)
+        # Write yaml extension wildcards to wcet.txt if found
+        wildcards_yaml_ext = get_ext_wildcard_tags()
+        if wildcards_yaml_ext:
+            write_to_temp_file('wcet.txt', wildcards_yaml_ext)
+
+    if HYP_PATH.exists():
+        hypernets = get_hypernetworks()
+        if hypernets:
+            write_to_temp_file('hyp.txt', hypernets)
+
+    if LORA_PATH is not None and LORA_PATH.exists():
+        lora = get_lora()
+        if lora:
+            write_to_temp_file('lora.txt', lora)
+
+    if LYCO_PATH is not None and LYCO_PATH.exists():
+        lyco = get_lyco()
+        if lyco:
+            write_to_temp_file('lyco.txt', lyco)
+
+
+write_temp_files()
 
 # Register autocomplete options
 def on_ui_settings():
@@ -366,5 +374,6 @@ def on_ui_settings():
         shared.opts.add_option("tac_keymap", shared.OptionInfo(keymapDefault, keymapLabel, gr.Textbox, section=TAC_SECTION))
         shared.opts.add_option("tac_colormap", shared.OptionInfo(colorDefault, colorLabel, gr.Textbox, section=TAC_SECTION))
 
-
+    shared.opts.add_option("tac_refreshTempFiles", shared.OptionInfo("Refresh TAC temp files", "Refresh internal temp files", gr.HTML, {}, refresh=refresh_temp_files, section=TAC_SECTION))
+    
 script_callbacks.on_ui_settings(on_ui_settings)