feat: enhance QR code generation with TECH_FLORIPA logo integration and update certificate generation process

Author: maxsonferovante

certified_builder/certified_builder.py

@@ -1,10 +1,9 @@
 import logging
-import tempfile
 import os
 from typing import List
 from PIL import Image, ImageDraw, ImageFont
-from io import BytesIO
 from models.participant import Participant
+from config import config
 from certified_builder.utils.fetch_file_certificate import fetch_file_certificate
 from certified_builder.certificates_on_solana import CertificatesOnSolana
 from certified_builder.make_qrcode import MakeQRCode
@@ -33,6 +32,7 @@ def build_certificates(self, participants: List[Participant]):
             # Cache for background and logo if they are the same for all participants
             certificate_template = None
             logo = None
+            logo_tech_floripa = None
 
             # Check if all participants share the same background and logo
             if participants:
@@ -46,6 +46,9 @@ def build_certificates(self, participants: List[Participant]):
                 if all_same_logo:
                     logo = self._download_image(first_participant.certificate.logo)
 
+                if not logo_tech_floripa:
+                    logo_tech_floripa = self._download_image(config.TECH_FLORIPA_LOGO_URL)
+
             for participant in participants:
                 try:
                     # Register certificate on Solana, with returned data extract url for verification
@@ -57,7 +60,7 @@ def build_certificates(self, participants: List[Participant]):
                             "certificate_code": participant.formated_validation_code()
                         }
                     )
-                                                           
+                
                     # alteração: agora usamos a função renomeada que apenas extrai o explorer_url
                     participant.authenticity_verification_url = extract_solana_explorer_url(solana_response=solana_response)
 
@@ -71,7 +74,7 @@ def build_certificates(self, participants: List[Participant]):
                         logo = self._download_image(participant.certificate.logo)
 
                     # Generate and save certificate
-                    certificate_generated = self.generate_certificate(participant, certificate_template, logo)
+                    certificate_generated = self.generate_certificate(participant, certificate_template, logo, logo_tech_floripa)
                     certificate_path = self.save_certificate(certificate_generated, participant)
 
                     results.append({
@@ -121,7 +124,7 @@ def _ensure_valid_rgba(self, img: Image) -> Image:
             new_img.paste(img.convert('RGB'), (0, 0))
             return new_img
 
-    def generate_certificate(self, participant: Participant, certificate_template: Image, logo: Image):
+    def generate_certificate(self, participant: Participant, certificate_template: Image, logo: Image, logo_tech_floripa: Image):
         """Generate a certificate for a participant."""
         try:
             # Ensure images have valid transparency channels
@@ -145,9 +148,10 @@ def generate_certificate(self, participant: Participant, certificate_template: I
                 # Fallback without using the logo as its own mask
                 overlay.paste(logo, (50, 50))
 
-           
+            url_qr_code = f"{config.TECH_FLORIPA_CERTIFICATE_VALIDATE_URL}?validate_code={participant.formated_validation_code()}"
             qrcode_size = (150, 150)                       
-            qr_code_image_io = MakeQRCode.generate_qr_code(participant.authenticity_verification_url)            
+            logger.info(f"URL do QR code: {url_qr_code} para o certificado de {participant.name_completed()}")
+            qr_code_image_io = MakeQRCode.generate_qr_code(url_qr_code, logo_tech_floripa=logo_tech_floripa)                            
             qr_code_image = Image.open(qr_code_image_io).convert("RGBA")            
             # comentário: para manter o QR nítido, usamos NEAREST ao redimensionar
             if qr_code_image.size != qrcode_size:

certified_builder/make_qrcode.py

@@ -2,18 +2,19 @@
 import qrcode
 import logging
 from io import BytesIO
+from PIL import Image
 from qrcode.image.pil import PilImage
 
 logger = logging.getLogger(__name__)
 
 class MakeQRCode:
     @staticmethod
-    def generate_qr_code(data: str) -> BytesIO:
+    def generate_qr_code(data: str, logo_tech_floripa: Image) -> BytesIO:        
         try:
-            logger.info("Generating QR code ")
+            logger.info(f"Generating QR code for {data}")
             qr = qrcode.QRCode(
                 version=1,
-                error_correction=qrcode.constants.ERROR_CORRECT_L,
+                error_correction=qrcode.constants.ERROR_CORRECT_H,
                 box_size=10,
                 border=4,
             )
@@ -22,10 +23,27 @@ def generate_qr_code(data: str) -> BytesIO:
             qr.make(fit=True)
             img = qr.make_image(fill_color="black", back_color="transparent", image_factory=PilImage)
             img = img.convert("RGBA")
+            
+            # Redimensiona o logo para aproximadamente 30% do tamanho do QR code
+            # Tamanho limitado para não interferir nos padrões de detecção nos cantos
+            qr_width, qr_height = img.size
+            logo_size = int(min(qr_width, qr_height) * 0.3)
+            logo_resized = logo_tech_floripa.copy()
+            logo_resized.thumbnail((logo_size, logo_size), Image.Resampling.LANCZOS)
+            logo_resized = logo_resized.convert("RGBA")
+            
+            # Calcula a posição central para colar o logo
+            logo_width, logo_height = logo_resized.size
+            position = ((qr_width - logo_width) // 2, (qr_height - logo_height) // 2)
+            
+            # Cola o logo no centro do QR code mantendo transparência
+            # Com ERROR_CORRECT_H (30% redundância), o QR code permanece legível mesmo com o logo
+            img.paste(logo_resized, position, logo_resized)
+            
             byte_io = BytesIO()
             img.save(byte_io, format='PNG')
             byte_io.seek(0)
-            logger.info("QR code generated successfully")
+            logger.info(f"QR code generated successfully for {data}")
             return byte_io
         except Exception as e:
             logging.error(f"Failed to generate QR code: {e}")