diff --git a/main.go b/main.go index fd025f0..98a62e4 100644 --- a/main.go +++ b/main.go @@ -13,7 +13,7 @@ import ( ) const ( - NEGATIVE_PROMPT = "(nsfw:1.1), (porn:1.1), (naked:1.1), (nude:1.1), (nipple:1.1), (penis:1.1), (dick:1.1), (vagina:1.1), (asshole:1.1), visible nipple, nsfl, not safe for work, nudity, artifact, deformed, multiple limbs, ugly, gore, blood, sex, pornography, penis, dick, genitalia, male genitalia, anus, penetration, double penetration, cock" + NEGATIVE_PROMPT = "easynegative, (nsfw:1.1), (porn:1.1), (naked:1.1), (nude:1.1), (nipple:1.1), (penis:1.1), (dick:1.1), (vagina:1.1), (asshole:1.1), visible nipple, nsfl, not safe for work, nudity, artifact, deformed, multiple limbs, ugly, gore, blood, sex, pornography, penis, dick, genitalia, male genitalia, anus, penetration, double penetration, cock" ) var nodes [1]rtypes.SDNode = [...]rtypes.SDNode{ @@ -81,13 +81,14 @@ func main() { func generateProdiaImage(c *gin.Context) { imageDescriptior := c.PostForm("image_description") + negativePrompt := c.PostForm("negative_prompt") if goaway.IsProfane(imageDescriptior) { imageDescriptior = goaway.Censor(imageDescriptior) fmt.Printf("Found profanity, censoring to: %s\n", imageDescriptior) } - resp, err := prodia.RequestGeneration(imageDescriptior, NEGATIVE_PROMPT) + resp, err := prodia.RequestGeneration(imageDescriptior, fmt.Sprintf("%s,%s", negativePrompt, NEGATIVE_PROMPT)) if err != nil { c.AbortWithError(500, err) return diff --git a/templates/home/index.html b/templates/home/index.html index 4e76967..9628032 100644 --- a/templates/home/index.html +++ b/templates/home/index.html @@ -62,7 +62,11 @@
- + Prompt + + + Negative Prompt +