File size: 4,300 Bytes
6fef025
30e73c8
f5b8400
53b0019
 
 
 
 
 
 
 
 
6fef025
cebd60e
8980225
 
e594368
8fe517c
cebd60e
8fe517c
8980225
 
 
 
 
779cd45
8980225
 
f5b8400
53b0019
f5b8400
 
 
 
 
 
 
 
 
 
6fef025
 
 
19aac56
 
81987e1
 
6fef025
81987e1
19aac56
f5b8400
6fef025
 
 
f5b8400
 
6fef025
 
f5b8400
 
fc42fea
6fef025
 
f5b8400
 
 
d2777ff
1485f13
6fef025
 
 
8fc64ff
 
bdcf524
6b70d61
8980225
225d490
8980225
53b0019
 
 
 
 
8f8f343
 
cebd60e
3977ac5
 
 
 
 
8f8f343
3977ac5
 
 
 
 
 
8f8f343
 
 
 
 
 
 
 
e7bc2e8
8f8f343
 
654d3d1
 
 
 
6ce4171
654d3d1
 
 
f5b8400
8fc64ff
e5beca8
8fc64ff
 
f5b8400
 
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import gradio as gr
import random
from all_models import models
from datetime import datetime

def get_current_time():
    # Get the current time
    now = datetime.now()
    # Format the time as a string
    current_time = now.strftime("%Y-%m-%d %H:%M:%S")
    return current_time



# Function to generate a string with text and a random number
def generate_text_with_random_number():
    #random_number = randint(1, 100)
    kii = "nude femboy sissy with small penis,  bound anal sex orgy party, perfect penis, oral sex, threesum, head between thighs, glans inside mouth, cuming from glans"
    ki=f'{kii} {random.randint(1,100)}'
    return ki

# Function to process the input and return it (or do some other processing)
def process_text(input_text):
    return f"You entered: {input_text}"




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('') #str(randint(0, 99999999999))
    return models_load[model_str](f'{prompt} {noise}')





def make_me():
    with gr.Tab('The Dream'): 
        fn=process_text,
        txt_input = gr.Textbox(label = 'Your prompt:', lines=4, value=generate_text_with_random_number()   ).style(container=False,min_width=1200)
                               #"nude femboy sissy with small penis,  bound anal sex orgy party, perfect penis, oral sex, threesum, head between thighs, glans inside mouth, cuming from glans"

        time_textbox = gr.Textbox(label="Current Time")
        refresh_button = gr.Button("Refresh Time")
        refresh_button.click(fn=get_current_time, outputs=time_textbox)

        gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
        stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
        gr.HTML(
        """
            <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
              <div>
                <body>
                <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images and select models.</p>
                </div>
                </body>
              </div>
            </div>
        """
               )
        with gr.Row():
            output = [gr.Image(label = m, min_width=480) 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>
    """
)

with gr.Blocks() as demo:
    make_me()


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