summaryrefslogtreecommitdiff
path: root/internal/image/gemini_attribution.go
diff options
context:
space:
mode:
Diffstat (limited to 'internal/image/gemini_attribution.go')
-rw-r--r--internal/image/gemini_attribution.go6
1 files changed, 5 insertions, 1 deletions
diff --git a/internal/image/gemini_attribution.go b/internal/image/gemini_attribution.go
index 2aefeac..15d1bb1 100644
--- a/internal/image/gemini_attribution.go
+++ b/internal/image/gemini_attribution.go
@@ -22,7 +22,11 @@ func (c *GeminiProvider) buildAttribution(result *SearchResult, prompt string) s
attribution.WriteString("Image generated by Google Gemini Nano Banana\n\n")
fmt.Fprintf(&attribution, "Model: %s\n", c.modelName())
fmt.Fprintf(&attribution, "Text model: %s\n", c.textModelName())
- fmt.Fprintf(&attribution, "Aspect ratio: %s\n", geminiAspectRatio)
+ aspectRatio := geminiAspectRatio
+ if c != nil && c.config != nil && strings.TrimSpace(c.config.AspectRatio) != "" {
+ aspectRatio = strings.TrimSpace(c.config.AspectRatio)
+ }
+ fmt.Fprintf(&attribution, "Aspect ratio: %s\n", aspectRatio)
fmt.Fprintf(&attribution, "Size: %dx%d\n", result.Width, result.Height)
if result.Description != "" {
fmt.Fprintf(&attribution, "Result: %s\n", result.Description)