Update handler.py

handler.py

@@ -3,7 +3,7 @@ from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDisc
 import base64
 from io import BytesIO
 from PIL import Image
-from scipy.spatial import KDTree
+from scipy.spatial import KDT
 from webcolors import CSS3_HEX_TO_NAMES, hex_to_rgb, hex_to_name
 import cv2
 
@@ -71,6 +71,7 @@ class EndpointHandler():
         torch.manual_seed(seed)
         images = self.pipe(result_prompt, negative_prompt = negative_prompt, image=raw_images, num_inference_steps=num_inference_steps, guidance_scale = guidance_scale, image_guidance_scale = image_guidance_scale).images
         img=images[0]
+        
         img.save("./1.png")
         logo.save("./logo.png")
         
@@ -133,7 +134,7 @@ class EndpointHandler():
 
         # Helpers.
         
-    def draw_text(img, text, font=cv2.FONT_HERSHEY_PLAIN, pos=(20, 45), font_scale=1, font_thickness=1, text_color=(0, 0, 255)):
+    def draw_text(self, img, text, font=cv2.FONT_HERSHEY_PLAIN, pos=(20, 45), font_scale=1, font_thickness=1, text_color=(0, 0, 255)):
         x, y = pos
         text_size, _ = cv2.getTextSize(text, font, font_scale, font_thickness)
         text_w, text_h = text_size
@@ -141,7 +142,7 @@ class EndpointHandler():
         return text_size
     
     
-    def smooth_corners(image, radius, alpha=255):
+    def smooth_corners(self, image, radius, alpha=255):
         factor = 5  # Factor to increase the image size for anti-aliasing
         radius = radius * factor
         size = (image.width, image.height)
@@ -175,7 +176,7 @@ class EndpointHandler():
     
         return result_image
 
-    def add_logo_to_image(base_image_path, logo_path, corner_radius=30):
+    def add_logo_to_image(self, base_image_path, logo_path, corner_radius=30):
 
         base_image = Image.open(base_image_path)
         smoothed_image = smooth_corners(base_image, radius=30)
@@ -230,7 +231,7 @@ class EndpointHandler():
         return result_logo_image
 
 
-    def create_template(logo_and_image, punchline_text, punchline_text_color, punchline_text_max_width, spacing_image_text):
+    def create_template(self, logo_and_image, punchline_text, punchline_text_color, punchline_text_max_width, spacing_image_text):
     
         result_logo_image_resized = logo_and_image.resize((325, 425))
         # Create a white background with a size of 645x645
@@ -292,7 +293,7 @@ class EndpointHandler():
         
         return template, punchline_text_height, len(lines)
 
-    def draw_button_with_text(button_color, button_text, button_font=cv2.FONT_HERSHEY_TRIPLEX, button_font_scale = 0.7, button_font_thickness=1, button_text_color=(0, 0, 255)):
+    def draw_button_with_text(self, button_color, button_text, button_font=cv2.FONT_HERSHEY_TRIPLEX, button_font_scale = 0.7, button_font_thickness=1, button_text_color=(0, 0, 255)):
         # Create a blank 300x80 white image (background)
         background_image = Image.new("RGB", (300, 80), "white")
         button_image_pil = smooth_corners(background_image, 20, alpha=255)
@@ -313,7 +314,7 @@ class EndpointHandler():
         
         return button_image_pil
 
-    def concat_template_with_button(template, button, line_count, spacing_between_punchline_and_button, punchline_text_height):
+    def concat_template_with_button(self, template, button, line_count, spacing_between_punchline_and_button, punchline_text_height):
     
         result_template = template.copy()
         
@@ -329,7 +330,7 @@ class EndpointHandler():
         
         return result_template
 
-    def create_image_template(
+    def create_image_template(self, 
         base_image_path,
         logo_path,
         punchline_text,