Upload prompt_selector_auto_1129.py

Danbooru Prompt Selector/prompt_selector_auto_1129.py

@@ -72,8 +72,10 @@ def generate(width, height, positive, negative, button):
     global temp_clipboard_image
     global current_sampler
     global NAI_width, NAI_height
-    scale = entry_CFG_value.get()
-    if type(scale) is not float or type(scale) is not int: scale = 5.0
+    try:
+        scale = float(entry_CFG_value.get())
+    except:
+        scale = 5.0
     if scale > 7.0: scale = 7.0
     elif scale < 4.0: scale = 4.0
     else: scale = round(scale, 1)
@@ -87,7 +89,7 @@ def generate(width, height, positive, negative, button):
         "extra_noise_seed": random.randint(0,9999999999),
         "sampler": current_sampler,
         "steps": 28,
-        "scale": 5.5,
+        "scale": scale,
         "uncond_scale": 1.0,
         "negative_prompt": negative_text.get("1.0", "end-1c").strip(),
         "sm" : sema_button_var.get(),
@@ -156,7 +158,7 @@ def generate(width, height, positive, negative, button):
     button.config(state=tk.NORMAL)
     if mac_var.get():
         random_function()
-        time.sleep(random.uniform(0.1, 3.5))
+        time.sleep(random.uniform(5.1, 8.5))
         button.config(state=tk.DISABLED)
         window.event_generate(GENERATE_EVENT, when="tail")