databank.py

import random
import catalogo

def obtenPrompt(endpoint, **kwargs): #Son opcionales todos los demás argumentos que recibirá. 
   
    outfit_color = kwargs["outfit_color"] if "outfit_color" in kwargs else ""
    hair_style = kwargs["hair_style"] if "hair_style" in kwargs else ""

    if endpoint == "partynight":
        return f"A manga picture of a beautiful girl with a sexy {outfit_color} sequins cocktail dress"
    elif endpoint == "babydoll":
        return f"A manga picture of a beautiful girl with a sexy {outfit_color} babydoll dress"
    elif endpoint == "leather":
        return f"A {defineEstilo('leather')} picture of a beautiful girl with a neck chocker, a black leather skirt and black deep-cut long sleeved blouse"
    elif endpoint == "oktoberfest":
        return f"A Ghibli manga of a beautiful Oktoberfest waitress with big boobs serving beers"
    elif endpoint == "ghibli":
        return f"A Ghibli manga of a beautiful girl"
    elif endpoint == "wildwest":
        return f"A Ghibli manga of a beautiful wild west girl with a poncho, a revolver and bullets bandolier"
    elif endpoint == "stewardess":  
        return f"A Ghibli manga of a beautiful flight attendant in a {outfit_color} uniform skirt {obtenAccesorio('stewardess')}"
    elif endpoint == "bride":
        return f"A {defineEstilo('bride')} art nouveau of a beautiful bride in a spectacular wedding dress"
    elif endpoint == "swimsuit":
        return f"A manga picture of a beautiful girl in {outfit_color} swimsuit"
    elif endpoint == "lingerie":
        return f"A manga picture of a girl in corset and hosiery"
    elif endpoint == "schooluniform":
        return f"A manga picture of a girl in school uniform"
    elif endpoint == "frenchmaid":
        return f"A manga picture of a girl in french maid uniform with maid's tiara"
    elif endpoint == "cheerleader":
        return f"A manga picture of a cheerleader with {outfit_color} uniform and {hair_style} hair style."
    
def obtenAccesorio(endpoint):

    if endpoint == "partynight":
        return 
    elif endpoint == "babydoll":
        return 
    elif endpoint == "leather":
        return 
    elif endpoint == "oktoberfest":
        return 
    elif endpoint == "ghibli":
        return 
    elif endpoint == "wildwest":
        return 
    elif endpoint == "stewardess":
        accesorio = random.choice(catalogo.accesorios_stewardess)
        if accesorio !='':
            print("Si hay accesorio...")
            return 'with a ' + random.choice(catalogo.colores) + ' ' + accesorio
        else: 
            return ''

    elif endpoint == "bride":
        return 
    elif endpoint == "swimsuit":
        return 
    elif endpoint == "lingerie":
        return 
    elif endpoint == "schooluniform":
        return 
    elif endpoint == "frenchmaid":
        return 
    elif endpoint == "cheerleader":
        return 
    
def defineEstilo(endpoint):
    if endpoint == "partynight":
        return 
    elif endpoint == "babydoll":
        return 
    elif endpoint == "leather":
        estilo = ["watercolor", "manga"]
        return random.choice(estilo) 
    elif endpoint == "oktoberfest":
        return 
    elif endpoint == "ghibli":
        return 
    elif endpoint == "wildwest":
        estilo = ["watercolor picture", "manga picture"]
        return random.choice(estilo) 
    elif endpoint == "stewardess":
        return 
    elif endpoint == "bride":
        estilo = ["anime", "manga"]
        return random.choice(estilo)
    elif endpoint == "swimsuit":
        return 
    elif endpoint == "lingerie":
        return 
    elif endpoint == "schooluniform":
        return 
    elif endpoint == "frenchmaid":
        return 
    elif endpoint == "cheerleader":
        return 
    return random.choice(catalogo.estilo)