Update app.py

app.py

@@ -391,15 +391,20 @@ class CertificateGenerator:
     def _add_photo(self, certificate: Image.Image, photo_path: str):
         """Add a clear circular profile photo in the top-right corner"""
         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)
+            print(f"Loaded photo: {photo.size}, {photo.mode}")  # Debug info
             
-            # Define size for circular photo (slightly larger for better quality)
+            # Define size for circular photo
             size = (120, 120)
             
-            # Convert to RGBA if not already
-            if photo.mode != 'RGBA':
-                photo = photo.convert('RGBA')
+            # 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)
@@ -409,47 +414,43 @@ class CertificateGenerator:
             # Resize photo maintaining aspect ratio
             aspect = photo.width / photo.height
             if aspect > 1:
-                # Width is greater
                 new_height = size[1]
                 new_width = int(new_height * aspect)
-                photo = photo.resize((new_width, new_height), Image.Resampling.LANCZOS)
-                # Center crop
-                left = (new_width - size[0]) // 2
-                photo = photo.crop((left, 0, left + size[0], size[1]))
             else:
-                # Height is greater
                 new_width = size[0]
                 new_height = int(new_width / aspect)
-                photo = photo.resize((new_width, new_height), Image.Resampling.LANCZOS)
-                # Center crop
+            
+            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 output image with transparency
+            # Create circular photo
             output = Image.new('RGBA', size, (0, 0, 0, 0))
             output.paste(photo, (0, 0))
-            
-            # Apply antialiasing to the mask
-            mask = mask.filter(ImageFilter.GaussianBlur(radius=1))
             output.putalpha(mask)
             
-            # Position in top-right corner with padding
-            padding_right = 60
-            padding_top = 40
-            position = (certificate.width - size[0] - padding_right, padding_top)
+            # Position in top-right corner
+            photo_x = certificate.width - size[0] - 60  # 60px from right
+            photo_y = 40  # 40px from top
             
-            # Add a subtle white background circle for better visibility
+            # Add white background circle
             bg = Image.new('RGBA', size, (255, 255, 255, 255))
-            bg_mask = mask.copy()
-            certificate.paste(bg, position, mask=bg_mask)
+            certificate.paste(bg, (photo_x, photo_y), mask=mask)
             
             # Paste the photo
-            certificate.paste(output, position, mask=output)
-        
+            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: {str(e)}")
-            print(f"Photo path: {photo_path}")
-            print(f"Photo details: {getattr(photo, 'size', 'unknown size')}, {getattr(photo, 'mode', 'unknown mode')}")
+            import traceback
+            traceback.print_exc()
     
     def generate(
         self,
@@ -459,43 +460,37 @@ class CertificateGenerator:
         company_logo: Optional[str] = None,
         participant_photo: Optional[str] = None
     ) -> str:
-        """Generate certificate with improved layout"""
+        """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 first
+            # Add border
             self._add_professional_border(draw)
             
-            # Add photo if provided (now centered)
-            if participant_photo:
-                self._add_photo(certificate, participant_photo)
-                # Adjust vertical spacing for content when photo is present
-                content_start_y = 180  # Increased to accommodate photo
-            else:
-                content_start_y = 140  # Original spacing when no photo
-            
             # Load fonts
             fonts = self._load_fonts()
             
-            # Add content with adjusted vertical position
-            self._add_content(
-                draw, 
-                fonts, 
-                str(name), 
-                str(course_name), 
-                float(score),
-                y_offset=content_start_y
-            )
-            
             # Add company logo if provided
-            if company_logo:
+            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:
@@ -516,10 +511,15 @@ class CertificateGenerator:
         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):