Asahina2K commited on
Commit
0cfdbbe
1 Parent(s): 50c0853

rollback app.py, change text metadata at utils to parameters

Browse files
Files changed (2) hide show
  1. app.py +10 -10
  2. utils.py +1 -1
app.py CHANGED
@@ -107,7 +107,7 @@ def generate(
107
 
108
  if use_upscaler:
109
  upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
110
- parameters = {
111
  "prompt": prompt,
112
  "negative_prompt": negative_prompt,
113
  "resolution": f"{width} x {height}",
@@ -123,20 +123,20 @@ def generate(
123
  if use_upscaler:
124
  new_width = int(width * upscale_by)
125
  new_height = int(height * upscale_by)
126
- parameters["use_upscaler"] = {
127
  "upscale_method": "nearest-exact",
128
  "upscaler_strength": upscaler_strength,
129
  "upscale_by": upscale_by,
130
  "new_resolution": f"{new_width} x {new_height}",
131
  }
132
  else:
133
- parameters["use_upscaler"] = None
134
- parameters["Model"] = {
135
  "Model": DESCRIPTION,
136
  "Model hash": "e3c47aedb0",
137
  }
138
 
139
- logger.info(json.dumps(parameters, indent=4))
140
 
141
  try:
142
  if use_upscaler:
@@ -175,14 +175,14 @@ def generate(
175
 
176
  if images:
177
  image_paths = [
178
- utils.save_image(image, parameters, OUTPUT_DIR, IS_COLAB)
179
  for image in images
180
  ]
181
 
182
  for image_path in image_paths:
183
  logger.info(f"Image saved as {image_path} with metadata")
184
 
185
- return image_paths, parameters
186
  except Exception as e:
187
  logger.exception(f"An error occurred: {e}")
188
  raise
@@ -334,11 +334,11 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
334
  show_label=False
335
  )
336
  with gr.Accordion(label="Generation Parameters", open=False):
337
- gr_parameters = gr.JSON(label="parameters", show_label=False)
338
  gr.Examples(
339
  examples=config.examples,
340
  inputs=prompt,
341
- outputs=[result, gr_parameters],
342
  fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
343
  cache_examples=CACHE_EXAMPLES,
344
  )
@@ -387,7 +387,7 @@ with gr.Blocks(css="style.css", theme="NoCrypt/[email protected]") as demo:
387
  upscale_by,
388
  add_quality_tags,
389
  ],
390
- outputs=[result, gr_parameters],
391
  api_name="run",
392
  )
393
 
 
107
 
108
  if use_upscaler:
109
  upscaler_pipe = StableDiffusionXLImg2ImgPipeline(**pipe.components)
110
+ metadata = {
111
  "prompt": prompt,
112
  "negative_prompt": negative_prompt,
113
  "resolution": f"{width} x {height}",
 
123
  if use_upscaler:
124
  new_width = int(width * upscale_by)
125
  new_height = int(height * upscale_by)
126
+ metadata["use_upscaler"] = {
127
  "upscale_method": "nearest-exact",
128
  "upscaler_strength": upscaler_strength,
129
  "upscale_by": upscale_by,
130
  "new_resolution": f"{new_width} x {new_height}",
131
  }
132
  else:
133
+ metadata["use_upscaler"] = None
134
+ metadata["Model"] = {
135
  "Model": DESCRIPTION,
136
  "Model hash": "e3c47aedb0",
137
  }
138
 
139
+ logger.info(json.dumps(metadata, indent=4))
140
 
141
  try:
142
  if use_upscaler:
 
175
 
176
  if images:
177
  image_paths = [
178
+ utils.save_image(image, metadata, OUTPUT_DIR, IS_COLAB)
179
  for image in images
180
  ]
181
 
182
  for image_path in image_paths:
183
  logger.info(f"Image saved as {image_path} with metadata")
184
 
185
+ return image_paths, metadata
186
  except Exception as e:
187
  logger.exception(f"An error occurred: {e}")
188
  raise
 
334
  show_label=False
335
  )
336
  with gr.Accordion(label="Generation Parameters", open=False):
337
+ gr_metadata = gr.JSON(label="metadata", show_label=False)
338
  gr.Examples(
339
  examples=config.examples,
340
  inputs=prompt,
341
+ outputs=[result, gr_metadata],
342
  fn=lambda *args, **kwargs: generate(*args, use_upscaler=True, **kwargs),
343
  cache_examples=CACHE_EXAMPLES,
344
  )
 
387
  upscale_by,
388
  add_quality_tags,
389
  ],
390
+ outputs=[result, gr_metadata],
391
  api_name="run",
392
  )
393
 
utils.py CHANGED
@@ -169,7 +169,7 @@ def save_image(image, metadata, output_dir, is_colab):
169
  filepath = os.path.join(output_dir, filename)
170
  metadata_str = json.dumps(metadata)
171
  info = PngImagePlugin.PngInfo()
172
- info.add_text("metadata", metadata_str)
173
  image.save(filepath, "PNG", pnginfo=info)
174
  return filepath
175
 
 
169
  filepath = os.path.join(output_dir, filename)
170
  metadata_str = json.dumps(metadata)
171
  info = PngImagePlugin.PngInfo()
172
+ info.add_text("parameters", metadata_str)
173
  image.save(filepath, "PNG", pnginfo=info)
174
  return filepath
175