Update app3.py

app3.py

@@ -45,35 +45,48 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
     
     # Define your data (this could also be a JSON payload)
     print("SD_processing")
+    # data = {
+    #     "input": {
+    #         "api": {
+    #             "method": "POST",
+    #             "endpoint": "/sdapi/v1/txt2img"
+    #         },
+    #         "payload": {
+    #             "override_settings": {
+    #                 "sd_model_checkpoint": "CyberRealistic",
+    #                 "sd_vae": ""
+    #             },
+    #             "override_settings_restore_afterwards": True,
+    #             "refiner_checkpoint": "",
+    #             "refiner_switch_at": 0.8,
+    #             "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}",
+    #             "seed": -1,
+    #             "batch_size": 1,
+    #             "steps": 30,
+    #             "cfg_scale": 7,
+    #             "width": 520,
+    #             "height": 520,
+    #             "sampler_name": "DPM++ SDE Karras",
+    #             "sampler_index": "DPM++ SDE Karras",
+    #             "restore_faces": False
+    #         }
+    #     }
+    # }
+
     data = {
-        "input": {
-            "api": {
-                "method": "POST",
-                "endpoint": "/sdapi/v1/txt2img"
-            },
-            "payload": {
-                "override_settings": {
-                    "sd_model_checkpoint": "CyberRealistic",
-                    "sd_vae": ""
-                },
-                "override_settings_restore_afterwards": True,
-                "refiner_checkpoint": "",
-                "refiner_switch_at": 0.8,
-                "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}",
-                "seed": -1,
-                "batch_size": 1,
-                "steps": 30,
-                "cfg_scale": 7,
-                "width": 520,
-                "height": 520,
-                "sampler_name": "DPM++ SDE Karras",
-                "sampler_index": "DPM++ SDE Karras",
-                "restore_faces": False
-            }
-        }
+    "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}",
+        "width": 512,
+        "height": 720,
+        "guidance_scale": 7.5,
+        "num_inference_steps": 50,
+        "num_outputs": 1,
+        "prompt_strength": 0.8,
+        "scheduler": "K-LMS"
+    }
     }
-
 
 
 
@@ -89,6 +102,7 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
         # Poll the status until it's not 'IN_QUEUE'
         while response_data['status'] == 'IN_QUEUE':
             time.sleep(5)  # Wait for 5 seconds before checking again
+            print("1")
             response = requests.get(f"{url}/{msg_id}", headers=headers)
             
             try:
@@ -100,6 +114,7 @@ def SD_call(prompt, image_prompt, age, weight, color, hair_color,hair_length,hai
 
         # Check if the response contains images
         if 'images' in response_data.get('output', {}):
+            print("image")
             base64_image = response_data['output']['images'][0]
             image_bytes = base64.b64decode(base64_image)
             
@@ -125,7 +140,11 @@ 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
+    #return "Image generated successfully", image_path
+    if image_path is not None:
+        return "Image generated successfully", image_path
+    else:
+        return "No images found in the response.", None
 
 
 demo = gr.Interface(