Revert image api style

1 files changed, 12 insertions, 5 deletions

diff --git a/indoteknik_api/controllers/controller.py b/indoteknik_api/controllers/controller.py
index 50e86b68..fe0f1203 100644
--- a/indoteknik_api/controllers/controller.py
+++ b/indoteknik_api/controllers/controller.py
@@ -210,7 +210,12 @@ class Controller(http.Controller):
     def add_watermark_to_image(self, image, ratio, version = '1'):
         if not image: return ''
         
-        logo_path = get_module_resource('indoteknik_api', 'static', 'src', 'images', 'logo-indoteknik-gray.png')
+        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_img = Image.open(logo_path).convert('RGBA')
         
         img_data = io.BytesIO(base64.b64decode(image))
@@ -238,13 +243,15 @@ 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 - 20), logo_footer_img)
+            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
             
         logo_img.thumbnail((logo_img_w, logo_img_h))
         
-        if version == '1':
-            # Add watermark
-            new_img.paste(logo_img, (12, 10), logo_img)
+        # Add watermark
+        new_img.paste(logo_img, (12, 10), logo_img)
         
         buffered = io.BytesIO()
         new_img.save(buffered, format="PNG")