more on this

1 files changed, 13 insertions, 0 deletions

diff --git a/internal/image/openai.go b/internal/image/openai.go
index 637b8a1..d964b55 100644
--- a/internal/image/openai.go
+++ b/internal/image/openai.go
@@ -23,6 +23,9 @@ type OpenAIClient struct {
 	quality    string // standard or hd (dall-e-3 only)
 	style      string // natural or vivid (dall-e-3 only)
 	lastPrompt string // Store the last used prompt for attribution
+
+	// PromptCallback is called when the prompt is generated, before the image is created
+	PromptCallback func(prompt string)
 }
 
 // OpenAIConfig holds configuration for the OpenAI image provider
@@ -120,6 +123,11 @@ func (c *OpenAIClient) Search(ctx context.Context, opts *SearchOptions) ([]Searc
 	// Store the prompt for attribution
 	c.lastPrompt = prompt
 
+	// Call the callback if set
+	if c.PromptCallback != nil {
+		c.PromptCallback(prompt)
+	}
+
 	// Log the prompt to stdout for debugging
 	fmt.Printf("OpenAI Image Generation Prompt (%d chars): %s\n", len(prompt), prompt)
 	fmt.Printf("OpenAI Image Generation: Using model '%s' with size '%s'\n", c.model, c.size)
@@ -220,6 +228,11 @@ func (c *OpenAIClient) GetLastPrompt() string {
 	return c.lastPrompt
 }
 
+// SetPromptCallback sets a callback function that will be called when the prompt is generated
+func (c *OpenAIClient) SetPromptCallback(callback func(prompt string)) {
+	c.PromptCallback = callback
+}
+
 // createEducationalPrompt generates a prompt optimized for language learning
 func (c *OpenAIClient) createEducationalPrompt(bulgarianWord, englishTranslation string) string {
 	// Generate a scene description for the word