import gradio as gr import random from time import time, ctime def promptgen(choice, num): t = time() print(ctime(t)) if choice == "Prompt Generator v0.1(Better quality)": prompt = open('prompttext.txt').read().splitlines() elif choice == "Prompt Generator v0.2(More tags)": prompt = open('prompttext2.txt').read().splitlines() if int(num) < 1 or int(num) > 20: num = 10 vocab = len(prompt) generated = [] while len(sorted(set(generated), key=lambda d: generated.index(d))) < num: rand = random.randint(0, vocab) generated.append(prompt[rand-1]) return ' '.join(sorted(set(generated), key=lambda d: generated.index(d))) demo = gr.Blocks() with demo: gr.HTML( """
Simple prompt generation script for Midjourney, DALLe, Stable and Disco diffusion and etc neural networks.
More examples in Github and Project site