/meta - Generation info extraction

[wizz13150]

Hey,

lol, it's me again.
So, doing this.
First cause it's annoying for me to extract them (I just thought that I should have looked for a website to do that, lol)
Then, because why not.

Just sharing what i'm doing.

• /Meta command to extract generation infos from a pic.
• 2 options: init_image & init_url

I may add some broken buttons to this, LOL !

Very first raw output of a functionnal command:

metacog.py :

import discord
from discord.ext import commands
from PIL import Image
from io import BytesIO
from typing import Optional
from discord import option
import requests

class MetaCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.slash_command(name='meta', description='Extract metadata from an image', guild_only=True)
    @option(
        'init_image',
        discord.Attachment,
        description='The image to extract metadata from.',
        required=False,
    )
    @option(
        'init_url',
        str,
        description='The URL image to extract metadata from. This overrides init_image!',
        required=False,
    )
    async def meta_handler(self, ctx, init_image: Optional[discord.Attachment] = None, init_url: Optional[str] = None):
        has_image = True
        if init_url:
            try:
                response = requests.get(init_url)
                image_data = BytesIO(response.content)
            except:
                await ctx.respond("Couldn't fetch image from the URL.", ephemeral=True)
                return
        elif init_image:
            image_data = BytesIO()
            await init_image.save(image_data)
        else:
            await ctx.respond("No image provided.", ephemeral=True)
            return
        
        image_data.seek(0)
        image = Image.open(image_data)
        metadata = image.info  # This should contain the metadata

        if metadata:
            metadata_str = ""
            for key, value in metadata.items():
                metadata_str += f"{key}: {value}\n"

            embed = discord.Embed(title='Image Metadata', description=metadata_str)
            await ctx.respond(embed=embed)
        else:
            await ctx.respond("No metadata found.", ephemeral=True)

def setup(bot):
    bot.add_cog(MetaCog(bot))

and self.load_extension('core.metacog') in aiya.py

I'll see what to do with this and post it later.
May be included in the /identify command.
I really don't know at this point.

Cheers ! 🥂

[wizz13150]

v0.2

Code :

import discord
from discord import ui
from discord.ext import commands
from PIL import Image
from io import BytesIO
from typing import Optional
from discord import option
import requests
import random


class MetaView(ui.View):
    def __init__(self, metadata_raw):
        super().__init__()
        self.metadata_raw = metadata_raw

    @ui.button(label='Draw From Meta', style=ui.ButtonStyle.primary)
    async def draw_from_meta(self, button: ui.Button, interaction: ui.Interaction):
        # Le code pour générer une image depuis les données métadonnées
        await interaction.response.send_message('Drawing from metadata... (this is a placeholder)')

    @ui.button(label='Copy Generation Datas', style=ui.ButtonStyle.secondary)
    async def copy_generation_data(self, button: ui.Button, interaction: ui.Interaction):
        # Le code pour copier les données métadonnées en format brut
        await interaction.response.send_message(f'Here are your raw generation data:```{self.metadata_raw}```')

class MetaCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    @commands.slash_command(name='meta', description='Extract Generation datas from an image', guild_only=True)
    @option(
        'init_image',
        discord.Attachment,
        description='The image to extract metadata from.',
        required=False,
    )
    @option(
        'init_url',
        str,
        description='The URL image to extract metadata from. This overrides init_image!',
        required=False,
    )
    async def meta_handler(self, ctx, init_image: Optional[discord.Attachment] = None, init_url: Optional[str] = None):
        has_image = True
        if init_url:
            try:
                response = requests.get(init_url)
                image_data = BytesIO(response.content)
            except:
                await ctx.respond("🚫 Couldn't fetch image from the URL.", ephemeral=True)
                return
        elif init_image:
            image_data = BytesIO()
            await init_image.save(image_data)
        else:
            await ctx.respond("🚫 No image provided.", ephemeral=True)
            return
        
        image_data.seek(0)
        image = Image.open(image_data)
        metadata = image.info.get('parameters', '')  # Assume the metadata you want is under the 'parameters' key
        
        if metadata:
            # Extract different parts as you specified
            prompt = metadata.split("Negative prompt:")[0].strip()
            negative_prompt = metadata.split("Negative prompt:")[1].split("Steps:")[0].strip()
            steps_and_beyond = metadata.split("Steps:")[1].strip()
            steps = steps_and_beyond.split(", ")[0]  # First value after "Steps:"
            
            # Process the remaining key-value pairs
            key_value_pairs = [seg.split(": ", 1) for seg in steps_and_beyond.split(", ") if ": " in seg]
            metadata_dict = {k: v for k, v in key_value_pairs}
            
            # Filter the keys you want
            keys_to_extract = ['Model', 'Seed', 'Steps', 'CFG scale', 'Sampler', 'Size', 'Clip skip']
            extracted_metadata = {k: metadata_dict.get(k, 'N/A') for k in keys_to_extract}
            
            # Create the Embed description from the extracted metadata
            extracted_metadata_str = f"🌟 **Model:** `{metadata_dict.get('Model', 'N/A')}`\n"
            extracted_metadata_str += f"🎨 **Prompt:**\n`{prompt}`\n🚫 **Negative Prompt:**\n`{negative_prompt}`\n"
            extracted_metadata_str += f"🌱 **Seed:** `{metadata_dict.get('Seed', 'N/A')}`\n🔢 **Steps:** `{steps}`\n🔍 **CFG scale:** `{metadata_dict.get('CFG scale', 'N/A')}`\n"
            for key in ['Sampler', 'Size', 'Clip skip']:
                extracted_metadata_str += f"🛠️ **{key}:** `{metadata_dict.get(key, 'N/A')}`\n"
            
            # Group the other keys and values into a single block
            other_metadata_str = "Model hash:" + steps_and_beyond.split("Model hash:")[1].strip()  # Start at "Model hash"
            other_metadata_str = other_metadata_str.replace(f"Steps: {steps}, ", "")
            other_metadata_str = other_metadata_str.rstrip(', ')  # Remove the last comma
            
            # Create an Embed with a random color
            embed = discord.Embed(title='🌠 Generation Datas 🌠', color=random.randint(0, 0xFFFFFF))
            embed.add_field(name='\u200b', value=extracted_metadata_str, inline=False)  # \u200b is a non-breaking space
            embed.add_field(name="🔍 Other Metadata 🔍", value=other_metadata_str, inline=False)
            
            # Rewind the BytesIO object to the beginning
            image_data.seek(0)
            file = discord.File(image_data, filename="preview.png")
            embed.set_thumbnail(url="attachment://preview.png")
            
            await ctx.respond(embed=embed, file=file)
        else:
            await ctx.respond("🚫 No metadata found. Please try again with a valid image! 📸", ephemeral=True)

def setup(bot):
    bot.add_cog(MetaCog(bot))

I'll add 3 buttons for this:

• Draw from Meta
• Copy Generation Datas
• Delete

[wizz13150]

v0.2.5

import discord
import requests
import random
from PIL import Image
from io import BytesIO
from typing import Optional
from discord import option, ui, Button, Embed
from discord.ext import commands
from discord.ui import Button, View


# Define a View for the metadata buttons
class MetaView(ui.View):
    def __init__(self, metadata_raw):
        super().__init__(timeout=None)
        self.metadata_raw = metadata_raw

    # Define a button for drawing from metadata
    @discord.ui.button(
        custom_id="button_draw_from_meta",
        emoji="🎨",
        label="Draw From Meta")
    async def draw_from_meta(self, button: ui.Button, interaction):
        await interaction.response.send_message('SOON !')

    # Define a button for copying generation data
    @discord.ui.button(
        custom_id="button_copy_generation_data_meta",
        emoji="📋",
        label="Copy Generation Datas")
    async def copy_generation_data(self, button: ui.Button, interaction):
        metadata_cleaned = self.metadata_raw.replace("\n", ", ")
        embed = Embed(title="──── Generation Datas────", description=f'```\nPrompt: {metadata_cleaned}\n```', color=random.randint(0, 0xFFFFFF))
        await interaction.response.send_message(embed=embed)

    # Define a button for deleting the metadata message
    @discord.ui.button(
        custom_id="delete_meta",
        emoji="❌",
        label="Delete")
    async def delete_meta(self, button: ui.Button, interaction):
        await self.message.delete()


# Define the MetaCog class
class MetaCog(commands.Cog):
    def __init__(self, bot):
        self.bot = bot

    # Define a slash command for extracting metadata
    @commands.slash_command(name='meta', description='Extract Generation datas from an image', guild_only=True)
    @option(
        'init_image',
        discord.Attachment,
        description='The image to extract metadata from.',
        required=False,
    )
    @option(
        'init_url',
        str,
        description='The URL image to extract metadata from. This overrides init_image!',
        required=False,
    )
    async def meta_handler(self, ctx, init_image: Optional[discord.Attachment] = None, init_url: Optional[str] = None):
        has_image = True
        if init_url:
            try:
                response = requests.get(init_url)
                image_data = BytesIO(response.content)
            except:
                await ctx.respond("🚫 Couldn't fetch image from the URL.", ephemeral=True)
                return
        elif init_image:
            image_data = BytesIO()
            await init_image.save(image_data)
        else:
            await ctx.respond("🚫 No image provided.", ephemeral=True)
            return
        
        image_data.seek(0)
        image = Image.open(image_data)
        metadata = image.info.get('parameters', '')  # Assume the metadata you want is under the 'parameters' key
        
        if metadata:
            # Extract different parts as you specified
            prompt = metadata.split("Negative prompt:")[0].strip()
            negative_prompt = metadata.split("Negative prompt:")[1].split("Steps:")[0].strip()
            steps_and_beyond = metadata.split("Steps:")[1].strip()
            steps = steps_and_beyond.split(", ")[0]  # First value after "Steps:"
            
            # Process the remaining key-value pairs
            key_value_pairs = [seg.split(": ", 1) for seg in steps_and_beyond.split(", ") if ": " in seg]
            metadata_dict = {k: v for k, v in key_value_pairs}
            
            # Filter the keys you want
            keys_to_extract = ['Model', 'Seed', 'Steps', 'CFG scale', 'Sampler', 'Size', 'Clip skip']
            extracted_metadata = {k: metadata_dict.get(k, 'N/A') for k in keys_to_extract}
            
            # Create the Embed description from the extracted metadata
            extracted_metadata_str = f"🌟 **Model:** `{metadata_dict.get('Model', 'N/A')}`\n"
            extracted_metadata_str += f"🎨 **Prompt:**\n`{prompt}`\n🚫 **Negative Prompt:**\n`{negative_prompt}`\n"
            extracted_metadata_str += f"🌱 **Seed:** `{metadata_dict.get('Seed', 'N/A')}`\n🔢 **Steps:** `{steps}`\n🔍 **CFG scale:** `{metadata_dict.get('CFG scale', 'N/A')}`\n"
            for key in ['Sampler', 'Size', 'Clip skip']:
                extracted_metadata_str += f"🛠️ **{key}:** `{metadata_dict.get(key, 'N/A')}`\n"
            
            # Group the other keys and values into a single block
            other_metadata_str = "Model hash:" + steps_and_beyond.split("Model hash:")[1].strip()  # Start at "Model hash"
            other_metadata_str = other_metadata_str.replace(f"Steps: {steps}, ", "")
            other_metadata_str = other_metadata_str.rstrip(', ')  # Remove the last comma
            
            # Create an Embed with a random color
            embed = discord.Embed(title="──── Generation Datas ────", color=random.randint(0, 0xFFFFFF))
            embed.add_field(name='\u200b', value=extracted_metadata_str, inline=False)  # \u200b is a non-breaking space
            embed.add_field(name="🔍 Other Metadata 🔍", value=other_metadata_str, inline=False)
            
            # Rewind the BytesIO object to the beginning
            image_data.seek(0)
            file = discord.File(image_data, filename="preview.png")
            embed.set_thumbnail(url="attachment://preview.png")
            
            # Create an instance of MetaView and pass the raw metadata
            view = MetaView(metadata)

            # Add the view when you send the message
            await ctx.respond(embed=embed, file=file, view=view)
        else:
            await ctx.respond("🚫 No metadata found. Please try again with a valid image! 📸", ephemeral=True)

def setup(bot):
    bot.add_cog(MetaCog(bot))

Still broken buttons, ofc. No surprise here, until i found how to fix this for /generate too.