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

更新 21.3.:

• 複数の入力ファイルをサポート

• デフォルトでサンプル範囲を 10,000 まで拡張

プロンプトが記載されたファイルからランダムな数のプロンプトを選択するのに役立つツール。私はアップロードしているさまざまなプロンプトパッケージをテストする際に使用しています。十分なサンプル数を取得して数枚の画像を生成します。明らかな不適切なプロンプトを除外・修正し、その繰り返しです。

要件

• pip install gradio

使用法

• gradio guitoolkit.py

• または python guitoolkit.py を使用

使用手順

1. このファイルをダウンロードする / 以下のコードを guitoolkit.py（または任意の名前）というファイルにコピーする

2. 仮想環境を作成するか使用する python -m venv venv

3. 環境をアクティベートする venv\Scripts\activate

4. 本ツールを使用するために必要な gradio ライブラリをインストールするため、pip install gradio コマンドを実行する

5. インストールが完了したら、gradio guitoolkit.py または python guitoolkit.py のいずれかを実行する

6. 以下のメッセージが表示されたら、ツールが使用可能になります：gradio .\guitoolkit.py
reload mode で起動中: http://127.0.0.1:7861 (終了するには CTRL+C を押す)

7. これで http://127.0.0.1:7861 にアクセスして、ツールを使用できるようになります

8. シャッフルしたいファイルを入力し、必要な数を指定。出力をコピーして、たとえば Automatic1111 に貼り付ける

9. 

ソースコード

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)