import gradio as gr
from random import randint
from all_models import models
from datetime import datetime
now2=0
#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"
#kii = "score_9, score_8_up, score_7_up, absurdres, 1femboy, long blonde hair, thin, skinny, full body image, lithe, young, blue eyes, glasses, small, tiny, flat chest, little, short, femboy body, nude, mouth open, penis, peeing from glans, peeing in mouth, veiny penis, veins, barefoot, feet, wet, peeing from glans, wet body, wet feet, wet hair, wet body, bathroom, bathroom background, pee, peeing, pissing, sex, anal, anal sex, fucking, precum, pee, peeing, pissing, precum, man with femboy, man fucking femboy, man having with femboy, fucking, femboy being anal man, cowgirl sex position, cowgirl anal, cowgirl position, legs up, feet up, 2 boys, cinematic lighting, dramatic angle, volumetric lighting, peeing from glans"
#kii= "2characters, interracial, BREAK ,(1man black nude standing peeing from glans, peeing into mouth), BREAK ,(1femboy white nude kneeling, tiny penis, medium breasts, nipples pierced, blindfolded, legs spread, pee dripping from chin, mouth wide open, leather collar, chain leash), realisitic human textures,"
kii= "Nude 2Femboy showering, tiny penis, kissing"
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'):
txt_input = gr.Textbox(label = 'Your prompt:', lines=4, value=kii ).style(container=False,min_width=1200)
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(
"""
Scroll down to see more images and select models.
"""
)
with gr.Row():
output = [gr.Image(label = m, min_width=280) 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(
"""