Update app.py

app.py

@@ -16,7 +16,6 @@ def get_current_time():
     now = datetime.now()
     now2 = now
     current_time = now2.strftime("%y-%m-%d %H:%M:%S")
-    # ki = f'{kii} {current_time}'
     return current_time
 
 def load_fn(models):
@@ -37,10 +36,9 @@ load_fn(models)
 
 num_models = 6
 max_images = 6
-inference_timeout = 300
+inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
-loopcounter = 1
 
 
 def extend_choices(choices):
@@ -66,10 +64,15 @@ 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
-    task = asyncio.create_task(asyncio.to_thread(models_load[model_str].fn,
-                               prompt=prompt, negative_prompt=nprompt, **kwargs, token=HF_TOKEN))
+        
+    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:
         result = await asyncio.wait_for(task, timeout=timeout)
@@ -87,9 +90,9 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     if task.done() and result is not None and not isinstance(result, tuple):
         with lock:
             # png_path = "img.png"
-            global loopcounter 
-            png_path = get_current_time() + "_" + str(loopcounter) + ".png"
-            loopcounter += 1
+            # 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() + "_" + str(theSeed) + ".png"
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
         return image
     return None
@@ -127,7 +130,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,13 +137,29 @@ function refresh() {
 }
 """
 
+js_AutoSave="""
+        console.log("Yo");
+        
+        var img1 = document.querySelector("div#component-355 .svelte-1kpcxni button.svelte-1kpcxni .svelte-1kpcxni img"),
+        observer = new MutationObserver((changes) => {
+            changes.forEach(change => {
+                    if(change.attributeName.includes('src')){
+                        console.log(img1.src);
+                        document.querySelector("div#component-355 .svelte-1kpcxni .svelte-sr71km a.svelte-1s8vnbx button").click();
+                    }
+            });
+        });
+        observer.observe(img1, {attributes : true});
+        
+"""
+
 with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
 # with gr.Blocks(theme='JohnSmith9982/small_and_pretty', fill_width=True, css=CSS, js=js_func) as demo:
     gr.HTML("")  
     with gr.Tab('6 Models'):
         with gr.Column(scale=2):
             with gr.Group():
-                txt_input = gr.Textbox(label='Your prompt:', value=preSetPrompt, lines=2, autofocus=1)
+                txt_input = gr.Textbox(label='Your prompt:', value=preSetPrompt, lines=3, autofocus=1)
                 neg_input = gr.Textbox(label='Negative prompt:', value=negPreSetPrompt, lines=1)
                 with gr.Accordion("Advanced", open=False, visible=True):
                     with gr.Row():
@@ -193,7 +211,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
             with gr.Group():
                 # global preSetPrompt
                 # global negPreSetPrompt
-                txt_input2 = gr.Textbox(label='Your prompt:', value = preSetPrompt, lines=1, autofocus=1)
+                txt_input2 = gr.Textbox(label='Your prompt:', value = preSetPrompt, lines=3, autofocus=1)
                 neg_input2 = gr.Textbox(label='Negative prompt:', value=negPreSetPrompt, lines=1)
                 with gr.Accordion("Advanced", open=False, visible=True):
                     with gr.Row():
@@ -203,11 +221,11 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                         steps2 = gr.Slider(label="Number of inference steps", info="If 0, the default value is used.", maximum=100, step=1, value=0)
                         cfg2 = gr.Slider(label="Guidance scale", info="If 0, the default value is used.", maximum=30.0, step=0.1, value=0)
                         seed2 = gr.Slider(label="Seed", info="Randomize Seed if -1.", minimum=-1, maximum=MAX_SEED, step=1, value=-1)
-                        seed_rand2 = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
+                        seed_rand2 = gr.Button("Randomize Seed", size="sm", variant="secondary")
                         seed_rand2.click(randomize_seed, None, [seed2], queue=False)
             num_images = gr.Slider(1, max_images, value=max_images, step=1, label='Number of images')
             with gr.Row():
-                gen_button2 = gr.Button('Generate', variant='primary', scale=2)
+                gen_button2 = gr.Button('Let the machine halucinate', variant='primary', scale=2)
                 #stop_button2 = gr.Button('Stop', variant='stop', interactive=False, scale=1)
                 #gen_button2.click(lambda: gr.update(interactive=True), None, stop_button2)
 
@@ -233,8 +251,10 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                                        concurrency_limit=None, queue=False)  # Be sure to delete ", queue=False" when activating the stop button
             o.change(add_gallery, [o, model_choice2, gallery2], [gallery2])
             #stop_button2.click(lambda: gr.update(interactive=False), None, stop_button2, cancels=[gen_event2])
-
+    
+    # gr.Markdown(js_AutoSave)
     gr.Markdown("")
 
-#demo.queue(default_concurrency_limit=200, max_size=200)
+# 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)