Updated

lib/prodia/prodia.go

@@ -22,15 +22,15 @@ func getProdiaKey() string {
 	return os.Getenv("PRODIA_KEY")
 }
 
-func RequestGeneration() (*rtypes.ProdiaGenerateResponse, error) {
+func RequestGeneration(prompt string, negative string) (*rtypes.ProdiaGenerateResponse, error) {
 	req := rtypes.ProdiaGenerateRequest{
 		Model:          rtypes.Deliberate_v2,
-		Prompt:         "beautiful woman wearing military clothing, woman, girl, beautiful, masterpiece, pretty, blue eyes, military, uniform, army",
-		NegativePrompt: "man, ugly, destroyed, nsfw, nudity",
-		Steps:          25,
+		Prompt:         prompt,
+		NegativePrompt: negative,
+		Steps:          35,
 		CFGScale:       6.5,
 		Seed:           -1,
-		Upscale:        true,
+		Upscale:        false,
 		Sampler:        rtypes.DPMpp2MK,
 		AspectRatio:    rtypes.Square,
 	}
@@ -121,7 +121,6 @@ func FetchGeneration(jobId string) (*rtypes.ProdiaRetrieveResponseX, error) {
 			return nil, errors.New("API Error")
 		}
 
-		fmt.Println("Not done yet, waiting... ::" + responseJson.Status)
 		time.Sleep(time.Second)
 	}
 

lib/rtypes/prodia_request.go

@@ -42,13 +42,13 @@ const (
 
 type ProdiaGenerateRequest struct {
 	Model          ProdiaModel       `json:"model"`
-	Prompt         ProdiaSampler     `json:"prompt"`
+	Prompt         string            `json:"prompt"`
 	NegativePrompt string            `json:"negative_prompt"`
 	Steps          int               `json:"steps"`
 	CFGScale       float32           `json:"cfg_scale"`
 	Seed           int64             `json:"seed"`
 	Upscale        bool              `json:"upscale"`
-	Sampler        string            `json:"sampler"`
+	Sampler        ProdiaSampler     `json:"sampler"`
 	AspectRatio    ProdiaAspectRatio `json:"aspect_ratio"`
 }