Embed SillyTavern generation metadata
Browse files
app.py
CHANGED
|
@@ -694,6 +694,7 @@ def default_loras():
|
|
| 694 |
|
| 695 |
class DirectRequest(BaseModel):
|
| 696 |
prompt: str
|
|
|
|
| 697 |
second_prompt: str = ""
|
| 698 |
regions: list[RegionRequest] = Field(default_factory=list, max_length=3)
|
| 699 |
regional_mode: str = REGIONAL_MODES[0]
|
|
@@ -2015,7 +2016,7 @@ def generate_images(request, from_api=True):
|
|
| 2015 |
second_style_image,
|
| 2016 |
)
|
| 2017 |
|
| 2018 |
-
metadata_config = request.model_dump()
|
| 2019 |
if request.style_images:
|
| 2020 |
metadata_config["style_images"] = [
|
| 2021 |
f"style-reference-{index}.png"
|
|
@@ -2041,6 +2042,11 @@ def generate_images(request, from_api=True):
|
|
| 2041 |
ENVIRONMENT_START,
|
| 2042 |
REGIONAL_GLOBAL_STRENGTH,
|
| 2043 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2044 |
results = [
|
| 2045 |
Image.fromarray(
|
| 2046 |
np.clip(item.detach().numpy() * 255, 0, 255).astype(np.uint8)
|
|
|
|
| 694 |
|
| 695 |
class DirectRequest(BaseModel):
|
| 696 |
prompt: str
|
| 697 |
+
sillytavern: dict = Field(default_factory=dict)
|
| 698 |
second_prompt: str = ""
|
| 699 |
regions: list[RegionRequest] = Field(default_factory=list, max_length=3)
|
| 700 |
regional_mode: str = REGIONAL_MODES[0]
|
|
|
|
| 2016 |
second_style_image,
|
| 2017 |
)
|
| 2018 |
|
| 2019 |
+
metadata_config = request.model_dump(exclude={"sillytavern"})
|
| 2020 |
if request.style_images:
|
| 2021 |
metadata_config["style_images"] = [
|
| 2022 |
f"style-reference-{index}.png"
|
|
|
|
| 2042 |
ENVIRONMENT_START,
|
| 2043 |
REGIONAL_GLOBAL_STRENGTH,
|
| 2044 |
)
|
| 2045 |
+
if request.sillytavern:
|
| 2046 |
+
metadata["sillytavern"] = json.dumps(
|
| 2047 |
+
request.sillytavern,
|
| 2048 |
+
separators=(",", ":"),
|
| 2049 |
+
)
|
| 2050 |
results = [
|
| 2051 |
Image.fromarray(
|
| 2052 |
np.clip(item.detach().numpy() * 255, 0, 255).astype(np.uint8)
|