aboutsummaryrefslogtreecommitdiff
path: root/modules/styles.py
diff options
context:
space:
mode:
Diffstat (limited to 'modules/styles.py')
-rw-r--r--modules/styles.py11
1 files changed, 6 insertions, 5 deletions
diff --git a/modules/styles.py b/modules/styles.py
index 60bd8a7f..4ce32b8e 100644
--- a/modules/styles.py
+++ b/modules/styles.py
@@ -3,14 +3,15 @@ from modules import errors
import csv
import os
import typing
+from typing import Optional, Union
import shutil
class PromptStyle(typing.NamedTuple):
name: str
- prompt: str | None
- negative_prompt: str | None
- path: str | None = None
+ prompt: Optional[str]
+ negative_prompt: Optional[str]
+ path: Optional[str] = None
def merge_prompts(style_prompt: str, prompt: str) -> str:
@@ -79,7 +80,7 @@ def extract_original_prompts(style: PromptStyle, prompt, negative_prompt):
class StyleDatabase:
- def __init__(self, paths: list[str | Path]):
+ def __init__(self, paths: list[Union[str, Path]]):
self.no_style = PromptStyle("None", "", "", None)
self.styles = {}
self.paths = paths
@@ -128,7 +129,7 @@ class StyleDatabase:
if styles_file.is_file():
self.load_from_csv(styles_file)
- def load_from_csv(self, path: str | Path):
+ def load_from_csv(self, path: Union[str, Path]):
try:
with open(path, "r", encoding="utf-8-sig", newline="") as file:
reader = csv.DictReader(file, skipinitialspace=True)