using JSON block for code ouput

app.py

@@ -9,19 +9,19 @@ def device_change(device, config):
     
     config = set_config(config, 'device', device)
     
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def model_refiner_change(refiner, config):
     
     config = set_config(config, 'refiner', refiner)
     
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def cpu_offload_change(cpu_offload, config):
     
     config = set_config(config, 'cpu_offload', cpu_offload)
     
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def models_change(model, scheduler, config):
 
@@ -55,85 +55,85 @@ def models_change(model, scheduler, config):
     # safety_checker_change(in_safety_checker.value, config)
     # requires_safety_checker_change(in_requires_safety_checker.value, config)
 
-    return model_description, refiner, trigger_token, use_safetensors, scheduler, config, str(config), assemble_code(config)
+    return model_description, refiner, trigger_token, use_safetensors, scheduler, config, config, assemble_code(config)
 
 def data_type_change(data_type, config):
 
     config = set_config(config, 'data_type', data_type)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def tensorfloat32_change(allow_tensorfloat32, config):  
         
     config = set_config(config, 'allow_tensorfloat32', allow_tensorfloat32)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def inference_steps_change(inference_steps, config):
     
     config = set_config(config, 'inference_steps', inference_steps)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def manual_seed_change(manual_seed, config):
     
     config = set_config(config, 'manual_seed', manual_seed)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def guidance_scale_change(guidance_scale, config):
 
     config = set_config(config, 'guidance_scale', guidance_scale)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def enable_vae_slicing_change(enable_vae_slicing, config):
 
     config = set_config(config, 'enable_vae_slicing', enable_vae_slicing)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def enable_vae_tiling_change(enable_vae_tiling, config):
 
     config = set_config(config, 'enable_vae_tiling', enable_vae_tiling)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def prompt_change(prompt, config):
     
     config = set_config(config, 'prompt', prompt)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def trigger_token_change(trigger_token, config):
     
     config = set_config(config, 'trigger_token', trigger_token)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def negative_prompt_change(negative_prompt, config):
     
     config = set_config(config, 'negative_prompt', negative_prompt)
     
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def variant_change(variant, config):
     
     config = set_config(config, 'variant', variant)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
     
 def safety_checker_change(safety_checker, config):
         
     config = set_config(config, 'safety_checker', safety_checker)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def requires_safety_checker_change(requires_safety_checker, config):
 
     config = set_config(config, 'requires_safety_checker', requires_safety_checker)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def auto_encoders_change(auto_encoder, config):
     
@@ -146,7 +146,7 @@ def auto_encoders_change(auto_encoder, config):
 
     config = set_config(config, 'auto_encoder', auto_encoder)
 
-    return auto_encoder_description, config, str(config), assemble_code(config)
+    return auto_encoder_description, config, config, assemble_code(config)
 
 def schedulers_change(scheduler, config):
     
@@ -159,7 +159,7 @@ def schedulers_change(scheduler, config):
         
     config = set_config(config, 'scheduler', scheduler)
 
-    return scheduler_description, config, str(config), assemble_code(config)
+    return scheduler_description, config, config, assemble_code(config)
     
 def adapters_textual_inversion_change(adapter_textual_inversion, config):
     
@@ -174,13 +174,13 @@ def adapters_textual_inversion_change(adapter_textual_inversion, config):
     
     config = set_config(config, 'adapter_textual_inversion', adapter_textual_inversion)
     
-    return adapter_textual_inversion_description, in_adapters_textual_inversion_token, config, str(config), assemble_code(config)
+    return adapter_textual_inversion_description, in_adapters_textual_inversion_token, config, config, assemble_code(config)
 
 def textual_inversion_token_change(adapter_textual_inversion_token, config):
     
     config = set_config(config, 'adapter_textual_inversion_token', adapter_textual_inversion_token)
 
-    return config, str(config), assemble_code(config)
+    return config, config, assemble_code(config)
 
 def run_inference(config, config_history, progress=gr.Progress(track_tqdm=True)):
     
@@ -382,7 +382,8 @@ with gr.Blocks(analytics_enabled=False) as demo:
         out_image = gr.Image()
         out_code = gr.Code(assemble_code(config.value), label="Code")
     with gr.Row():
-        out_config = gr.Code(value=str(config.value), label="Current config")
+        # out_config = gr.Code(value=str(config.value), label="Current config")
+        out_config = gr.JSON(value=config.value, label="Current config")
     with gr.Row():
         out_config_history = gr.Markdown(dict_list_to_markdown_table(config_history.value))