Update app3.py

app3.py

@@ -12,13 +12,13 @@ token = '5UAYO8UWHNQKT3UUS9H8V360L76MD72DRIUY9QC2'
 
     
 
-##############################################################
+#############################################################
 #################################################
 
-def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hair_texture,skin_details,eye_colors,NSFW):
+def SD_call(prompt, image_prompt, age, weight,  w_w, color, hair_color,h_c_w, hair_length, h_l_w,hair_texture, h_t_w,skin_details, eye_colors, e_c_w,NSFW):
 
     positive = "clothes"
-    negative = "naked, nsfw, porn"
+    negative = "(((naked))), (((nsfw))), porn"
     serverless_api_id = '7c9nnp0b3ordr8'
     # Define the URL you want to send the request to
     url = f"https://api.runpod.ai/v2/{serverless_api_id}/runsync"
@@ -30,9 +30,15 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
         "Content-Type": "application/json"
     }
 
+    weight = f"{'(' * w_w}{weight} woman{')' * w_w}"
+    hair_color=f"{'(' * h_c_w}{hair_color} hair{')' * h_c_w}"
+    hair_length=f"{'(' * h_l_w}{hair_length} hair{')' * h_l_w}"
+    hair_texture =f"{'(' * h_t_w}{hair_texture} hair{')' * h_t_w}"
+    eye_colors=f"{'(' * e_c_w}{eye_colors} eyes{')' * e_c_w}"
+
     if NSFW == True:
-        positive = "naked, nsfw"
-        negative = "clothes"
+        positive = "((naked)), ((nsfw))"
+        negative = "((clothes))"
 
     if prompt.strip():
         total_prompt = prompt
@@ -40,7 +46,7 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
     else:
         color = ", ".join(color)
         skin_details = ", ".join(skin_details)
-        total_prompt = f"masterpiece, best quality, 8k, (looking at viewer:1.1), gorgeous, hot, seductive, {age} years old american {color} woman, {weight} woman, (eye contact:1.1), beautiful face, hyper detailed, best quality, ultra high res, {hair_length} {hair_color} {hair_texture} hair,{eye_colors} eyes, {skin_details} photorealistic, high resolution, detailed, raw photo, 1girl,{image_prompt}, amateur cellphone photography. f8.0, samsung galaxy, noise, jpeg artefacts, poor lighting, low light, underexposed, high contrast "
+        total_prompt = f"masterpiece, best quality, 8k, (looking at viewer:1.1), gorgeous, hot, seductive, {age} years old american {color} woman, {weight}, (eye contact:1.1), beautiful face, hyper detailed, best quality, ultra high res, {hair_length}, {hair_color}, {hair_texture},{eye_colors}, {skin_details}, photorealistic, high resolution, detailed, raw photo, 1girl,{image_prompt}, amateur cellphone photography. f8.0, samsung galaxy, noise, jpeg artefacts, poor lighting, low light, underexposed, high contrast "
 
     
     # Define your data (this could also be a JSON payload)
@@ -77,11 +83,11 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
     data = {
     "input": {
         "prompt": f"{total_prompt}, {positive}",
-        "negative_prompt": f"EasyNegative, fat, paintings, sketches, lowres, ((monochrome)), ((grayscale)), bad anatomy, text, error, cropped, signature, watermark, username, blurry, bad feet, poorly drawn face, bad proportions, gross proportions, ng_deepnegative_v1_75t, badhandsv5-neg, {negative}",
+        "negative_prompt": f"paintings, sketches, lowres, ((monochrome)), ((grayscale)), bad anatomy, text, error, cropped, signature, watermark, username, blurry, bad feet, poorly drawn face, bad proportions, gross proportions, badhandsv5-neg, {negative}",
         "width": 512,
         "height": 720,
         "guidance_scale": 7.5,
-        "num_inference_steps": 50,
+        "num_inference_steps": 30,
         "num_outputs": 1,
         "prompt_strength": 0.8,
         "scheduler": "K-LMS"
@@ -138,13 +144,12 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
 
 
 
-def greet(prompt, image_prompt, age, weight, color, hair_color,hair_length,hair_texture,skin_details,eye_colors,NSFW):
-    image_path = SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hair_texture,skin_details,eye_colors,NSFW)
-    #return "Image generated successfully", image_path
+def greet(prompt, image_prompt, age, weight, w_w, color, hair_color, h_c_w, hair_length, h_l_w,hair_texture, h_t_w,skin_details, eye_colors, e_c_w,NSFW):
+    image_path = SD_call(prompt, image_prompt, age, weight,  w_w, color, hair_color,h_c_w, hair_length, h_l_w,hair_texture, h_t_w,skin_details, eye_colors, e_c_w,NSFW)
     if image_path is not None:
-        return "Image generated successfully", image_path
+        return image_path
     else:
-        return "No images found in the response.", None
+        return None
 
 
 demo = gr.Interface(
@@ -154,16 +159,21 @@ demo = gr.Interface(
         gr.Textbox(label="Girl_prompt", lines=3),
         gr.Slider(label="Age", value=22, minimum=18, maximum=75),
         gr.Radio(["skinny", "slim", "athletic", "muscular", "average", "curvy", "chubby", "overweight", "obese"],label="Body Type",type="value"),
+        gr.Slider(label="Body Type weight", value=2, minimum=1, maximum=4, step=1),
         gr.CheckboxGroup(choices=["asian", "white", "black", "latina", "middle eastern","indigenous", "Mixed"],label="Color",type="value"),
         gr.Radio(["black", "brown", "brunette", "dark brown", "light brown", "blonde", "dirty blonde", "platinum blonde", "red", "auburn", "ginger", "strawberry blonde", "gray", "silver", "white", "blue", "green", "purple", "pink", "rainbow", "multicolored"],label="Hair Color",type="value"),
+        gr.Slider(label="Hair Color weight", value=2, minimum=1, maximum=4, step=1),
         gr.Radio(["short", "long", "mi-long"],label="Hair length", type="value"),
+        gr.Slider(label="Hair length weight", value=2, minimum=1, maximum=4, step=1),
         gr.Radio(["straight", "curvy", "wavy"],label="Hair texture", type="value"),
+        gr.Slider(label="Hair texture weight", value=2, minimum=1, maximum=4, step=1),
         gr.CheckboxGroup(choices=["((tattoos))", "((birthmark))", "freckles", "((scars))"],label="Skin details", type="value"),
-        gr.Radio(["brown", "hazel", "green", "blue", "gray", "amber", "black", "red", "violet"],label="Eye Color", type="value"),
+        gr.Radio(["brown", "hazel", "green", "blue", "gray", "amber", "black", "red", "violet"],label="Eyes Color", type="value"),
+        gr.Slider(label="Eyes color weight", value=2, minimum=1, maximum=4, step=1),
         gr.Checkbox(label="NSFW", info="👀👀👀")
     ],
     flagging_options=["blurry", "incorrect", "other"],
-    outputs=[gr.Textbox(label="Answer", lines=3), gr.Image(label="Generated Image", type="filepath")],
+    outputs=[gr.Image(label="Generated Image", type="filepath")],
 )
 
 demo.launch(share=True)