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)
 	}