Update app.py

app.py

@@ -6,21 +6,16 @@ import os
 from threading import RLock
 from datetime import datetime
 
-preSetPrompt = "cute tall slender athletic 20+ caucasian woman. gorgeous face. perky tits. sensual expression. lifting shirt. 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"
-
-# H. R. Giger prompt:
-# preSetPrompt = "a tall slender athletic caucasian nude 18+ female cyborg. gorgeous face. perky tits. wet skin. sensual expression. she is entangled in rusty chains, rusty barbed wire and electric cables. old dark dusty decaying spaceship designed by h.r. giger. rusty metal dildos. wet tubes and wet plastic hoses. dark, gloomy teal cinematic light. photorealistic."
-
+preSetPrompt = "tall slender athletic 18+ caucasian woman. gorgeous face. perky tits. short wet messy hair. open jeans. torn shirt. challenging. explicit. artistic. photorealistic. 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")
+    current_time = now.strftime("%y-%m-%d %H:%M:%S")
     return current_time
 
 def load_fn(models):
@@ -38,9 +33,8 @@ def load_fn(models):
 
 load_fn(models)
 
-
-num_models = 6
-max_images = 6
+num_models = 12
+max_images = 12
 inference_timeout = 400
 default_models = models[:num_models]
 MAX_SEED = 2**32-1
@@ -61,21 +55,18 @@ def random_choices():
     return random.choices(models, k=num_models)
 
 
-# https://huggingface.co/docs/api-inference/detailed_parameters
-# https://huggingface.co/docs/huggingface_hub/package_reference/inference_client
 async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1, timeout=inference_timeout):
     kwargs = {}
     if height > 0: kwargs["height"] = height
     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:
         theSeed = randomize_seed()
-        kwargs["seed"] = theSeed
     else: 
-        kwargs["seed"] = seed
         theSeed = seed
+    kwargs["seed"] = theSeed
         
     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)
@@ -83,26 +74,23 @@ async def infer(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0
         result = await asyncio.wait_for(task, timeout=timeout)
     except asyncio.TimeoutError as e:
         print(e)
-        print(f"Task timed out: {model_str}")
+        print(f"infer: Task timed out: {model_str}")
         if not task.done(): task.cancel()
         result = None
         raise Exception(f"Task timed out: {model_str}") from e
     except Exception as e:
         print(e)
+        print(f"infer: exception: {model_str}")
         if not task.done(): task.cancel()
         result = None
         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 = 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)
+            image = save_image(result, png_path, model_str, prompt, nprompt, height, width, steps, cfg, theSeed)
         return image
     return None
 
-
 def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, seed=-1):
     try:
         loop = asyncio.new_event_loop()
@@ -110,7 +98,7 @@ def gen_fn(model_str, prompt, nprompt="", height=0, width=0, steps=0, cfg=0, see
                                          height, width, steps, cfg, seed, inference_timeout))
     except (Exception, asyncio.CancelledError) as e:
         print(e)
-        print(f"Task aborted: {model_str}")
+        print(f"gen_fn: Task aborted: {model_str}")
         result = None
         raise gr.Error(f"Task aborted: {model_str}, Error: {e}")
     finally:
@@ -124,44 +112,88 @@ def add_gallery(image, model_str, gallery):
         if image is not None: gallery.insert(0, (image, model_str))
     return gallery
 
+JS="""
+<script>
 
-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=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;
+/*
+function simulateButtonPress_() {
+    const button = document.getElementById('simulate-button');
+    if (button) {
+        button.click();  // Simulate the button press
+        console.log('Button Pressed!');
     }
 }
-"""
+*/
 
-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();
-                    }
-            });
+function simulateButtonPress() {
+    console.log('Button Pressed!');
+}
+
+// Function to observe image changes
+function observeImageChanges() {
+    // Select all images with the 'image-monitor' class
+    const images = document.querySelectorAll('.svelte-1pijsyv');
+    
+    // Create a MutationObserver to watch for changes in the image src
+    const observer = new MutationObserver((mutationsList, observer) => {
+        mutationsList.forEach(mutation => {
+            if (mutation.type === 'attributes' && mutation.attributeName === 'src') {
+                // If the image src changes, simulate button press
+                console.log('Image changed!');
+                simulateButtonPress();
+            }
         });
-        observer.observe(img1, {attributes : true});
-        
+    });
+    
+    // Observer options: observe changes to attributes (like src)
+    const config = { attributes: true };
+
+    // Start observing each image
+    images.forEach(image => {
+        observer.observe(image, config);
+    });
+}
+
+// Start observing
+
+
+window.addEventListener('load', () => {
+    observeImageChanges();
+    console.log("Yo");
+});
+    
+</script>
+"""
+
+CSS="""
+<style>
+    .image-monitor {
+        border:1px solid red;
+    }
+
+    /*
+    .svelte-1pijsyv{
+        border:1px solid green;
+    }
+    */
+    
+    .gallery-container{
+        max-height: 512px;
+    }
+
+    .butt{
+        background-color:#2b4764 !important
+    }
+    .butt:hover{
+        background-color:#3a6c9f !important;
+    }
+    
+</style>
 """
 
-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.Blocks(fill_width=True, head=js) as demo:
+with gr.Blocks(head=CSS + JS) as demo:
+    with gr.Tab(str(num_models) + ' Models'):
         with gr.Column(scale=2):
             with gr.Group():
                 txt_input = gr.Textbox(label='Your prompt:', value=preSetPrompt, lines=3, autofocus=1)
@@ -177,31 +209,27 @@ 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', variant='primary', scale=3)
+                gen_button = gr.Button(f'Generate up to {int(num_models)} images', variant='primary', scale=3, elem_classes=["butt"])
                 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("", elem_classes="guide")
 
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    output = [gr.Image(label=m, show_download_button=True, elem_classes="output",
+                    output = [gr.Image(label=m, show_download_button=True, elem_classes=["image-monitor"],
                               interactive=False, width=112, height=112, show_share_button=False, format="png",
                               visible=True) for m in default_models]
                     current_models = [gr.Textbox(m, visible=False) for m in default_models]
 
         with gr.Column(scale=2):
-            gallery = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+            gallery = gr.Gallery(label="Output", show_download_button=True,
                                 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):
             gen_event = gr.on(triggers=[gen_button.click, txt_input.submit], fn=gen_fn,
                               inputs=[m, txt_input, neg_input, height, width, steps, cfg, seed], outputs=[o],
-                              concurrency_limit=None, queue=False) # Be sure to delete ", queue=False" when activating the stop button
+                              concurrency_limit=None, queue=False)
             o.change(add_gallery, [o, m, gallery], [gallery])
-            #stop_button.click(lambda: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
 
         with gr.Column(scale=4):
             with gr.Accordion('Model selection'):
@@ -214,8 +242,6 @@ 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():
-                # 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):
@@ -230,19 +256,17 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                         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('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)
+                gen_button2 = gr.Button('Let the machine halucinate', variant='primary', scale=2, elem_classes=["butt"])
 
         with gr.Column(scale=1):
             with gr.Group():
                 with gr.Row():
-                    output2 = [gr.Image(label='', show_download_button=True, elem_classes="output",
+                    output2 = [gr.Image(label='', show_download_button=True,
                                interactive=False, width=112, height=112, visible=True, format="png",
                                show_share_button=False, show_label=False) for _ in range(max_images)]
 
         with gr.Column(scale=2):
-            gallery2 = gr.Gallery(label="Output", show_download_button=True, elem_classes="gallery",
+            gallery2 = gr.Gallery(label="Output", show_download_button=True,
                                 interactive=False, show_share_button=True, container=True, format="png",
                                 preview=True, object_fit="cover", columns=2, rows=2) 
 
@@ -253,13 +277,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                                fn=lambda i, n, m, t1, t2, n1, n2, n3, n4, n5: gen_fn(m, t1, t2, n1, n2, n3, n4, n5) if (i < n) else None,
                                inputs=[img_i, num_images, model_choice2, txt_input2, neg_input2,
                                        height2, width2, steps2, cfg2, seed2], outputs=[o],
-                                       concurrency_limit=None, queue=False)  # Be sure to delete ", queue=False" when activating the stop button
+                                       concurrency_limit=None, queue=False)
             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.launch(show_api=False, max_threads=400)
-# demo.launch(show_api=False, max_threads=400, js=js_AutoSave)