File size: 4,279 Bytes
6fef025
89c5f18
f5b8400
53b0019
 
0318f31
ae8b9b8
0edfd0c
ba4f952
2944f54
b5a428c
53b0019
 
0318f31
c469318
0318f31
b21c027
 
f5b8400
 
 
 
 
 
 
 
d6c665c
f5b8400
6fef025
81987e1
6fef025
19aac56
f5b8400
6fef025
f5b8400
 
6fef025
f5b8400
 
0318f31
6fef025
f5b8400
 
 
25a1729
6177a01
6fef025
bdcf524
132d5bb
 
e367d1b
4424a6b
e367d1b
730e2da
 
132d5bb
 
 
 
 
 
 
 
 
8f8f343
2944f54
0b760cf
8f8f343
bb69629
0318f31
8f8f343
0318f31
8f8f343
 
556ec8a
0318f31
 
 
 
 
 
2944f54
 
 
 
 
 
 
 
 
 
6eab85d
47b7d2c
e5beca8
8fc64ff
26b4194
 
0318f31
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import gradio as gr
from random import randint
from all_models import models
from datetime import datetime

now2 = 0

#kii = "femboys crawling, anal oral sex orgy, pulling anus open with fingers, laying on couch, glans in mouth, tiny penis, tiny testicles, long flowing hair, perfect hands , perfect fingers, ass above face, holding legs back, squirting dairy cream from anus, peeing from glans, peeing in mouth, "
#kii = "femboys, mohawk, apocolytic desert, comic illustration, anal oral sex orgy, pulling anus open with fingers, bending over hood car,  tiny penis, tiny testicles, perfect hands , perfect fingers,  peeing from glans, peeing in mouth, "
kii = "femboys, mohawk, apocolytic desert, comic illustration, anal oral sex orgy, pulling anus open with fingers, bending over hood car,  tiny penis, tiny testicles, perfect hands , perfect fingers,  peeing from glans, peeing in mouth, "

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 ki

def load_fn(models):
    global models_load
    models_load = {}
    for model in models:
        if model not in models_load.keys():
            try:
                m = gr.load(f'models/{model}')
            except Exception as error:
                m = gr.Interface(lambda txt: None, ['text'], ['image'])
            models_load.update({model: m})

load_fn(models)

num_models = len(models)
default_models = models[:num_models]

def extend_choices(choices):
    return choices + (num_models - len(choices)) * ['NA']

def update_imgbox(choices):
    choices_plus = extend_choices(choices)
    return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]

def gen_fn(model_str, prompt):
    if model_str == 'NA':
        return None
    noise = str(randint(0, 9999))
    return models_load[model_str](f'{prompt} {noise}')

def make_me():
   # with gr.Tab('The Dream'): 
        with gr.Row():
            txt_input = gr.Textbox(lines=2, value=kii, width=300)
            #txt_input = gr.Textbox(label='Your prompt:', lines=2, value=kii)
        
            gen_button = gr.Button('Generate images', width=150, height=30)
            stop_button = gr.Button('Stop', variant='secondary', interactive=False, width=150, height=30)
            gen_button.click(lambda s: gr.update(interactive=True), None, stop_button)
        #gr.HTML("""
        #    <div style="text-align: center; max-width: 100%; margin: 0 auto;">
        #      <!--div>
        #        <body>
        #        </body>
        #      </div-->
        #    </div>
        #""")
        with gr.Row():
            output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]
            current_models = [gr.Textbox(m, visible=False) for m in default_models]
            for m, o in zip(current_models, output):
                gen_event = gen_button.click(gen_fn, [m, txt_input], o)
                stop_button.click(lambda s: gr.update(interactive=False), None, stop_button, cancels=[gen_event])
        with gr.Accordion('Model selection'):
            model_choice = gr.CheckboxGroup(models, label=f'Choose up to {num_models} different models from the 800 available!', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
            model_choice.change(update_imgbox, model_choice, output)
            model_choice.change(extend_choices, model_choice, current_models)
        with gr.Row():
            gr.HTML("""
                <div class="footer">
                <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77 and Omnibus's Maximum Multiplier!
                </p>
            """)



js_code = """
document.addEventListener('pointerdown', function() {
    
    console.log('ghgh')
});
"""


with gr.Blocks(css="div.float.svelte-1mwvhlq {    position: absolute;    top: var(--block-label-margin);    left: var(--block-label-margin);    background: none;    border: none;}") as demo: 
  gr.Markdown("<script>" + js_code + "</script>")
    make_me()



demo.queue(concurrency_count=200)
demo.launch()