feat: Add more templates (#14)

- feat: Add more templates (82e52844617435e1ce20194ca796650f9b058b05)

.gitattributes

@@ -40,3 +40,6 @@ assets/template_1.png filter=lfs diff=lfs merge=lfs -text
 assets/template_2.png filter=lfs diff=lfs merge=lfs -text
 assets/template_b_avon.png filter=lfs diff=lfs merge=lfs -text
 assets/template_b_natura.png filter=lfs diff=lfs merge=lfs -text
+assets/template_c_natura.png filter=lfs diff=lfs merge=lfs -text
+assets/template_c_outros_sem_cupom.png filter=lfs diff=lfs merge=lfs -text
+assets/template_c_outros.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -133,8 +133,14 @@ with gr.Blocks() as demo:
                 image_final_price_input = gr.Number(label="Final Price", placeholder="Enter final price...")
                 image_coupon_code_input = gr.Textbox(label="Coupon Code", placeholder="Enter coupon code...")
                 template_selection = gr.Radio(
-                    choices=["lidi_promo", "natura", "avon"],
-                    value="lidi_promo",
+                    choices=[
+                            "outros", 
+                             "outros_sem_cupom", 
+                             "natura", 
+                             "avon",                              
+                             "lidi_promo",
+                             "natura_cliente"],
+                    value="outros",
                     label="Template Selection"
                 )
                 gen_image_btn = gr.Button("Generate Image")

template_system.py

@@ -219,7 +219,7 @@ class LidiPromoTemplate(Template):
         }
 
 
-class NaturaTemplate(Template):
+class NaturaClienteTemplate(Template):
     """
     Template implementation for Natura promotional images.
     Uses template_b_natura.png with different configuration.
@@ -353,7 +353,92 @@ class AvonTemplate(Template):
         }
 
 
+class TemplateC(Template):
+    """
+    Template implementation for Avon promotional images.
+    Uses template_b_avon.png with Avon-specific configuration.
+    """
+    
+    def __init__(self, template_path: str):
+        super().__init__(template_path)
+    
+    def get_box_config(self) -> Dict[str, Any]:
+        """Return box configuration for product image."""
+        return {
+            'size': (602, 434),
+            'position': (54, 304)  # (x, y) from top-left corner
+        }
+    
+    def get_text_config(self) -> Dict[str, Dict[str, Any]]:
+        """Return text configuration for all text elements."""
+        return {
+            'product_name': {
+                'position': (72, 777),
+                'color': '#000000',
+                'font': 'font_name',
+                'anchor': 'ls'
+            },
+            'original_price': {
+                'position': (72, 815),
+                'color': '#666666',
+                'font': 'font_price_from',
+                'anchor': 'ls'
+            },
+            'final_price': {
+                'position': (90, 887),
+                'color': '#FFFFFF',
+                'font': 'font_price',
+                'anchor': 'lm'
+            },
+            'coupon_code': {
+                'position': (471, 1020),
+                'color': '#FFFFFF',
+                'font': 'font_cupom',
+                'anchor': 'ms'
+            }
+        }
+    
+    def get_font_config(self) -> Dict[str, Dict[str, Any]]:
+        """Return font configuration for different text elements."""
+        return {
+            'font_name': {
+                'path': 'assets/Montserrat-Bold.ttf',
+                'size': 32
+            },
+            'font_price_from': {
+                'path': 'assets/Montserrat-Regular.ttf',
+                'size': 22
+            },
+            'font_price': {
+                'path': 'assets/Montserrat-Bold.ttf',
+                'size': 40
+            },
+            'font_cupom': {
+                'path': 'assets/Montserrat-Bold.ttf',
+                'size': 42
+            }
+        }
+        
+        
+class TemplateOutros(TemplateC):
+    
+    def __init__(self, template_path: str = None):
+        super().__init__("assets/template_c_outros.png")
+
+class TemplateOutrosSemCupom(TemplateC):
+    
+    def __init__(self, template_path: str = None):
+        super().__init__("assets/template_c_outros_sem_cupom.png")
+        
+class TemplateNatura(TemplateC):
+    
+    def __init__(self, template_path: str = None):
+        super().__init__("assets/template_c_natura.png")
+
 # Register additional templates
 TemplateRegistry.register('lidi_promo', LidiPromoTemplate)
-TemplateRegistry.register('natura', NaturaTemplate)
-TemplateRegistry.register('avon', AvonTemplate)
+TemplateRegistry.register('natura', TemplateNatura)
+TemplateRegistry.register('avon', AvonTemplate)
+TemplateRegistry.register('outros', TemplateOutros)
+TemplateRegistry.register('outros_sem_cupom', TemplateOutrosSemCupom)
+TemplateRegistry.register('natura_cliente', NaturaClienteTemplate)