mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
refactor: rename parse_json to to_json, rename parse_string to to_string
This commit is contained in:
@@ -261,11 +261,11 @@ class MetadataParser(ABC):
|
|||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
@abstractmethod
|
@abstractmethod
|
||||||
def parse_json(self, metadata: dict | str) -> dict:
|
def to_json(self, metadata: dict | str) -> dict:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
@abstractmethod
|
@abstractmethod
|
||||||
def parse_string(self, metadata: dict) -> str:
|
def to_string(self, metadata: dict) -> str:
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
def set_data(self, raw_prompt, full_prompt, raw_negative_prompt, full_negative_prompt, steps, base_model_name,
|
def set_data(self, raw_prompt, full_prompt, raw_negative_prompt, full_negative_prompt, steps, base_model_name,
|
||||||
@@ -328,7 +328,7 @@ class A1111MetadataParser(MetadataParser):
|
|||||||
'version': 'Version'
|
'version': 'Version'
|
||||||
}
|
}
|
||||||
|
|
||||||
def parse_json(self, metadata: str) -> dict:
|
def to_json(self, metadata: str) -> dict:
|
||||||
metadata_prompt = ''
|
metadata_prompt = ''
|
||||||
metadata_negative_prompt = ''
|
metadata_negative_prompt = ''
|
||||||
|
|
||||||
@@ -422,7 +422,7 @@ class A1111MetadataParser(MetadataParser):
|
|||||||
|
|
||||||
return data
|
return data
|
||||||
|
|
||||||
def parse_string(self, metadata: dict) -> str:
|
def to_string(self, metadata: dict) -> str:
|
||||||
data = {k: v for _, k, v in metadata}
|
data = {k: v for _, k, v in metadata}
|
||||||
|
|
||||||
width, height = eval(data['resolution'])
|
width, height = eval(data['resolution'])
|
||||||
@@ -502,7 +502,7 @@ class FooocusMetadataParser(MetadataParser):
|
|||||||
def get_scheme(self) -> MetadataScheme:
|
def get_scheme(self) -> MetadataScheme:
|
||||||
return MetadataScheme.FOOOCUS
|
return MetadataScheme.FOOOCUS
|
||||||
|
|
||||||
def parse_json(self, metadata: dict) -> dict:
|
def to_json(self, metadata: dict) -> dict:
|
||||||
for key, value in metadata.items():
|
for key, value in metadata.items():
|
||||||
if value in ['', 'None']:
|
if value in ['', 'None']:
|
||||||
continue
|
continue
|
||||||
@@ -517,7 +517,7 @@ class FooocusMetadataParser(MetadataParser):
|
|||||||
|
|
||||||
return metadata
|
return metadata
|
||||||
|
|
||||||
def parse_string(self, metadata: list) -> str:
|
def to_string(self, metadata: list) -> str:
|
||||||
for li, (label, key, value) in enumerate(metadata):
|
for li, (label, key, value) in enumerate(metadata):
|
||||||
# remove model folder paths from metadata
|
# remove model folder paths from metadata
|
||||||
if key.startswith('lora_combined_'):
|
if key.startswith('lora_combined_'):
|
||||||
|
|||||||
@@ -27,7 +27,7 @@ def log(img, metadata, metadata_parser: MetadataParser | None = None, output_for
|
|||||||
date_string, local_temp_filename, only_name = generate_temp_filename(folder=path_outputs, extension=output_format)
|
date_string, local_temp_filename, only_name = generate_temp_filename(folder=path_outputs, extension=output_format)
|
||||||
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
|
os.makedirs(os.path.dirname(local_temp_filename), exist_ok=True)
|
||||||
|
|
||||||
parsed_parameters = metadata_parser.parse_string(metadata.copy()) if metadata_parser is not None else ''
|
parsed_parameters = metadata_parser.to_string(metadata.copy()) if metadata_parser is not None else ''
|
||||||
image = Image.fromarray(img)
|
image = Image.fromarray(img)
|
||||||
|
|
||||||
if output_format == OutputFormat.PNG.value:
|
if output_format == OutputFormat.PNG.value:
|
||||||
|
|||||||
@@ -713,7 +713,7 @@ with shared.gradio_root:
|
|||||||
parsed_parameters = {}
|
parsed_parameters = {}
|
||||||
else:
|
else:
|
||||||
metadata_parser = modules.meta_parser.get_metadata_parser(metadata_scheme)
|
metadata_parser = modules.meta_parser.get_metadata_parser(metadata_scheme)
|
||||||
parsed_parameters = metadata_parser.parse_json(parameters)
|
parsed_parameters = metadata_parser.to_json(parameters)
|
||||||
|
|
||||||
return modules.meta_parser.load_parameter_button_click(parsed_parameters, state_is_generating)
|
return modules.meta_parser.load_parameter_button_click(parsed_parameters, state_is_generating)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user