q6 commited on
Commit
c78d1eb
·
1 Parent(s): a16f06e

Embed SillyTavern generation metadata

Browse files
Files changed (1) hide show
  1. app.py +7 -1
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)