Update app.py

app.py

@@ -300,8 +300,63 @@ class CertificateGenerator:
             [(1160, 780 - corner_size), (1160, 780)]
         ]
         
-        for corner in corners:
-            draw.line(corner, fill=self.border_color, width=2)
+class CertificateGenerator:
+    def __init__(self):
+        self.certificate_size = (1200, 800)
+        self.border_color = '#FFFFFF'  # White background
+        self.background_color = '#FFFFFF'
+        
+        # Install fonts if needed
+        FontManager.install_fonts()
+        self.font_paths = FontManager.get_font_paths()
+    
+    def _load_fonts(self) -> Dict[str, ImageFont.FreeTypeFont]:
+        """Load fonts with fallbacks"""
+        fonts = {}
+        try:
+            if self.font_paths['regular'] and self.font_paths['bold']:
+                fonts['title'] = ImageFont.truetype(self.font_paths['bold'], 36)  # Smaller for course name
+                fonts['subtitle'] = ImageFont.truetype(self.font_paths['regular'], 14)  # For labels
+                fonts['text'] = ImageFont.truetype(self.font_paths['regular'], 20)  # For regular text
+                fonts['name'] = ImageFont.truetype(self.font_paths['bold'], 32)  # For participant name
+            else:
+                raise ValueError("No suitable fonts found")
+        except Exception as e:
+            print(f"Font loading error: {e}. Using default font.")
+            default = ImageFont.load_default()
+            fonts = {
+                'title': default,
+                'subtitle': default,
+                'text': default,
+                'name': default
+            }
+        return fonts
+        
+    def generate(
+        self,
+        score: float,
+        name: str,
+        course_name: str,
+        company_logo: Optional[str] = None,
+        participant_photo: Optional[str] = None
+    ) -> str:
+        try:
+            certificate = self._create_base_certificate()
+            draw = ImageDraw.Draw(certificate)
+            
+            fonts = self._load_fonts()
+            self._add_content(draw, fonts, str(name), str(course_name), float(score))
+            
+            if company_logo:
+                self._add_logo(certificate, company_logo)
+            
+            return self._save_certificate(certificate)
+        except Exception as e:
+            print(f"Error generating certificate: {e}")
+            return None
+    
+    def _create_base_certificate(self) -> Image.Image:
+        return Image.new('RGB', self.certificate_size, self.background_color)
     
     def _add_content(
         self,
@@ -311,45 +366,49 @@ class CertificateGenerator:
         course_name: str,
         score: float
     ):
-        # Title and headers
-        draw.text((600, 100), "CertifyMe AI", font=fonts['title'], fill=self.border_color, anchor="mm")
-        draw.text((600, 160), "Certificate of Achievement", font=fonts['subtitle'], fill=self.border_color, anchor="mm")
-        
-        # Clean inputs
-        name = str(name).strip() or "Participant"
-        course_name = str(course_name).strip() or "Assessment"
-        
-        # Main content
-        content = [
-            (300, "This is to certify that", 'black'),
-            (380, name, self.border_color),
-            (460, "has successfully completed", 'black'),
-            (540, course_name, self.border_color),
-            (620, f"with a score of {float(score):.1f}%", 'black'),
-            (700, datetime.now().strftime("%B %d, %Y"), 'black')
-        ]
+        # Add certificate ID at top right
+        certificate_id = f"Certificate no: {datetime.now().strftime('%Y%m%d')}-{abs(hash(name)) % 10000:04d}"
+        draw.text((1100, 40), certificate_id, font=fonts['subtitle'], fill='#666666', anchor="ra")
         
-        for y, text, color in content:
-            draw.text((600, y), text, font=fonts['text'], fill=color, anchor="mm")
-    
-    def _add_images(
-        self,
-        certificate: Image.Image,
-        company_logo: Optional[str],
-        participant_photo: Optional[str]
-    ):
-        if company_logo:
-            self._add_image(certificate, company_logo, (50, 50))
-        if participant_photo:
-            self._add_image(certificate, participant_photo, (1000, 50))
+        # Add "CERTIFICATE OF COMPLETION" text
+        draw.text((60, 120), "CERTIFICATE OF COMPLETION", font=fonts['subtitle'], fill='#666666')
+        
+        # Add course name (large and bold)
+        course_name = course_name.strip() or "Assessment"
+        draw.multiline_text((60, 180), course_name, font=fonts['title'], fill='#1C1D1F', spacing=10)
+        
+        # Add instructor info
+        draw.text((60, 280), "Instructor", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, 310), "CertifyMe AI", font=fonts['text'], fill='#1C1D1F')
+        
+        # Add participant name (large)
+        name = name.strip() or "Participant"
+        draw.text((60, 400), name, font=fonts['name'], fill='#1C1D1F')
+        
+        # Add date and score info with spacing
+        date_str = datetime.now().strftime("%b. %d, %Y")
+        
+        # Date section
+        draw.text((60, 480), "Date", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, 510), date_str, font=fonts['text'], fill='#1C1D1F')
+        
+        # Score section
+        draw.text((300, 480), "Score", font=fonts['subtitle'], fill='#666666')
+        draw.text((300, 510), f"{float(score):.1f}%", font=fonts['text'], fill='#1C1D1F')
+
+        # Add reference number at bottom
+        ref_number = f"Reference Number: {abs(hash(name + date_str)) % 10000:04d}"
+        draw.text((60, 700), ref_number, font=fonts['subtitle'], fill='#666666')
     
-    def _add_image(self, certificate: Image.Image, image_path: str, position: Tuple[int, int]):
+    def _add_logo(self, certificate: Image.Image, logo_path: str):
         try:
-            img = Image.open(image_path)
-            img.thumbnail((150, 150))
-            certificate.paste(img, position)
+            logo = Image.open(logo_path)
+            # Resize logo to appropriate size
+            logo.thumbnail((150, 80))
+            # Position in top-left corner
+            certificate.paste(logo, (50, 30))
         except Exception as e:
-            print(f"Error adding image: {e}")
+            print(f"Error adding logo: {e}")
     
     def _save_certificate(self, certificate: Image.Image) -> str:
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')