wip: add metadata mapping, reading and writing

applying data after reading currently not functional for A1111

webui.py

@@ -14,6 +14,7 @@ import modules.gradio_hijack as grh
 import modules.advanced_parameters as advanced_parameters
 import modules.style_sorter as style_sorter
 import modules.meta_parser
+import modules.metadata
 import args_manager
 import copy
 
@@ -21,6 +22,7 @@ from modules.sdxl_styles import legal_style_names
 from modules.private_logger import get_current_html_path
 from modules.ui_gradio_extensions import reload_javascript
 from modules.auth import auth_enabled, check_auth
+from modules.util import is_json
 
 
 def generate_clicked(*args):
@@ -208,6 +210,28 @@ with shared.gradio_root:
                                     value=flags.desc_type_photo)
                                 desc_btn = gr.Button(value='Describe this Image into Prompt')
                                 gr.HTML('<a href="https://github.com/lllyasviel/Fooocus/discussions/1363" target="_blank">\U0001F4D4 Document</a>')
+                    with gr.TabItem(label='Load Metadata') as load_tab:
+                        with gr.Column():
+                            metadata_input_image = grh.Image(label='Drag any image generated by Fooocus here', source='upload', type='filepath')
+                            metadata_json = gr.JSON(label='Metadata')
+                            metadata_import_button = gr.Button(value='Overwrite Input Values')
+
+                        def trigger_metadata_preview(filepath):
+                            parameters, items, metadata_scheme = modules.metadata.read_info_from_image(filepath)
+
+                            results = {}
+                            if parameters is not None:
+                                results['parameters'] = parameters
+                            if items:
+                                results['items'] = items
+                            if metadata_scheme is not None:
+                                results['metadata_scheme'] = metadata_scheme
+
+                            return results
+
+                        metadata_input_image.upload(trigger_metadata_preview, inputs=metadata_input_image,
+                                                    outputs=metadata_json)
+
             switch_js = "(x) => {if(x){viewer_to_bottom(100);viewer_to_bottom(500);}else{viewer_to_top();} return x;}"
             down_js = "() => {viewer_to_bottom();}"
 
@@ -548,14 +572,8 @@ with shared.gradio_root:
 
         def parse_meta(raw_prompt_txt, is_generating):
             loaded_json = None
-            try:
-                if '{' in raw_prompt_txt:
-                    if '}' in raw_prompt_txt:
-                        if ':' in raw_prompt_txt:
-                            loaded_json = json.loads(raw_prompt_txt)
-                            assert isinstance(loaded_json, dict)
-            except:
-                loaded_json = None
+            if is_json(raw_prompt_txt):
+                loaded_json = json.loads(raw_prompt_txt)
 
             if loaded_json is None:
                 if is_generating:
@@ -567,31 +585,30 @@ with shared.gradio_root:
 
         prompt.input(parse_meta, inputs=[prompt, state_is_generating], outputs=[prompt, generate_button, load_parameter_button], queue=False, show_progress=False)
 
-        load_parameter_button.click(modules.meta_parser.load_parameter_button_click, inputs=[prompt, state_is_generating], outputs=[
-            advanced_checkbox,
-            image_number,
-            prompt,
-            negative_prompt,
-            style_selections,
-            performance_selection,
-            aspect_ratios_selection,
-            overwrite_width,
-            overwrite_height,
-            sharpness,
-            guidance_scale,
-            adm_scaler_positive,
-            adm_scaler_negative,
-            adm_scaler_end,
-            base_model,
-            refiner_model,
-            refiner_switch,
-            sampler_name,
-            scheduler_name,
-            seed_random,
-            image_seed,
-            generate_button,
-            load_parameter_button
-        ] + lora_ctrls, queue=False, show_progress=False)
+        load_data_outputs = [advanced_checkbox, image_number, prompt, negative_prompt, style_selections,
+                             performance_selection, aspect_ratios_selection, overwrite_width, overwrite_height,
+                             sharpness, guidance_scale, adm_scaler_positive, adm_scaler_negative, adm_scaler_end,
+                             base_model, refiner_model, refiner_switch, sampler_name, scheduler_name, seed_random,
+                             image_seed, generate_button, load_parameter_button] + lora_ctrls
+
+        load_parameter_button.click(modules.meta_parser.load_parameter_button_click, inputs=[prompt, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=False)
+
+        def trigger_metadata_import(filepath, state_is_generating):
+            parameters, items, metadata_scheme = modules.metadata.read_info_from_image(filepath)
+
+            if parameters is None:
+                pass
+
+            if metadata_scheme is None and isinstance(parameters, dict):
+                metadata_scheme = modules.metadata.MetadataScheme.FOOOCUS.value
+
+
+            metadata_parser = modules.metadata.get_metadata_parser(metadata_scheme)
+            parsed_parameters = metadata_parser.parse_json(parameters)
+            return modules.meta_parser.load_parameter_button_click(parsed_parameters, state_is_generating)
+
+
+        metadata_import_button.click(trigger_metadata_import, inputs=[metadata_input_image, state_is_generating], outputs=load_data_outputs, queue=False, show_progress=False)
 
         generate_button.click(lambda: (gr.update(visible=True, interactive=True), gr.update(visible=True, interactive=True), gr.update(visible=False, interactive=False), [], True),
                               outputs=[stop_button, skip_button, generate_button, gallery, state_is_generating]) \