Update app.py

app.py

@@ -66,8 +66,14 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if width > 0: kwargs["width"] = width
     if steps > 0: kwargs["num_inference_steps"] = steps
     if cfg > 0: cfg = kwargs["guidance_scale"] = cfg
-    if seed == -1: kwargs["seed"] = randomize_seed()
-    else: kwargs["seed"] = seed
+        
+    if seed == -1:
+        theSeed = randomize_seed()
+        kwargs["seed"] = theSeed
+    else: 
+        kwargs["seed"] = seed
+        theSeed = seed
+        
     task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn, prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
     await asyncio.sleep(0)
     try:
@@ -88,7 +94,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
             # png_path = "img.png"
             # png_path = get_current_time() + "_" + model_str.replace("/", "_") + ".png"
             # png_path =  model_str.replace("/", "_") + " - " +  prompt + " - " + get_current_time() + ".png"
-            png_path =  model_str.replace("/", "_") + " - " + get_current_time() + ".png"
+            png_path =  model_str.replace("/", "_") + " - " + get_current_time() + "_" + str(theSeed) + ".png"
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
         return image
     return None
@@ -126,7 +132,6 @@ CSS="""
 js_func = """
 function refresh() {
     const url = new URL(window.location);
-
     if (url.searchParams.get('__theme') !== 'dark') {
         url.searchParams.set('__theme', 'dark');
         window.location.href = url.href;
@@ -135,7 +140,6 @@ function refresh() {
 """
 
 js_AutoSave="""
-
         console.log("Yo");
         
         var img1 = document.querySelector("div#component-355 .svelte-1kpcxni button.svelte-1kpcxni .svelte-1kpcxni img"),
@@ -255,4 +259,4 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
 
 # demo.queue(default_concurrency_limit=200, max_size=200)
 demo.launch(show_api=False, max_threads=400)
-# demo.launch(show_api=False, max_threads=400, js=js_AutoSave)
+# demo.launch(show_api=False, max_threads=400, js=js_AutoSave)