Update app.py

app.py

@@ -115,8 +115,8 @@ def realesrgan(img, model_name, denoise_strength, face_enhance, outscale, ext):
 
         #out_filename = f"output_{rnd_string(8)}.{extension}"
         out_filename = f"output_{rnd_string(8)}." + ext
-        metadata = PngInfo()
         if parameters != '':
+            metadata = PngInfo()
             metadata.add_text("parameters", parameters)
             cv2.imwrite(out_filename, output, [int(cv2.IMWRITE_PNG_COMPRESSION), 0], pnginfo=metadata)
         else:
@@ -183,6 +183,9 @@ def image_properties(img):
         properties = f"Width: {img.size[0]}, Height: {img.size[1]}  |  Color Mode: {img_mode}"
         return properties
 
+def image_metadata(img):
+    return img.text['parameters']
+
 
 def main():
     # Gradio Interface
@@ -212,12 +215,14 @@ def main():
             with gr.Group():
                 input_image = gr.Image(label="Source Image", type="pil", image_mode="RGBA")
                 input_image_properties = gr.Textbox(label="Image Properties", max_lines=1)
+                input_image_metadata = gr.Textbox(label="Image Metadata", max_lines=1)
             output_image = gr.Image(label="Restored Image", image_mode="RGBA", show_share_button=False)
         with gr.Row():
             restore_btn = gr.Button("Upscale")
 
         # Event listeners:
         input_image.change(fn=image_properties, inputs=input_image, outputs=input_image_properties)
+        input_image.change(fn=image_metadata, inputs=input_image, outputs=input_image_metadata)
         restore_btn.click(fn=realesrgan,
                           inputs=[input_image, model_name, denoise_strength, face_enhance, outscale, ext],
                           outputs=output_image,