From 5c8ca65904a5f6107ae64868d4e81bf2175b9bad Mon Sep 17 00:00:00 2001
From: zanllp <qc@zanllp.cn>
Date: Wed, 27 Mar 2024 21:03:59 +0800
Subject: [PATCH] Align metadata generated by various parsers

---
 scripts/iib/db/update_image_data.py | 1 +
 scripts/iib/parsers/fooocus.py      | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/scripts/iib/db/update_image_data.py b/scripts/iib/db/update_image_data.py
index 615b030..7f5c2a8 100644
--- a/scripts/iib/db/update_image_data.py
+++ b/scripts/iib/db/update_image_data.py
@@ -96,6 +96,7 @@ def update_image_data(search_dirs: List[str], is_rebuild = False):
                         "Source Identifier",
                         "Postprocess upscale by",
                         "Postprocess upscaler",
+                        "Size"
                     ]:
                         v = meta.get(k)
                         if not v:
diff --git a/scripts/iib/parsers/fooocus.py b/scripts/iib/parsers/fooocus.py
index 93120a6..af9e626 100644
--- a/scripts/iib/parsers/fooocus.py
+++ b/scripts/iib/parsers/fooocus.py
@@ -49,7 +49,8 @@ class FooocusParser:
                     v = v.replace(",", "，")
                     params["meta"][k] = v
                     metadata_list_str += f" {k}: {v},"
-
+        params["meta"]["Model"] = params["meta"]["Base Model"]
+        params["meta"]["Size"] = str(params["meta"]["Resolution"]).replace("(", "").replace(")", "").replace("，", " * ")
         metadata_list_str = metadata_list_str.strip()
         info = f"""{params['Prompt']}\nNegative prompt: {params['Negative Prompt']}\n{metadata_list_str}""".strip()
         return ImageGenerationInfo(