Create app.py

app.py

@@ -0,0 +1,126 @@
+import io
+import gradio as gr
+from PIL import Image, PngImagePlugin
+
+# --- Helper functions ---
+
+def load_image(file_obj):
+    """
+    Open the image and extract its 'parameters' metadata if available.
+    Only supports PNG images.
+    """
+    if file_obj is None:
+        return None, ""
+    try:
+        img = Image.open(file_obj)
+    except Exception as e:
+        return None, f"Error loading image: {e}"
+    if img.format != "PNG":
+        return None, "Error: Only PNG images are supported."
+    # Extract the 'parameters' text if it exists
+    metadata = img.info.get("parameters", "")
+    return img, metadata
+
+def update_png_metadata(img, new_metadata):
+    """
+    Prepare PNG metadata: copy existing text chunks (except 'parameters')
+    and add/update the 'parameters' key.
+    """
+    pnginfo = PngImagePlugin.PngInfo()
+    for key, value in img.info.items():
+        if key != "parameters":
+            pnginfo.add_text(key, value)
+    pnginfo.add_text("parameters", new_metadata)
+    return pnginfo
+
+def save_image_with_metadata(img, new_metadata):
+    """
+    Save the given image with updated metadata into a BytesIO stream,
+    then return the stream for download.
+    """
+    output = io.BytesIO()
+    pnginfo = update_png_metadata(img, new_metadata)
+    # Save with updated PNG text chunks
+    img.save(output, format="PNG", pnginfo=pnginfo)
+    output.seek(0)
+    return output
+
+def copy_metadata(metadata):
+    """
+    Return a cleaned version of metadata by removing the 'Used embeddings:' part
+    (if present). This mimics the original behavior.
+    """
+    # Look for a line starting with "Used embeddings:"
+    embed_index = metadata.find("\nUsed embeddings: ")
+    if embed_index < 0:
+        embed_index = metadata.find("\r\nUsed embeddings: ")
+    if embed_index > 0:
+        return metadata[:embed_index]
+    return metadata
+
+# --- Gradio Interface functions ---
+
+def load_image_wrapper(file_obj):
+    """
+    Loads the image and returns:
+      - an image preview (PIL Image),
+      - the metadata text for editing,
+      - and the original image saved in state.
+    """
+    img, meta = load_image(file_obj)
+    if img is None:
+        return None, meta, None
+    return img, meta, img
+
+def update_image(img, metadata):
+    """
+    Updates the image with new metadata and returns a file-like object for download.
+    """
+    if img is None:
+        return None
+    updated_img = save_image_with_metadata(img, metadata)
+    return updated_img
+
+def delete_metadata_wrapper(img):
+    """
+    Clears the metadata (i.e. sets it to empty) and returns both an empty string
+    (to clear the textbox) and an updated image file.
+    """
+    if img is None:
+        return "", None
+    updated_file = save_image_with_metadata(img, "")
+    return "", updated_file
+
+# --- Build the Gradio Blocks interface ---
+
+with gr.Blocks() as demo:
+    gr.Markdown("# SD EXIF Editor (Gradio Version)")
+    gr.Markdown("Upload a PNG image, view and edit the 'parameters' metadata, and download the updated image.")
+    
+    with gr.Row():
+        with gr.Column():
+            image_input = gr.Image(label="Upload PNG Image", type="file")
+            load_btn = gr.Button("Load Image")
+            image_preview = gr.Image(label="Image Preview")
+            download_btn = gr.Button("Download Updated Image")
+        with gr.Column():
+            metadata_box = gr.Textbox(label="Metadata (parameters)", lines=10, placeholder="Metadata will appear here...")
+            copy_btn = gr.Button("Copy Clean Metadata")
+            delete_btn = gr.Button("Delete Metadata")
+    
+    # A hidden state to store the PIL image object
+    image_state = gr.State(None)
+    
+    # When the user uploads and clicks load, update preview, metadata, and state.
+    load_btn.click(load_image_wrapper, inputs=image_input, outputs=[image_preview, metadata_box, image_state])
+    
+    # Download updated image (with new metadata) when the user clicks "Download Updated Image"
+    download_btn.click(update_image, inputs=[image_state, metadata_box], outputs=gr.File(label="Updated PNG Image"))
+    
+    # When the user clicks "Copy Clean Metadata," remove the "Used embeddings:" part.
+    copy_btn.click(lambda meta: copy_metadata(meta), inputs=metadata_box, outputs=metadata_box)
+    
+    # When the user clicks "Delete Metadata," clear the textbox and update the file.
+    delete_btn.click(delete_metadata_wrapper, inputs=image_state, outputs=[metadata_box, gr.File(label="Updated PNG Image")])
+    
+demo.launch()