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
reloading 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)