feature: added flag, config and ui update for image extension change #1789

modules/config.py

@@ -248,6 +248,11 @@ default_max_image_number = get_config_item_or_set_default(
     default_value=32,
     validator=lambda x: isinstance(x, int) and x >= 1
 )
+default_image_extension = get_config_item_or_set_default(
+    key='default_image_extension',
+    default_value='png',
+    validator=lambda x: x in modules.flags.image_extensions
+)
 default_image_number = get_config_item_or_set_default(
     key='default_image_number',
     default_value=2,
@@ -330,6 +335,7 @@ possible_preset_keys = [
     "default_sampler",
     "default_scheduler",
     "default_performance",
+    "default_image_extension",
     "default_prompt",
     "default_prompt_negative",
     "default_styles",

modules/flags.py

@@ -34,6 +34,7 @@ default_parameters = {
 
 inpaint_engine_versions = ['None', 'v1', 'v2.5', 'v2.6']
 performance_selections = ['Speed', 'Quality', 'Extreme Speed']
+image_extensions = ['png', 'jpg']
 
 inpaint_option_default = 'Inpaint or Outpaint (default)'
 inpaint_option_detail = 'Improve Detail (face, hand, eyes, etc.)'

modules/private_logger.py

@@ -13,7 +13,7 @@ log_cache = {}
 
 def get_current_html_path():
     date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs,
-                                                                         extension='png')
+                                                                         extension=modules.config.default_image_extension)
     html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')
     return html_name
 
@@ -22,7 +22,7 @@ def log(img, dic):
     if args_manager.args.disable_image_log:
         return
 
-    date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension='png')
+    date_string, local_temp_filename, only_name = generate_temp_filename(folder=modules.config.path_outputs, extension=modules.config.default_image_extension)
     os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
     Image.fromarray(img).save(local_temp_filename)
     html_name = os.path.join(os.path.dirname(local_temp_filename), 'log.html')