25 lines
736 B
Python
25 lines
736 B
Python
from __future__ import annotations
|
|
from typing import List
|
|
from pathlib import Path
|
|
import csv
|
|
|
|
from dynamicprompts import constants
|
|
|
|
|
|
def write_prompts(
|
|
path: Path | str,
|
|
positive_prompt: str,
|
|
negative_prompt: str,
|
|
positive_prompts: List[str],
|
|
negative_prompts: List[str],
|
|
) -> Path:
|
|
path = Path(path)
|
|
with path.open("w", encoding=constants.DEFAULT_ENCODING, errors="ignore") as f:
|
|
writer = csv.writer(f)
|
|
writer.writerow(["positive_prompt", "negative_prompt"])
|
|
writer.writerow([positive_prompt, negative_prompt])
|
|
for positive_prompt, negative_prompt in zip(positive_prompts, negative_prompts):
|
|
writer.writerow([positive_prompt, negative_prompt])
|
|
|
|
return path
|