Tool that shuffles and picks x amount of prompts from input prompt file(s)

Updated 21.3.:

• Support for multiple input files added

• Extended sample range to 10 000 by default

Tool that helps with selecting a random amount of prompts from a file that contains prompts. I am using it when testing the different prompt packages I am uploading. I'll take a big enough sample to generate a few images. Remove and fix obvious maligned prompts, rinse and repeat.

Requirements

• pip install gradio

Usage

• gradio guitoolkit.py

• or use python guitoolkit.py

How to guide

1. Download this file / copy the code below into a file called guitoolkit.py (or whatever you want to call it)

2. Make/use a virtual environment python -m venv venv

3. Activate environment venv\Scripts\activate

4. Run the command pip install gradio to install the gradio library which is required to use this

5. When you have installed that, run either gradio guitoolkit.py or python guitoolkit.py

6. You should now have the tool ready to use if you get the following: gradio .\guitoolkit.py
launching in reload mode on: http://127.0.0.1:7861 (Press CTRL+C to quit)

7. You can now visit http://127.0.0.1:7861 where the tool is ready to use

8. Input the file(s) you want to shuffle, select how many you want, copy the output, insert it into e.g. Automatic1111

9. 

Source code

import gradio as gr
import random

def shuffle_file(file_obj, no_prompts):
    prompts = []
    for file in file_obj:
        with open(file.name) as infile:
            in_prompts = infile.readlines()
        prompts.extend(list(set(in_prompts)))
    
    prompts = random.sample(prompts, no_prompts)
    random.shuffle(prompts)
    print(type(prompts))
    return "".join(prompts)

demo = gr.Interface(
    fn=shuffle_file,
    inputs=["files",  gr.Slider(5, 10000)],
    outputs=["code"],
)

if __name__ == "__main__":
    demo.launch(server_port=9800)