dev(narugo): add empty metadata support for a41 webui images

deepghs · Sep 7, 2024 · c02b2b2 · c02b2b2
1 parent a1720fd
commit c02b2b2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/imgutils/sd/metadata.py b/imgutils/sd/metadata.py
@@ -195,7 +195,13 @@ def parse_sdmeta_from_text(x: str) -> SDMetaData:
         <class 'imgutils.sd.metadata.SDMetaData'>
     """
     x = textwrap.dedent(x).strip()
-    *prompt_lines, argument_line = x.splitlines(keepends=False)
+    all_lines = x.splitlines(keepends=False)
+    if len(all_lines) >= 1:
+        prompt_lines = all_lines[:-1]
+        argument_line = all_lines[-1]
+    else:
+        prompt_lines = []
+        argument_line = ''
     if len(_PARAM_PATTERN.findall(argument_line)) < 3:
         prompt_lines.append(argument_line)
         argument_line = ''

diff --git a/test/sd/test_metadata.py b/test/sd/test_metadata.py
@@ -397,3 +397,6 @@ def _test_pnginfo(pinfo):
             assert get_sdmeta_from_image(f) == sdimg_3_std
         with _test_pnginfo(sdimg_4_std.pnginfo) as f:
             assert get_sdmeta_from_image(f) == sdimg_4_std
+
+    def test_empty_info_parse(self):
+        assert parse_sdmeta_from_text('') == SDMetaData('', '', {})