atridad/himbot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed S3 upload names

Browse Source
This commit is contained in:
Atridad Lahiji
2024-01-19 14:33:37 -07:00
parent 96e8318f06
commit f31739a59b
2 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/replicate.go
View File

@ -54,7 +54,7 @@ func ReplicateTextGeneration(prompt string) (string, error) {
return result, nil
}
func ReplicateImageGeneration(prompt string) (*bytes.Buffer, error) {
func ReplicateImageGeneration(prompt string, filename string) (*bytes.Buffer, error) {
client, clientError := replicate.NewClient(replicate.WithTokenFromEnv())
if clientError != nil {
return nil, clientError
@ -102,10 +102,19 @@ func ReplicateImageGeneration(prompt string) (*bytes.Buffer, error) {
}
// Save image to a temporary file
tmpfile, err := os.CreateTemp("", "image.*.jpg")
var tmpfile *os.File
var err error
if filename != "" {
tmpfile, err = os.CreateTemp("", filename)
} else {
tmpfile, err = os.CreateTemp("", "image.*.jpg")
}
if err != nil {
log.Fatal(err)
}
defer os.Remove(tmpfile.Name())
if _, err := tmpfile.Write(imageBytes); err != nil {