Update app.py

app.py

@@ -105,32 +105,64 @@ class TelegramChatGenerator:
         """Draw bubble with shadow and gradient effect"""
         x1, y1, x2, y2 = coords
         
-        # Draw shadow first
-        shadow_coords = (x1 + shadow_offset, y1 + shadow_offset, 
-                        x2 + shadow_offset, y2 + shadow_offset)
-        self.draw_rounded_rectangle(draw, shadow_coords, radius, (0, 0, 0, 30))
+        # Ensure coordinates are valid
+        if x2 <= x1 or y2 <= y1:
+            return
+        
+        # Ensure radius doesn't exceed bubble dimensions
+        max_radius = min((x2 - x1) // 2, (y2 - y1) // 2, radius)
+        if max_radius < 1:
+            max_radius = 1
+        
+        # Draw shadow first (with bounds check)
+        if shadow_offset > 0:
+            shadow_coords = (x1 + shadow_offset, y1 + shadow_offset, 
+                            x2 + shadow_offset, y2 + shadow_offset)
+            self.draw_rounded_rectangle(draw, shadow_coords, max_radius, (0, 0, 0, 30))
         
         # Draw main bubble
-        self.draw_rounded_rectangle(draw, coords, radius, fill)
+        self.draw_rounded_rectangle(draw, coords, max_radius, fill)
         
-        # Add subtle highlight
-        highlight_coords = (x1, y1, x2, y1 + 2)
-        lighter_fill = tuple(min(255, c + 20) for c in fill)
-        self.draw_rounded_rectangle(draw, highlight_coords, radius, lighter_fill)
+        # Add subtle highlight (only if bubble is tall enough)
+        if y2 - y1 > 4:
+            highlight_coords = (x1, y1, x2, y1 + 2)
+            lighter_fill = tuple(min(255, c + 20) for c in fill)
+            self.draw_rounded_rectangle(draw, highlight_coords, max_radius, lighter_fill)
     
     def draw_rounded_rectangle(self, draw, coords, radius, fill):
-        """Draw a perfect rounded rectangle"""
+        """Draw a perfect rounded rectangle with bounds checking"""
         x1, y1, x2, y2 = coords
         
-        # Main rectangles
-        draw.rectangle([x1 + radius, y1, x2 - radius, y2], fill=fill)
-        draw.rectangle([x1, y1 + radius, x2, y2 - radius], fill=fill)
-        
-        # Corner circles
-        draw.ellipse([x1, y1, x1 + 2 * radius, y1 + 2 * radius], fill=fill)
-        draw.ellipse([x2 - 2 * radius, y1, x2, y1 + 2 * radius], fill=fill)
-        draw.ellipse([x1, y2 - 2 * radius, x1 + 2 * radius, y2], fill=fill)
-        draw.ellipse([x2 - 2 * radius, y2 - 2 * radius, x2, y2], fill=fill)
+        # Ensure coordinates are valid
+        if x2 <= x1 or y2 <= y1:
+            return
+        
+        # Limit radius to prevent invalid coordinates
+        width = x2 - x1
+        height = y2 - y1
+        max_radius = min(width // 2, height // 2, radius)
+        
+        if max_radius < 1:
+            # If radius is too small, just draw a regular rectangle
+            draw.rectangle([x1, y1, x2, y2], fill=fill)
+            return
+        
+        # Main rectangles (only if they have positive dimensions)
+        if width > 2 * max_radius:
+            draw.rectangle([x1 + max_radius, y1, x2 - max_radius, y2], fill=fill)
+        if height > 2 * max_radius:
+            draw.rectangle([x1, y1 + max_radius, x2, y2 - max_radius], fill=fill)
+        
+        # Corner circles (only if there's space for them)
+        if width >= 2 * max_radius and height >= 2 * max_radius:
+            # Top-left
+            draw.ellipse([x1, y1, x1 + 2 * max_radius, y1 + 2 * max_radius], fill=fill)
+            # Top-right
+            draw.ellipse([x2 - 2 * max_radius, y1, x2, y1 + 2 * max_radius], fill=fill)
+            # Bottom-left
+            draw.ellipse([x1, y2 - 2 * max_radius, x1 + 2 * max_radius, y2], fill=fill)
+            # Bottom-right
+            draw.ellipse([x2 - 2 * max_radius, y2 - 2 * max_radius, x2, y2], fill=fill)
     
     def get_text_dimensions(self, text, font):
         """Get accurate text dimensions"""
@@ -177,6 +209,10 @@ class TelegramChatGenerator:
         padding = 50
         max_bubble_width = self.image_size - (padding * 2) - 60
         
+        # Ensure minimum dimensions
+        if max_bubble_width < 200:
+            max_bubble_width = 200
+        
         # Wrap text intelligently
         wrapped_message = self.wrap_text(message, self.font, max_bubble_width - 60)
         msg_width, msg_height = self.get_text_dimensions(wrapped_message, self.font)
@@ -194,88 +230,118 @@ class TelegramChatGenerator:
             reply_msg_w, reply_msg_h = self.get_text_dimensions(wrapped_reply, self.reply_font)
             reply_height = reply_username_h + reply_msg_h + 25
         
-        # Calculate total bubble dimensions
+        # Calculate total bubble dimensions with minimums
         bubble_padding = 25
-        bubble_width = min(max_bubble_width, max(350, msg_width + bubble_padding * 2))
-        bubble_height = username_height + msg_height + bubble_padding * 2 + 30
+        min_bubble_width = 300
+        bubble_width = max(min_bubble_width, min(max_bubble_width, msg_width + bubble_padding * 2))
+        bubble_height = max(80, username_height + msg_height + bubble_padding * 2 + 50)
         
         if reply_to:
             bubble_height += reply_height + 15
         
+        # Ensure bubble fits in canvas
+        if bubble_width > self.image_size - padding:
+            bubble_width = self.image_size - padding
+        if bubble_height > self.image_size - padding:
+            bubble_height = self.image_size - padding
+        
         # Center the bubble vertically in the square
-        bubble_x = self.image_size - bubble_width - padding
-        bubble_y = (self.image_size - bubble_height) // 2
+        bubble_x = max(padding, self.image_size - bubble_width - padding)
+        bubble_y = max(padding, (self.image_size - bubble_height) // 2)
+        
+        # Ensure bubble coordinates are valid
+        if bubble_x + bubble_width > self.image_size:
+            bubble_x = self.image_size - bubble_width - 10
+        if bubble_y + bubble_height > self.image_size:
+            bubble_y = self.image_size - bubble_height - 10
         
         # Draw main bubble with shadow
-        self.draw_advanced_bubble(
-            draw,
-            [bubble_x, bubble_y, bubble_x + bubble_width, bubble_y + bubble_height],
-            radius=20,
-            fill=self.bubble_color,
-            shadow_offset=4
-        )
+        if bubble_width > 0 and bubble_height > 0:
+            self.draw_advanced_bubble(
+                draw,
+                [bubble_x, bubble_y, bubble_x + bubble_width, bubble_y + bubble_height],
+                radius=20,
+                fill=self.bubble_color,
+                shadow_offset=4
+            )
         
         # Current position for content
         content_y = bubble_y + bubble_padding
         
         # Draw reply section if present
-        if reply_to and replied_username:
+        if reply_to and replied_username and bubble_width > 60:
             reply_y = content_y
-            reply_bubble_height = reply_height + 10
+            reply_bubble_height = max(30, reply_height + 10)
             
-            # Reply background
-            self.draw_rounded_rectangle(
-                draw,
-                [bubble_x + 15, reply_y, bubble_x + bubble_width - 15, reply_y + reply_bubble_height],
-                radius=12,
-                fill=self.reply_bubble_color
-            )
-            
-            # Blue accent line
-            draw.rectangle([bubble_x + 20, reply_y + 5, bubble_x + 24, reply_y + reply_bubble_height - 5], 
-                         fill=self.reply_line_color)
-            
-            # Reply content
-            reply_content_x = bubble_x + 35
-            reply_content_y = reply_y + 8
+            # Reply background (with bounds checking)
+            reply_bubble_x1 = bubble_x + 15
+            reply_bubble_x2 = bubble_x + bubble_width - 15
+            reply_bubble_y1 = reply_y
+            reply_bubble_y2 = reply_y + reply_bubble_height
             
-            # Reply username
-            draw.text((reply_content_x, reply_content_y), replied_username, 
-                     fill=self.reply_line_color, font=self.reply_font)
-            reply_content_y += 20
-            
-            # Reply message
-            for line in wrapped_reply.split('\n'):
-                draw.text((reply_content_x, reply_content_y), line, 
-                         fill=self.reply_text_color, font=self.reply_font)
-                reply_content_y += 20
+            if reply_bubble_x2 > reply_bubble_x1 and reply_bubble_y2 > reply_bubble_y1:
+                self.draw_rounded_rectangle(
+                    draw,
+                    [reply_bubble_x1, reply_bubble_y1, reply_bubble_x2, reply_bubble_y2],
+                    radius=12,
+                    fill=self.reply_bubble_color
+                )
+                
+                # Blue accent line
+                if reply_bubble_height > 10:
+                    draw.rectangle([bubble_x + 20, reply_y + 5, bubble_x + 24, reply_y + reply_bubble_height - 5], 
+                                 fill=self.reply_line_color)
+                
+                # Reply content
+                reply_content_x = bubble_x + 35
+                reply_content_y = reply_y + 8
+                
+                # Reply username
+                if reply_content_x < bubble_x + bubble_width - 20:
+                    draw.text((reply_content_x, reply_content_y), replied_username, 
+                             fill=self.reply_line_color, font=self.reply_font)
+                    reply_content_y += 20
+                    
+                    # Reply message
+                    for line in wrapped_reply.split('\n'):
+                        if reply_content_y < bubble_y + bubble_height - 30:
+                            draw.text((reply_content_x, reply_content_y), line, 
+                                     fill=self.reply_text_color, font=self.reply_font)
+                            reply_content_y += 20
             
             content_y += reply_bubble_height + 15
         
         # Main message username
-        draw.text((bubble_x + bubble_padding, content_y), username, 
-                 fill=self.username_color, font=self.username_font)
-        content_y += username_height + 10
+        if content_y < bubble_y + bubble_height - 50:
+            draw.text((bubble_x + bubble_padding, content_y), username, 
+                     fill=self.username_color, font=self.username_font)
+            content_y += username_height + 10
         
         # Main message text with better spacing
         for line in wrapped_message.split('\n'):
-            draw.text((bubble_x + bubble_padding, content_y), line, 
-                     fill=self.text_color, font=self.font)
-            content_y += 28
+            if content_y < bubble_y + bubble_height - 40:
+                draw.text((bubble_x + bubble_padding, content_y), line, 
+                         fill=self.text_color, font=self.font)
+                content_y += 28
         
         # Add timestamp
         current_time = datetime.now().strftime("%H:%M")
         time_bbox = self.time_font.getbbox(current_time)
         time_width = time_bbox[2] - time_bbox[0]
         
-        draw.text((bubble_x + bubble_width - time_width - bubble_padding, 
-                  bubble_y + bubble_height - 25), 
-                 current_time, fill=self.time_color, font=self.time_font)
+        timestamp_x = bubble_x + bubble_width - time_width - bubble_padding
+        timestamp_y = bubble_y + bubble_height - 25
+        
+        if timestamp_x > bubble_x and timestamp_y > bubble_y:
+            draw.text((timestamp_x, timestamp_y), 
+                     current_time, fill=self.time_color, font=self.time_font)
         
         # Add checkmark (delivered status)
         check_x = bubble_x + bubble_width - 25
         check_y = bubble_y + bubble_height - 22
-        self.draw_checkmarks(draw, check_x, check_y)
+        
+        if check_x > bubble_x and check_y > bubble_y:
+            self.draw_checkmarks(draw, check_x, check_y)
         
         return img