ai

type GCPClient struct {
	PredictionClient   *aiplatform.PredictionClient
	TextToSpeechClient *texttospeech.Client
	PredictURL         string
}

type GPTResponse struct {
	Choices []struct {
		Message struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"message"`
		FinishReason string `json:"finish_reason"`
		Index        int    `json:"index"`
	} `json:"choices"`
	Data []struct {
		URL string `json:"url"`
	} `json:"data"`
}

type OpenAIClient struct {
	OpenAIKey string
}

type PereBodulClient interface {
	GenerateStory(ctx context.Context, params StoryParams) (string, error)
	GenerateAudio(ctx context.Context, params StoryParams, story string) ([]byte, error)
	GenerateImage(ctx context.Context, params StoryParams, imagePrompt string) (string, error)
	GenerateImagePrompt(ctx context.Context, params StoryParams, story string) (string, error)
}

type StoryParams struct {
	Hero     string
	Villain  string
	Location string
	Objects  string
	Wizard   Wizard
}

type VertexAIClient struct {
	GenAIClient        *genai.Client
	TextToSpeechClient *texttospeech.Client
}

type Wizard string