Update app.py

app.py

@@ -21,13 +21,13 @@ def promptgen(choice, num, artist):
   generated = []
   artists_num = 0
   while len(sorted(set(generated), key=lambda d: generated.index(d))) < num:
-    rand = random.randint(0, vocab)
-    if prompt[rand-1].startswith('art by') and artists_num < artist:
+    rand = random.choice(prompt)
+    if rand.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')
+        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()
@@ -38,7 +38,7 @@ with demo:
             <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.4 (Gradio Demo)
+                  Simple Prompt Generator v0.6 (Gradio Demo)
                 </h1>
               </div>
               <p style="margin-bottom: 10px; font-size: 94%; line-height: 23px;">