File size: 3,389 Bytes
7a9883b 90321c4 7a9883b 6d56b12 7a9883b 6d56b12 7a9883b 6d56b12 7a9883b 6d56b12 7a9883b 00230b5 6d56b12 00230b5 c1f0376 7a9883b 00230b5 7a9883b 4bbdf5d 7a9883b 6d56b12 7a9883b 6d56b12 7a9883b |
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 |
import gradio as gr
import random
from time import time, ctime
def promptgen(choice, num, artist):
t = time()
print(ctime(t))
if choice == "Prompt Generator v0.1(Better quality)":
prompt = open('pr1.txt').read().splitlines()
elif choice == "Prompt Generator v0.2(More tags)":
prompt = open('pr2.txt').read().splitlines()
if int(num) < 1 or int(num) > 20:
num = 10
if int(artist) < 0 or int(artist) > 40:
artist = 2
vocab = len(prompt)
generated = []
artists_num = 0
while len(sorted(set(generated), key=lambda d: generated.index(d))) < num:
rand = random.choice(prompt)
if rand.startswith('art by') and artists_num < artist:
artists_num +=1
generated.append(rand)
elif not rand.startswith('art by'):
generated.append(rand)
print(', '.join(set(generated)) + '\n\n')
return ', '.join(set(generated))
demo = gr.Blocks()
with demo:
gr.HTML(
"""
<div style="text-align: center; margin: 0 auto;">
<div style="display: inline-flex;align-items: center;gap: 0.8rem;font-size: 1.75rem;">
<h1 style="font-weight: 900; margin-bottom: 7px;margin-top:5px">
Simple Prompt Generator v0.6 (Gradio Demo)
</h1>
</div>
<p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">
Simple prompt generation script for Midjourney, DALLe, Stable and Disco diffusion and etc neural networks. <br> <p>More examples in <a class='link-info' href="https://github.com/WiNE-iNEFF/Simple_Prompt_Generator" target="_blank">Github</a> and <a class='link-info' href="https://wine-ineff.github.io/Simple_Prompt_Generator/" target="_blank">Project site</a></p>
</p>
<center>
<img style="display: inline-block, margin-right: 1%;" src='https://visitor-badge.glitch.me/badge?page_id=WiNE-iNEFF_Simple_Prompt_Generator&left_text=GithubVisitors' alt='visitor badge'><img style="display: inline-block;" src='https://visitor-badge.glitch.me/badge?page_id=WiNE-iNEFF_HF_Simple_Prompt_Generator&left_text=HuggingFaceVisitors' alt='visitor badge'>
</center>
</div>
"""
)
with gr.Column():
model_size = gr.Radio(["Prompt Generator v0.1(Better quality)", "Prompt Generator v0.2(More tags)"], label="Model Variant", value="Prompt Generator v0.1(Better quality)")
number = gr.Number(value="10", label="Num of tag (MAX 20)", show_label=True)
artist = gr.Number(value="2", label="Num of artist (Standart 2)", show_label=True)
out = gr.Textbox(lines=4, label="Generated Prompts")
greet_btn = gr.Button("Generate")
greet_btn.click(fn=promptgen, inputs=[model_size, number, artist], outputs=out)
gr.HTML(
"""
<div class="footer">
<div style='text-align: center;'>Simple Prompt Generator by <a href='https://twitter.com/wine_ineff' target='_blank'>Artsem Holub (WiNE-iNEFF)</a><br>More information about this demo and script your can find in <a class='link-info' href="https://github.com/WiNE-iNEFF/Simple_Prompt_Generator" target="_blank">Github</a> and <a class='link-info' href="https://wine-ineff.github.io/Simple_Prompt_Generator/" target="_blank">Project site</a></div>
</div>
"""
)
demo.queue(concurrency_count=4)
demo.launch() |