Merge pull request #6328 from lolsuffocate/fix-png-info-api

from secrets import compare_digest

import modules.shared as shared

from modules import sd_samplers, deepbooru, sd_hijack

from modules import sd_samplers, deepbooru, sd_hijack, images

from modules.api.models import *

from modules.processing import StableDiffusionProcessingTxt2Img, StableDiffusionProcessingImg2Img, process_images

from modules.extras import run_extras, run_pnginfo

from modules.extras import run_extras

from modules.textual_inversion.textual_inversion import create_embedding, train_embedding

from modules.textual_inversion.preprocess import preprocess

from modules.hypernetworks.hypernetwork import create_hypernetwork, train_hypernetwork

        if(not req.image.strip()):

            return PNGInfoResponse(info="")

        result = run_pnginfo(decode_base64_to_image(req.image.strip()))

        image = decode_base64_to_image(req.image.strip())

        if image is None:

            return PNGInfoResponse(info="")

        geninfo, items = images.read_info_from_image(image)

        if geninfo is None:

            geninfo = ""

        items = {**{'parameters': geninfo}, **items}

        return PNGInfoResponse(info=result[1])

        return PNGInfoResponse(info=geninfo, items=items)

    def progressapi(self, req: ProgressRequest = Depends()):

        # copy from check_progress_call of ui.py

    image: str = Field(title="Image", description="The base64 encoded PNG image")

class PNGInfoResponse(BaseModel):

    info: str = Field(title="Image info", description="A string with all the info the image had")

    info: str = Field(title="Image info", description="A string with the parameters used to generate the image")

    items: dict = Field(title="Items", description="An object containing all the info the image had")

class ProgressRequest(BaseModel):

    skip_current_image: bool = Field(default=False, title="Skip current image", description="Skip current image serialization")