fix(api): use correct case for PNG metadata

ssube · Aug 2, 2023 · dc9048c · dc9048c
1 parent ea99094
commit dc9048c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/api/onnx_web/output.py b/api/onnx_web/output.py
@@ -203,9 +203,9 @@ def save_image(
         exif = PngImagePlugin.PngInfo()
 
         if params is not None:
-            exif.add_text("Make", "onnx-web")
+            exif.add_text("make", "onnx-web")
             exif.add_text(
-                "Maker Note",
+                "maker note",
                 dumps(
                     json_params(
                         [output],
@@ -217,9 +217,9 @@ def save_image(
                     )
                 ),
             )
-            exif.add_text("Model", server.server_version)
+            exif.add_text("model", server.server_version)
             exif.add_text(
-                "Parameters",
+                "parameters",
                 str_params(server, params, size, inversions=inversions, loras=loras),
             )
 

diff --git a/gui/src/components/Profiles.tsx b/gui/src/components/Profiles.tsx
@@ -217,7 +217,7 @@ export function downloadParamsAsFile(data: DeepPartial<ImageMetadata>): void {
 export async function parseImageParams(file: File): Promise<DeepPartial<ImageMetadata>> {
   const tags = await ExifReader.load(file);
 
-  // handle lowercase variation from my earlier mistakes
+  // some parsers expect uppercase, some use lowercase, read both
   const makerNote = decodeTag(defaultTo(tags.MakerNote, tags['maker note']));
   // eslint-disable-next-line dot-notation, @typescript-eslint/strict-boolean-expressions
   const userComment = decodeTag(defaultTo(defaultTo(tags.UserComment, tags['Parameters']), tags['parameters']));