Update app.py

app.py

@@ -4,9 +4,21 @@ from externalmod import gr_Interface_load, save_image, randomize_seed
 import asyncio
 import os
 from threading import RLock
+from datetime import datetime
+
+
+preSetPrompt = "cute tall slender athletic 20+ caucasian woman. gorgeous face. perky tits. sly smile. lifting shirt. explicit pose. artistic. photorealistic. cinematic. f1.4"
+# preSetPrompt = "cute tall slender athletic 20+ nude caucasian woman. gorgeous face. perky tits. gaping outie pussy. pussy juice. sly smile. explicit pose. artistic. photorealistic. cinematic. f1.4"
+negPreSetPrompt = "[deformed | disfigured], poorly drawn, [bad : wrong] anatomy, [extra | missing | floating | disconnected] limb, (mutated hands and fingers), blurry, text, fuzziness"
+
 lock = RLock()
 HF_TOKEN = os.environ.get("HF_TOKEN") if os.environ.get("HF_TOKEN") else None # If private or gated models aren't used, ENV setting is unnecessary.
 
+def get_current_time():
+    now = datetime.now()
+    now2 = now
+    current_time = now2.strftime("%y-%m-%d %H:%M:%S")
+    return current_time
 
 def load_fn(models):
     global models_load
@@ -26,7 +38,7 @@ 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
 
@@ -56,8 +68,7 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
     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))
+    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)
@@ -74,7 +85,10 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         raise Exception() from e
     if task.done() and result is not None and not isinstance(result, tuple):
         with lock:
-            png_path = "img.png"
+            # 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"
             image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, seed)
         return image
     return None
@@ -105,19 +119,46 @@ def add_gallery(image, model_str, gallery):
 CSS="""
 .gradio-container { max-width: 1200px; margin: 0 auto; !important; }
 .output { width=112px; height=112px; max_width=112px; max_height=112px; !important; }
-.gallery { min_width=512px; min_height=512px; max_height=1024px; !important; }
+.gallery { min_width=512px; min_height=512px; max_height=512px; !important; }
 .guide { text-align: center; !important; }
 """
 
+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;
+    }
+}
+"""
+
+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:
-    gr.HTML(
-)  
-    with gr.Tab('Pr0n Diffusion'):
+# 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:', lines=4)
-                neg_input = gr.Textbox(label='Negative prompt:', lines=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():
                         width = gr.Slider(label="Width", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
@@ -129,11 +170,11 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                         seed_rand = gr.Button("Randomize Seed 🎲", size="sm", variant="secondary")
                         seed_rand.click(randomize_seed, None, [seed], queue=False)
             with gr.Row():
-                gen_button = gr.Button(f'Generate up to {int(num_models)} images in up to 3 minutes total', variant='primary', scale=3)
-                random_button = gr.Button(f'Random {int(num_models)} 🎲', variant='secondary', scale=1)
+                gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3)
+                random_button = gr.Button(f'Randomize Models', variant='secondary', scale=1)
                 #stop_button = gr.Button('Stop', variant='stop', interactive=False, scale=1)
                 #gen_button.click(lambda: gr.update(interactive=True), None, stop_button)
-            gr.Markdown("")
+            gr.Markdown("", elem_classes="guide")
 
         with gr.Column(scale=1):
             with gr.Group():
@@ -145,7 +186,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
 
         with gr.Column(scale=2):
             gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
-                                interactive=False, show_share_button=True, container=True, format="png",
+                                interactive=False, show_share_button=False, container=True, format="png",
                                 preview=True, object_fit="cover", columns=2, rows=2) 
 
         for m, o in zip(current_models, output):
@@ -166,8 +207,10 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
         with gr.Column(scale=2):
             model_choice2 = gr.Dropdown(models, label='Choose model', value=models[0])
             with gr.Group():
-                txt_input2 = gr.Textbox(label='Your prompt:', lines=4)
-                neg_input2 = gr.Textbox(label='Negative prompt:', lines=1)
+                # global preSetPrompt
+                # global negPreSetPrompt
+                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():
                         width2 = gr.Slider(label="Width", info="If 0, the default value is used.", maximum=1216, step=32, value=0)
@@ -176,11 +219,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)
 
@@ -206,8 +249,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)