1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54// Package generate deals with prompt generation.
package generate
import (
_ "embed"
"errors"
"strings"
"text/template"
)
// ErrNegativeInput happens when negative input occurs.
var ErrNegativeInput = errors.New("negative input")
// promptName is a name for the template below.
const promptName = "name_brainstorm_with_json"
//go:embed project.template
var promptTemplate string
// promptData is the input for the prompt template.
type promptData struct {
Purpose string // Project purpose
Theme string // Project theme
Demographics string // Target demographics
Interests string // Target interests
MaxSize int // Maximum number of names to generate
MaxLength int // Maximum length of each name
}
// Prompt generates the prompt for AI chatbots.
func Prompt(purpose, theme, demographics, interests string, size, length int) (string, error) {
if size < 0 || length < 0 {
return "", ErrNegativeInput
}
data := promptData{
Purpose: purpose,
Theme: theme,
Demographics: demographics,
Interests: interests,
MaxSize: size,
MaxLength: length,
}
tmpl, err := template.New(promptName).Parse(promptTemplate)
if err != nil {
return "", err
}
builder := strings.Builder{}
err = tmpl.Execute(&builder, data)
if err != nil {
return "", err
}
return strings.TrimSpace(builder.String()), err
}