Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from groq import Groq
 import os
-from PIL import Image, ImageDraw, ImageFont
+from PIL import Image, ImageDraw, ImageFont, ImageFilter
 from datetime import datetime
 import json
 import tempfile
@@ -9,6 +9,7 @@ from typing import List, Dict, Tuple, Optional
 from dataclasses import dataclass
 import subprocess
 import re
+import random
 
 @dataclass
 class Question:
@@ -44,7 +45,7 @@ class QuizGenerator:
                     }
                 ],
                 model="llama-3.2-3b-preview",
-                temperature=0,
+                temperature=0.1,
                 max_tokens=1024
             )
             
@@ -313,14 +314,26 @@ class CertificateGenerator:
         return fonts
     
     def _add_professional_border(self, draw: ImageDraw.Draw):
-        # Single elegant inner border with padding
+        """Add professional border with improved corners"""
         padding = 40
-        draw.rectangle(
-            [(padding, padding), 
-             (self.certificate_size[0] - padding, self.certificate_size[1] - padding)],
-            outline='#1C1D1F',
-            width=2
-        )
+        border_width = 2
+        corner_radius = 10
+        
+        # Draw rounded rectangle border
+        x0, y0 = padding, padding
+        x1, y1 = self.certificate_size[0] - padding, self.certificate_size[1] - padding
+        
+        # Draw corners
+        draw.arc((x0, y0, x0 + corner_radius * 2, y0 + corner_radius * 2), 180, 270, '#1C1D1F', border_width)
+        draw.arc((x1 - corner_radius * 2, y0, x1, y0 + corner_radius * 2), 270, 0, '#1C1D1F', border_width)
+        draw.arc((x0, y1 - corner_radius * 2, x0 + corner_radius * 2, y1), 90, 180, '#1C1D1F', border_width)
+        draw.arc((x1 - corner_radius * 2, y1 - corner_radius * 2, x1, y1), 0, 90, '#1C1D1F', border_width)
+        
+        # Draw lines
+        draw.line((x0 + corner_radius, y0, x1 - corner_radius, y0), '#1C1D1F', border_width)  # Top
+        draw.line((x0 + corner_radius, y1, x1 - corner_radius, y1), '#1C1D1F', border_width)  # Bottom
+        draw.line((x0, y0 + corner_radius, x0, y1 - corner_radius), '#1C1D1F', border_width)  # Left
+        draw.line((x1, y0 + corner_radius, x1, y1 - corner_radius), '#1C1D1F', border_width)  # Right
     
     def _add_content(
         self,
@@ -328,59 +341,40 @@ class CertificateGenerator:
         fonts: Dict[str, ImageFont.FreeTypeFont],
         name: str,
         course_name: str,
-        score: float
+        score: float,
+        y_offset: int = 140
     ):
+        """Add content with adjusted vertical positioning"""
         # Add "CERTIFICATE OF COMPLETION" text
-        draw.text((60, 140), "CERTIFICATE OF COMPLETION", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, y_offset), "CERTIFICATE OF COMPLETION", font=fonts['subtitle'], fill='#666666')
         
         # Add course name (large and bold)
         course_name = course_name.strip() or "Assessment"
-        text_width = draw.textlength(course_name, fonts['title'])
-        max_width = self.certificate_size[0] - 120  # Leave margins
-        if text_width > max_width:
-            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)
+        draw.text((60, y_offset + 60), course_name, font=fonts['title'], fill='#1C1D1F')
         
         # Add instructor info
-        draw.text((60, 300), "Instructor", font=fonts['subtitle'], fill='#666666')
-        draw.text((60, 330), "CertifyMe AI", font=fonts['text'], fill='#1C1D1F')
+        draw.text((60, y_offset + 160), "Instructor", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, y_offset + 190), "CertifyMe AI", font=fonts['text'], fill='#1C1D1F')
         
         # Add participant name (large)
         name = name.strip() or "Participant"
-        draw.text((60, 420), name, font=fonts['name'], fill='#1C1D1F')
+        draw.text((60, y_offset + 280), name, font=fonts['name'], fill='#1C1D1F')
         
-        # Add date and score info with spacing
+        # Add date and score info
         date_str = datetime.now().strftime("%b. %d, %Y")
         
         # Date section
-        draw.text((60, 500), "Date", font=fonts['subtitle'], fill='#666666')
-        draw.text((60, 530), date_str, font=fonts['text'], fill='#1C1D1F')
+        draw.text((60, y_offset + 360), "Date", font=fonts['subtitle'], fill='#666666')
+        draw.text((60, y_offset + 390), date_str, font=fonts['text'], fill='#1C1D1F')
         
         # Score section
-        draw.text((300, 500), "Score", font=fonts['subtitle'], fill='#666666')
-        draw.text((300, 530), f"{float(score):.1f}%", font=fonts['text'], fill='#1C1D1F')
-
-        # Footer section with certificate number and reference
+        draw.text((300, y_offset + 360), "Score", font=fonts['subtitle'], fill='#666666')
+        draw.text((300, y_offset + 390), f"{float(score):.1f}%", font=fonts['text'], fill='#1C1D1F')
+        
+        # Footer section
         certificate_id = f"Certificate no: {datetime.now().strftime('%Y%m%d')}-{abs(hash(name)) % 10000:04d}"
         ref_number = f"Reference Number: {abs(hash(name + date_str)) % 10000:04d}"
         
-        # Draw footer text aligned to left and right
         draw.text((60, 720), certificate_id, font=fonts['subtitle'], fill='#666666')
         draw.text((1140, 720), ref_number, font=fonts['subtitle'], fill='#666666', anchor="ra")
     
@@ -395,26 +389,67 @@ class CertificateGenerator:
             print(f"Error adding logo: {e}")
     
     def _add_photo(self, certificate: Image.Image, photo_path: str):
+        """Add a clear circular profile photo in the top-right corner with adjusted position"""
         try:
+            if not photo_path or not os.path.exists(photo_path):
+                print(f"Photo path does not exist: {photo_path}")
+                return
+            
+            # Open and process photo
             photo = Image.open(photo_path)
-            # Create circular mask
-            size = (100, 100)
+            
+            # Define size for circular photo
+            size = (120, 120)
+            
+            # Convert to RGB if not already
+            if photo.mode not in ('RGB', 'RGBA'):
+                photo = photo.convert('RGB')
+            
+            # Create high-quality circular mask
             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)
+            aspect = photo.width / photo.height
+            if aspect > 1:
+                new_height = size[1]
+                new_width = int(new_height * aspect)
+            else:
+                new_width = size[0]
+                new_height = int(new_width / aspect)
             
-            # Create a circular photo
+            photo = photo.resize((new_width, max(new_height, 1)), Image.Resampling.LANCZOS)
+            
+            # Center crop
+            if aspect > 1:
+                left = (new_width - size[0]) // 2
+                photo = photo.crop((left, 0, left + size[0], size[1]))
+            else:
+                top = (new_height - size[1]) // 2
+                photo = photo.crop((0, top, size[0], top + size[1]))
+            
+            # Create 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)
+            # Adjusted position - moved down from top
+            photo_x = certificate.width - size[0] - 60  # 60px from right
+            photo_y = 50  # Increased from 40 to 50px from top
+                
+            # Add white background circle
+            bg = Image.new('RGBA', size, (255, 255, 255, 255))
+            certificate.paste(bg, (photo_x, photo_y), mask=mask)
+            
+            # Paste the photo
+            certificate.paste(output, (photo_x, photo_y), mask=output)
+            print(f"Successfully added photo at position ({photo_x}, {photo_y})")
+            
         except Exception as e:
-            print(f"Error adding photo: {e}")
+            print(f"Error adding photo: {str(e)}")
+            import traceback
+            traceback.print_exc()
     
     def generate(
         self,
@@ -424,35 +459,66 @@ class CertificateGenerator:
         company_logo: Optional[str] = None,
         participant_photo: Optional[str] = None
     ) -> str:
+        """Generate certificate with improved photo handling"""
         try:
-            certificate = self._create_base_certificate()
+            # Create base certificate
+            certificate = Image.new('RGB', self.certificate_size, self.background_color)
             draw = ImageDraw.Draw(certificate)
             
-            # Add professional border
+            # Add border
             self._add_professional_border(draw)
             
+            # Load fonts
             fonts = self._load_fonts()
-            self._add_content(draw, fonts, str(name), str(course_name), float(score))
             
-            if company_logo:
+            # Add company logo if provided
+            if company_logo and os.path.exists(company_logo):
                 self._add_logo(certificate, company_logo)
             
+            # Add participant photo if provided
             if participant_photo:
+                print(f"Processing photo: {participant_photo}")  # Debug info
                 self._add_photo(certificate, participant_photo)
             
+            # Add content
+            self._add_content(draw, fonts, str(name), str(course_name), float(score))
+            
+            # Save certificate
             return self._save_certificate(certificate)
+            
         except Exception as e:
-            print(f"Error generating certificate: {e}")
+            print(f"Error generating certificate: {str(e)}")
+            import traceback
+            traceback.print_exc()
             return None
-    
+
     def _create_base_certificate(self) -> Image.Image:
-        return Image.new('RGB', self.certificate_size, self.background_color)
+        """Create base certificate with improved background"""
+        # Create base image
+        certificate = Image.new('RGB', self.certificate_size, self.background_color)
+        
+        # Add subtle gradient background (optional)
+        draw = ImageDraw.Draw(certificate)
+        
+        # Add very subtle grain texture for professional look (optional)
+        width, height = certificate.size
+        for x in range(0, width, 4):
+            for y in range(0, height, 4):
+                if random.random() > 0.5:
+                    draw.point((x, y), fill=(250, 250, 250))
+        
+        return certificate
     
     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
-
+        """Save certificate with improved error handling"""
+        try:
+            temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
+            certificate.save(temp_file.name, 'PNG', quality=95)
+            print(f"Certificate saved to: {temp_file.name}")  # Debug info
+            return temp_file.name
+        except Exception as e:
+            print(f"Error saving certificate: {str(e)}")
+            return None
     
 class QuizApp:
     def __init__(self, api_key: str):