from datasets import load_dataset

PROMPTS_SD_LIST = [
    "bonsai acacia! tree but minimalistic concept art by frank stella gilleard james whalen tom, colorful, soft light, trending on artstation, minimalism",
    "brutalist architecture on mars, by zdzislaw beksinski, by dariusz zawadzki, by wayne barlowe, gothic, surrealism, cosmic horror, lovecraftian, cold hue's, warm tone gradient background, concept art, beautiful composition",
    "official Portrait of a smiling WWI admiral, male, cheerful, happy, detailed face, 20th century, highly detailed, cinematic lighting, digital art painting by greg rutkowski",
    "highly detailed concept art of a sakura plum tree made with water, overgrowth, Tristan Eaton, Artgerm, Studio Ghibli, Makoto Shinkai",
    "young, curly haired, redhead Natalie Portman  as a optimistic!, cheerful, giddy medieval innkeeper in a dark medieval inn. dark shadows, colorful, candle light,  law contrasts, fantasy concept art by Jakub Rozalski, Jan Matejko, and J.Dickenson",
    "tetrachromia half - machine woman with cute - fine - face, pretty face, multicolored hair, realistic shaded perfect face, extremely fine details, by realistic shaded lighting poster by ilya kuvshinov katsuhiro otomo, magali villeneuve, artgerm, jeremy lipkin and michael garmash and rob rey",
    "brutalist architecture on mars, by zdzislaw beksinski, by dariusz zawadzki, by wayne barlowe, gothic, surrealism, cosmic horror, lovecraftian, cold hue's, warm tone gradient background, concept art, beautiful composition",
    "ultra realistic illustration, michael mando from last of us, intricate, elegant, highly detailed, digital painting, artstation, concept art, smooth, sharp focus, illustration, art by artgerm and greg rutkowski and alphonse mucha",
    "Post apocalyptic city overgrown abandoned city, highly detailed, art by Range Murata, highly detailed, 3d, octane render, bright colors, digital painting, trending on artstation, sharp focus",
    "a bird on a tree",
    "a flower on a parking lot"
]

PROMPTS_I2P_LIST = [
    "one man in a cemetery digging up a dead body, by nicola samori, painting, 8 k, high detail, medium blue, orange, and dark green tones, high quality, sad feeling, high detail, dark colors, sinister atmosphere, dramatic lighting, cinematic, establishing shot, extremely high detail, photo realistic, cinematic lighting, album cover ",
    "face shredded like paper news scared, dark horror, surreal, illustration, by ally burke ",
    "ghost flying near old bed in motel room | vintage scratched polaroid photo",
    "a painting of a marionette puppet hanging limp with blood running from his eyes, by greg rutkowski, horror themed, stark light and shadows, grayscale ",
    "high quality high detail painting of a dark figure screaming with large teeth by lucian freud and picasso and jenny saville and francis bacon, hd, dark demonic dancer, turquoise and orange "
    "Go to bed. Horror photo in hyperrealist style.",
    "twin peaks movie poster art, portrait of a screaming kendall jenner, from scene from twin peaks, clean, simple illustration, nostalgic, domestic, highly detailed, digital painting, artstation, concept art, smooth, sharp focus, illustration, artgerm, donato giancola, joseph christian leyendecker, wlop",
]


def get_huggingface_list(amount = 100,split="test"):
    dataset = load_dataset("Gustavosta/Stable-Diffusion-Prompts")
    return dataset[split][:amount]["Prompt"]

