From 9dfa1a99e386d07fcea5f5505697e63a6a0ffaa6 Mon Sep 17 00:00:00 2001
From: eddyizm <eddyizm@gmail.com>
Date: Wed, 10 Jan 2024 19:06:40 -0800
Subject: [PATCH] moved function to config module

---
 modules/config.py | 6 ++++++
 webui.py          | 7 +------
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/modules/config.py b/modules/config.py
index 35077cda..0dc9929e 100644
--- a/modules/config.py
+++ b/modules/config.py
@@ -151,6 +151,12 @@ def get_config_item_or_set_default(key, default_value, validator, disable_empty_
         return default_value
 
 
+def update_image_extension(value):
+    '''Updating default image extension used for saving output image'''
+    modules.config.default_image_extension = value
+    print(f'[CONFIG] Updated image extension value: {modules.config.default_image_extension}')
+
+
 default_base_model_name = get_config_item_or_set_default(
     key='default_model',
     default_value='juggernautXL_version6Rundiffusion.safetensors',
diff --git a/webui.py b/webui.py
index d21cd557..17c8a6ed 100644
--- a/webui.py
+++ b/webui.py
@@ -485,12 +485,7 @@ with shared.gradio_root:
                                          scheduler_name, adaptive_cfg, refiner_swap_method, negative_prompt
                                      ], queue=False, show_progress=False)
         
-        def update_image_extension(value):
-            print(f'this is x: {value}\nthis is config: {modules.config.default_image_extension}')
-            modules.config.default_image_extension = value
-            print(f'updated config value: {modules.config.default_image_extension}')
-
-        image_extension.input(lambda x: update_image_extension(x), 
+        image_extension.input(lambda x: modules.config.update_image_extension(x), 
                               inputs=image_extension)
                               
         advanced_checkbox.change(lambda x: gr.update(visible=x), advanced_checkbox, advanced_column,