Update app3.py

app3.py

@@ -15,8 +15,11 @@ token = '5UAYO8UWHNQKT3UUS9H8V360L76MD72DRIUY9QC2'
 ##############################################################
 #################################################
 
-def SD_call(image_prompt, age, color, hair_color):
+def SD_call(image_prompt, age, color, hair_color,NSFW):
 
+
+    postive = "clothes"
+    negative = "naked, nsfw, porn"
     serverless_api_id = '3g77weiulabzuk'
     # Define the URL you want to send the request to
     url = f"https://api.runpod.ai/v2/{serverless_api_id}/runsync"
@@ -30,6 +33,9 @@ def SD_call(image_prompt, age, color, hair_color):
     
     # Define your data (this could also be a JSON payload)
     print("SD_processing")
+    if NSFW:
+        postive = "naked, nsfw"
+        negative = "clothes"
     data = {
         "input": {
             "api": {
@@ -44,8 +50,8 @@ def SD_call(image_prompt, age, color, hair_color):
                 "override_settings_restore_afterwards": True,
                 "refiner_checkpoint": "",
                 "refiner_switch_at": 0.8,
-                "prompt": f"masterpiece, best quality, 8k, (looking at viewer:1.1), gorgeous, hot, seductive, {age} years old american {color} woman, (eye contact:1.1), beautiful face, hyper detailed, best quality, ultra high res, {hair_color} hair,blue eyes, photorealistic, high resolution, detailed, raw photo, 1girl,{image_prompt} ",
-                "negative_prompt": "EasyNegative, fat, paintings, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, ((monochrome)), ((grayscale)), bad anatomy, text, error, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, bad feet, poorly drawn face, bad proportions, gross proportions, ng_deepnegative_v1_75t, badhandsv5-neg, clothes",
+                "prompt": f"masterpiece, best quality, 8k, (looking at viewer:1.1), gorgeous, hot, seductive, {age} years old american {color} woman, (eye contact:1.1), beautiful face, hyper detailed, best quality, ultra high res, {hair_color} hair,blue eyes, photorealistic, high resolution, detailed, raw photo, 1girl,{image_prompt}, {positive} ",
+                "negative_prompt": f"EasyNegative, fat, paintings, sketches, (worst quality:2), (low quality:2), (normal quality:2), lowres, ((monochrome)), ((grayscale)), bad anatomy, text, error, cropped, worst quality, low quality, normal quality, jpeg artifacts, signature, watermark, username, blurry, bad feet, poorly drawn face, bad proportions, gross proportions, ng_deepnegative_v1_75t, badhandsv5-neg, {negative}",
                 "seed": -1,
                 "batch_size": 1,
                 "steps": 30,
@@ -197,9 +203,9 @@ def Chat_call(chat,temperature):
 
 
 
-def greet(chat, image_prompt, temperature, age, color, hair_color):
+def greet(chat, image_prompt, temperature, age, color, hair_color,NSFW):
     if image_prompt.strip():
-        image_path = SD_call(image_prompt,age, color, hair_color)
+        image_path = SD_call(image_prompt,age, color, hair_color,NSFW)
         return "Image generated successfully", image_path
 
     if temperature > 3:
@@ -216,7 +222,8 @@ demo = gr.Interface(
         gr.Slider(label="Text temperature", value=1, minimum=0, maximum=2),
         gr.Slider(label="Age", value=22, minimum=18, maximum=75),
         gr.Dropdown(["asian", "white", "black", "latina"], label="Color", info="Will add more later!"),
-        gr.Dropdown(["blond", "brune", "red", "white", "pink", "black", "blue", "green"], label="Hair color", info="Blond is cool")
+        gr.Dropdown(["blond", "brune", "red", "white", "pink", "black", "blue", "green"], label="Hair color", info="Blond is cool"),
+        gr.Checkbox(label="NSFW", info="👀👀👀")
     ],
     flagging_options=["blurry", "incorrect", "other"],
     outputs=[gr.Textbox(label="Answer", lines=3), gr.Image(label="Generated Image", type="filepath")],