Update app.py

app.py

@@ -56,17 +56,17 @@ def generateText(model, minLen=100, size=5):
 
     return output
 
-def sentence_builder(quantity, Type, negative):
+def sentence_builder(quantity, minLen, Type, negative):
     if Type == "NSFW": idx=1
     elif Type == "SFW": idx=2
     else: idx=0
     model = models[idx]
     output = ""
     for i in range(quantity):
-        prompt = generateText(model[0], minLen=300, size=1)[0]
+        prompt = generateText(model[0], minLen=minLen, size=1)[0]
         output+=f"PROMPT:  {prompt}\n\n"
         if negative: 
-            negative_prompt = generateText(model[1], minLen=300, size=5)[0]
+            negative_prompt = generateText(model[1], minLen=minLen, size=5)[0]
             output+=f"NEGATIVE PROMPT:  {negative_prompt}\n"
         output+="----------------------------------------------------------------"
         output+="\n\n\n"
@@ -78,6 +78,7 @@ ui = gr.Interface(
     sentence_builder,
     [
         gr.Slider(1, 10, value=4, label="Count", info="Choose between 1 and 10", step=1),
+        gr.Slider(100, 1000, value=300, label="minLen", info="Choose between 100 and 1000", step=50),
         gr.Radio(["NSFW", "SFW", "BOTH"], label="TYPE", info="NSFW stands for NOT SAFE FOR WORK, so choose any one you want?"),
         gr.Checkbox(label="negitive Prompt", info="Do you want to generate negative prompt as well as prompt?"),
     ],