Update app.py

app.py

@@ -2,25 +2,33 @@ import gradio as gr
 import random
 from time import time, ctime
 
-def promptgen(choice, num):
+def promptgen(choice, num, artist):
   t = time()
   print(ctime(t))
 
   if choice == "Prompt Generator v0.1(Better quality)":
-    prompt = open('prompttext.txt').read().splitlines()
+    prompt = open('pr1.txt').read().splitlines()
   elif choice == "Prompt Generator v0.2(More tags)":
-    prompt = open('prompttext2.txt').read().splitlines()
+    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.randint(0, vocab)
-    generated.append(prompt[rand-1])
-  print(' '.join(sorted(set(generated), key=lambda d: generated.index(d))))
-  return ' '.join(sorted(set(generated), key=lambda d: generated.index(d)))
+    if prompt[rand-1].startswith('art by') and artists_num < artist:
+        artists_num +=1 
+		generated.append(prompt[rand-1])
+	elif not prompt[rand-1].startswith('art by'):
+		generated.append(prompt[rand-1])
+  print(' '.join(set(generated)) + '\n')
+  return ' '.join(set(generated))
 
 demo = gr.Blocks()
 
@@ -45,9 +53,10 @@ with demo:
   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], outputs=out)
+  greet_btn.click(fn=promptgen, inputs=[model_size, number, artist], outputs=out)
   gr.HTML(
             """
                 <div class="footer">