Update app.py

app.py

@@ -7,6 +7,7 @@ import json
 import tempfile
 from typing import List, Dict, Tuple, Optional
 from dataclasses import dataclass
+import subprocess
 
 @dataclass
 class Question:
@@ -111,29 +112,115 @@ class QuizGenerator:
             all(isinstance(opt, str) for opt in question["options"])
         )
 
+class FontManager:
+    """Manages font installation and loading for the certificate generator"""
+    
+    @staticmethod
+    def install_fonts():
+        """Install required fonts if they're not already present"""
+        try:
+            # Install fonts package
+            subprocess.run([
+                "apt-get", "update", "-y"
+            ], check=True)
+            subprocess.run([
+                "apt-get", "install", "-y",
+                "fonts-liberation", # Provides Liberation fonts (Arial alternative)
+                "fontconfig"        # Font configuration
+            ], check=True)
+            
+            # Clear font cache
+            subprocess.run(["fc-cache", "-f"], check=True)
+            print("Fonts installed successfully")
+        except subprocess.CalledProcessError as e:
+            print(f"Warning: Could not install fonts: {e}")
+        except Exception as e:
+            print(f"Warning: Unexpected error installing fonts: {e}")
+    
+    @staticmethod
+    def get_font_paths() -> Dict[str, str]:
+        """Get the paths to the required fonts"""
+        # Liberation Sans is a metric-compatible replacement for Arial
+        font_paths = {
+            'regular': '/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf',
+            'bold': '/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf'
+        }
+        
+        # Fallback paths for different distributions
+        fallback_paths = {
+            'regular': [
+                '/usr/share/fonts/liberation-sans/LiberationSans-Regular.ttf',
+                '/usr/share/fonts/truetype/liberation/LiberationSans-Regular.ttf',
+                '/usr/share/fonts/TTF/LiberationSans-Regular.ttf'
+            ],
+            'bold': [
+                '/usr/share/fonts/liberation-sans/LiberationSans-Bold.ttf',
+                '/usr/share/fonts/truetype/liberation/LiberationSans-Bold.ttf',
+                '/usr/share/fonts/TTF/LiberationSans-Bold.ttf'
+            ]
+        }
+        
+        # Check fallback paths
+        for style in ['regular', 'bold']:
+            if not os.path.exists(font_paths[style]):
+                for path in fallback_paths[style]:
+                    if os.path.exists(path):
+                        font_paths[style] = path
+                        break
+        
+        return font_paths
+
 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:
+            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)
+        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,
-        name: str,
         score: float,
+        name: str,
         course_name: str,
         company_logo: Optional[str] = None,
         participant_photo: Optional[str] = None
     ) -> str:
-        certificate = self._create_base_certificate()
-        draw = ImageDraw.Draw(certificate)
-        
-        fonts = self._load_fonts()
-        self._add_borders(draw)
-        self._add_content(draw, fonts, name, score, course_name)
-        self._add_images(certificate, company_logo, participant_photo)
-        
-        return self._save_certificate(certificate)
+        """
+        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)
@@ -183,20 +270,24 @@ class CertificateGenerator:
         draw: ImageDraw.Draw,
         fonts: Dict[str, ImageFont.FreeTypeFont],
         name: str,
-        score: float,
-        course_name: str
+        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.strip() or "Participant", self.border_color),
+            (380, name, self.border_color),
             (460, "has successfully completed", 'black'),
-            (540, course_name.strip() or "Assessment", self.border_color),
-            (620, f"with a score of {score:.1f}%", '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')
         ]