ImageTypes.h

// OpenAI, Copyright LifeEXE. All Rights Reserved.
 
#pragma once
 
#include "CoreMinimal.h"
#include "Provider/OpenAIOptional.h"
#include "ImageTypes.generated.h"
 
UENUM(BlueprintType)
enum class EImageModelEnum : uint8
{
    DALL_E_2,
    DALL_E_3
};
 
UENUM(BlueprintType)
enum class EImageSizeDalle2 : uint8
{
    Size_256x256,
    Size_512x512,
    Size_1024x1024
};
 
UENUM(BlueprintType)
enum EImageSizeDalle3 : uint8
{
    Size_1024x1024,
    Size_1792x1024,
    Size_1024x1792
};
 
UENUM(BlueprintType)
enum EOpenAIImageQuality
{
    Standard,
    HD
};
 
UENUM(BlueprintType)
enum EOpenAIImageStyle
{
    // Vivid causes the model to lean towards generating hyper - real and dramatic images.
    Vivid,
    // Natural causes the model to produce more natural, less hyper - real looking images.
    Natural
};
 
UENUM(BlueprintType)
enum class EOpenAIImageFormat : uint8
{
    URL,
    B64_JSON
};
 
//                 REQUEST TYPES
 
USTRUCT(BlueprintType)
struct FOpenAIImage
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Required")
    FString Prompt;
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Model{"dall-e-2"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 N{1};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Quality{"standard"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Response_Format{"url"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Size{"1024x1024"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Style{"vivid"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FOptionalString User;
};
 
USTRUCT(BlueprintType)
struct FOpenAIImageEdit
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Required")
    FString Image;
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Required")
    FString Prompt;
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Mask;
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Model{"dall-e-2"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 N{1};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Size{"1024x1024"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Response_Format{"url"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FOptionalString User;
};
 
USTRUCT(BlueprintType)
struct FOpenAIImageVariation
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Required")
    FString Image;
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Model{"dall-e-2"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    int32 N{1};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Response_Format{"url"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FString Size{"1024x1024"};
 
    UPROPERTY(BlueprintReadWrite, Category = "OpenAI | Optional")
    FOptionalString User;
};
 
//                 RESPONSE TYPES
 
USTRUCT(BlueprintType)
struct FImageObject
{
    GENERATED_BODY()
 
    
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString B64_JSON;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString URL;
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    FString Revised_Prompt;
};
 
USTRUCT(BlueprintType)
struct FImageResponse
{
    GENERATED_BODY()
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    int32 Created{};
 
    UPROPERTY(BlueprintReadOnly, Category = "OpenAI")
    TArray<FImageObject> Data;
};
 
USTRUCT(BlueprintType)
struct FImageEditResponse : public FImageResponse
{
    GENERATED_BODY()
};
 
USTRUCT(BlueprintType)
struct FImageVariationResponse : public FImageResponse
{
    GENERATED_BODY()
};