Add type annotation
Browse files
app.py
CHANGED
|
@@ -63,8 +63,8 @@ default_style = styles[default_style_name]
|
|
| 63 |
style_names = list(styles.keys())
|
| 64 |
|
| 65 |
|
| 66 |
-
def apply_style(
|
| 67 |
-
p, n = styles.get(
|
| 68 |
return p.replace("{prompt}", positive), n + negative
|
| 69 |
|
| 70 |
|
|
|
|
| 63 |
style_names = list(styles.keys())
|
| 64 |
|
| 65 |
|
| 66 |
+
def apply_style(style_name: str, positive: str, negative: str = "") -> tuple[str, str]:
|
| 67 |
+
p, n = styles.get(style_name, default_style)
|
| 68 |
return p.replace("{prompt}", positive), n + negative
|
| 69 |
|
| 70 |
|