Update app.py

app.py

@@ -204,107 +204,11 @@ class QuizGenerationError(Exception):
     """Exception raised for errors in quiz generation"""
     pass
 
-
-class CertificateGenerator:
-    def __init__(self):
-        self.certificate_size = (1200, 800)
-        self.border_color = '#4682B4'
-        self.background_color = '#F0F8FF'
-        
-        # 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'], 60)
-                fonts['text'] = ImageFont.truetype(self.font_paths['regular'], 40)
-                fonts['subtitle'] = ImageFont.truetype(self.font_paths['regular'], 30)
-            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,
-                'text': default,
-                'subtitle': default
-            }
-        return fonts
-        
-    def generate(
-        self,
-        score: float,
-        name: str,
-        course_name: str,
-        company_logo: Optional[str] = None,
-        participant_photo: Optional[str] = None
-    ) -> str:
-        """
-        Generate a certificate with custom styling and optional logo/photo
-        """
-        try:
-            certificate = self._create_base_certificate()
-            draw = ImageDraw.Draw(certificate)
-            
-            fonts = self._load_fonts()
-            self._add_borders(draw)
-            self._add_content(draw, fonts, str(name), str(course_name), float(score))
-            self._add_images(certificate, company_logo, participant_photo)
-            
-            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 _load_fonts(self) -> Dict[str, ImageFont.FreeTypeFont]:
-        try:
-            return {
-                'title': ImageFont.truetype("arial.ttf", 60),
-                'text': ImageFont.truetype("arial.ttf", 40),
-                'subtitle': ImageFont.truetype("arial.ttf", 30)
-            }
-        except Exception as e:
-            print(f"Font loading error: {e}. Using default font.")
-            default = ImageFont.load_default()
-            return {'title': default, 'text': default, 'subtitle': default}
-    
-    def _add_borders(self, draw: ImageDraw.Draw):
-        # Main borders
-        draw.rectangle([20, 20, 1180, 780], outline=self.border_color, width=3)
-        draw.rectangle([40, 40, 1160, 760], outline=self.border_color, width=1)
-        
-        # Decorative corners
-        self._add_decorative_corners(draw)
-    
-    def _add_decorative_corners(self, draw: ImageDraw.Draw):
-        corner_size = 20
-        corners = [
-            # Top-left
-            [(20, 40), (20 + corner_size, 40)],
-            [(40, 20), (40, 20 + corner_size)],
-            # Top-right
-            [(1180 - corner_size, 40), (1180, 40)],
-            [(1160, 20), (1160, 20 + corner_size)],
-            # Bottom-left
-            [(20, 760), (20 + corner_size, 760)],
-            [(40, 780 - corner_size), (40, 780)],
-            # Bottom-right
-            [(1180 - corner_size, 760), (1180, 760)],
-            [(1160, 780 - corner_size), (1160, 780)]
-        ]
-        
 class CertificateGenerator:
     def __init__(self):
         self.certificate_size = (1200, 800)
-        self.border_color = '#FFFFFF'  # White background
         self.background_color = '#FFFFFF'
+        self.border_color = '#1C1D1F'
         
         # Install fonts if needed
         FontManager.install_fonts()
@@ -315,10 +219,10 @@ class CertificateGenerator:
         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
+                fonts['title'] = ImageFont.truetype(self.font_paths['bold'], 36)
+                fonts['subtitle'] = ImageFont.truetype(self.font_paths['regular'], 14)
+                fonts['text'] = ImageFont.truetype(self.font_paths['regular'], 20)
+                fonts['name'] = ImageFont.truetype(self.font_paths['bold'], 32)
             else:
                 raise ValueError("No suitable fonts found")
         except Exception as e:
@@ -344,12 +248,18 @@ class CertificateGenerator:
             certificate = self._create_base_certificate()
             draw = ImageDraw.Draw(certificate)
             
+            # Add professional border
+            self._add_professional_border(draw)
+            
             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)
             
+            if participant_photo:
+                self._add_photo(certificate, participant_photo)
+            
             return self._save_certificate(certificate)
         except Exception as e:
             print(f"Error generating certificate: {e}")
@@ -358,6 +268,33 @@ class CertificateGenerator:
     def _create_base_certificate(self) -> Image.Image:
         return Image.new('RGB', self.certificate_size, self.background_color)
     
+    def _add_professional_border(self, draw: ImageDraw.Draw):
+        # Outer border
+        draw.rectangle(
+            [(20, 20), (self.certificate_size[0] - 20, self.certificate_size[1] - 20)],
+            outline='#1C1D1F',
+            width=2
+        )
+        
+        # Inner border
+        draw.rectangle(
+            [(40, 40), (self.certificate_size[0] - 40, self.certificate_size[1] - 40)],
+            outline='#1C1D1F',
+            width=1
+        )
+        
+        # Corner decorations
+        corner_size = 20
+        for x, y in [(20, 20), (self.certificate_size[0] - 20, 20),
+                     (20, self.certificate_size[1] - 20),
+                     (self.certificate_size[0] - 20, self.certificate_size[1] - 20)]:
+            draw.rectangle(
+                [(x - corner_size, y - corner_size),
+                 (x + corner_size, y + corner_size)],
+                outline='#1C1D1F',
+                width=1
+            )
+    
     def _add_content(
         self,
         draw: ImageDraw.Draw,
@@ -368,53 +305,95 @@ class CertificateGenerator:
     ):
         # 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")
+        draw.text((1100, 60), certificate_id, font=fonts['subtitle'], fill='#666666', anchor="ra")
         
         # Add "CERTIFICATE OF COMPLETION" text
-        draw.text((60, 120), "CERTIFICATE OF COMPLETION", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, 140), "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)
+        text_width = draw.textlength(course_name, fonts['title'])
+        max_width = self.certificate_size[0] - 120  # Leave margins
+        if text_width > max_width:
+            # Split into multiple lines if too long
+            words = course_name.split()
+            lines = []
+            current_line = []
+            current_width = 0
+            for word in words:
+                word_width = draw.textlength(word + " ", fonts['title'])
+                if current_width + word_width <= max_width:
+                    current_line.append(word)
+                    current_width += word_width
+                else:
+                    lines.append(" ".join(current_line))
+                    current_line = [word]
+                    current_width = word_width
+            if current_line:
+                lines.append(" ".join(current_line))
+            course_name = "\n".join(lines)
+        
+        draw.multiline_text((60, 200), 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')
+        draw.text((60, 300), "Instructor", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, 330), "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')
+        draw.text((60, 420), 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')
+        draw.text((60, 500), "Date", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, 530), 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')
+        draw.text((300, 500), "Score", font=fonts['subtitle'], fill='#666666')
+        draw.text((300, 530), 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')
+        draw.text((60, 720), ref_number, font=fonts['subtitle'], fill='#666666')
     
     def _add_logo(self, certificate: Image.Image, logo_path: str):
         try:
             logo = Image.open(logo_path)
             # Resize logo to appropriate size
             logo.thumbnail((150, 80))
-            # Position in top-left corner
-            certificate.paste(logo, (50, 30))
+            # Position in top-left corner with padding
+            certificate.paste(logo, (60, 50), mask=logo if 'A' in logo.getbands() else None)
         except Exception as e:
             print(f"Error adding logo: {e}")
     
+    def _add_photo(self, certificate: Image.Image, photo_path: str):
+        try:
+            photo = Image.open(photo_path)
+            # Create circular mask
+            size = (100, 100)
+            mask = Image.new('L', size, 0)
+            draw = ImageDraw.Draw(mask)
+            draw.ellipse((0, 0, size[0], size[1]), fill=255)
+            
+            # Resize photo maintaining aspect ratio
+            photo.thumbnail(size)
+            
+            # Create a circular photo
+            output = Image.new('RGBA', size, (0, 0, 0, 0))
+            output.paste(photo, (0, 0))
+            output.putalpha(mask)
+            
+            # Position in top-right corner with padding
+            certificate.paste(output, (1000, 50), mask=output)
+        except Exception as e:
+            print(f"Error adding photo: {e}")
+    
     def _save_certificate(self, certificate: Image.Image) -> str:
         temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
         certificate.save(temp_file.name, 'PNG', quality=95)
         return temp_file.name
-
     
 class QuizApp:
     def __init__(self, api_key: str):