Update image watermark

1 files changed, 5 insertions, 12 deletions

diff --git a/indoteknik_api/controllers/controller.py b/indoteknik_api/controllers/controller.py
index a362e6a6..9ee56062 100644
--- a/indoteknik_api/controllers/controller.py
+++ b/indoteknik_api/controllers/controller.py
@@ -210,12 +210,7 @@ class Controller(http.Controller):
     def add_watermark_to_image(self, image, ratio, version = '1'):
         if not image: return ''
         
-        LOGO_FILENAME = {
-            '1': 'logo-indoteknik-gray.png',
-            '2': 'logo-indoteknik.png'
-        }
-        
-        logo_path = get_module_resource('indoteknik_api', 'static', 'src', 'images', LOGO_FILENAME.get(version))
+        logo_path = get_module_resource('indoteknik_api', 'static', 'src', 'images', 'logo-indoteknik-gray.png')
         logo_img = Image.open(logo_path).convert('RGBA')
         
         img_data = io.BytesIO(base64.b64decode(image))
@@ -243,15 +238,13 @@ class Controller(http.Controller):
             logo_footer_img = Image.open(logo__footer_path).convert('RGBA')
             logo_footer_img.thumbnail((img_width, img_height // 1))
             logo_footer_w, logo_footer_h = logo_footer_img.size
-            new_img.paste(logo_footer_img, (0, img_height - logo_footer_h), logo_footer_img)
-            
-            logo_img_w = img_width // 1.8
-            logo_img_h = img_height // 1.8
+            new_img.paste(logo_footer_img, (0, img_height - logo_footer_h - 20), logo_footer_img)
             
         logo_img.thumbnail((logo_img_w, logo_img_h))
         
-        # Add watermark
-        new_img.paste(logo_img, (12, 10), logo_img)
+        if version == '1':
+            # Add watermark
+            new_img.paste(logo_img, (12, 10), logo_img)
         
         buffered = io.BytesIO()
         new_img.save(buffered, format="PNG")