Update app.py

app.py

@@ -96,20 +96,12 @@ class TelegramChatGenerator:
         if max_radius < 1:
             max_radius = 1
         
-        # Draw multiple shadow layers for depth
-        for i in range(shadow_offset, 0, -1):
-            shadow_alpha = max(10, 40 - i * 5)  # Varying opacity
-            shadow_coords = (x1 + i, y1 + i, x2 + i, y2 + i)
-            shadow_color = (0, 0, 0, shadow_alpha)
-            
-            # Create shadow layer
-            shadow_layer = Image.new('RGBA', draw.im.size, (0, 0, 0, 0))
-            shadow_draw = ImageDraw.Draw(shadow_layer)
-            self.draw_rounded_rectangle(shadow_draw, shadow_coords, max_radius, shadow_color)
-            
-            # Blur the shadow
-            shadow_layer = shadow_layer.filter(ImageFilter.GaussianBlur(radius=1))
-            draw.im.paste(shadow_layer, (0, 0), shadow_layer)
+        # Draw simple shadow first
+        if shadow_offset > 0:
+            shadow_coords = (x1 + shadow_offset, y1 + shadow_offset, 
+                            x2 + shadow_offset, y2 + shadow_offset)
+            shadow_color = (0, 0, 0, 60)  # Semi-transparent black
+            self.draw_rounded_rectangle(draw, shadow_coords, max_radius, shadow_color)
         
         # Draw main bubble
         self.draw_rounded_rectangle(draw, coords, max_radius, fill)
@@ -138,22 +130,30 @@ class TelegramChatGenerator:
             draw.rectangle([x1, y1, x2, y2], fill=fill)
             return
         
+        # Handle both RGB and RGBA fills
+        if len(fill) == 3:
+            # RGB - add full opacity
+            fill_color = fill + (255,)
+        else:
+            # RGBA - use as is
+            fill_color = fill
+        
         # 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)
+            draw.rectangle([x1 + max_radius, y1, x2 - max_radius, y2], fill=fill_color)
         if height > 2 * max_radius:
-            draw.rectangle([x1, y1 + max_radius, x2, y2 - max_radius], fill=fill)
+            draw.rectangle([x1, y1 + max_radius, x2, y2 - max_radius], fill=fill_color)
         
         # 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)
+            draw.ellipse([x1, y1, x1 + 2 * max_radius, y1 + 2 * max_radius], fill=fill_color)
             # Top-right
-            draw.ellipse([x2 - 2 * max_radius, y1, x2, y1 + 2 * max_radius], fill=fill)
+            draw.ellipse([x2 - 2 * max_radius, y1, x2, y1 + 2 * max_radius], fill=fill_color)
             # Bottom-left
-            draw.ellipse([x1, y2 - 2 * max_radius, x1 + 2 * max_radius, y2], fill=fill)
+            draw.ellipse([x1, y2 - 2 * max_radius, x1 + 2 * max_radius, y2], fill=fill_color)
             # Bottom-right
-            draw.ellipse([x2 - 2 * max_radius, y2 - 2 * max_radius, x2, y2], fill=fill)
+            draw.ellipse([x2 - 2 * max_radius, y2 - 2 * max_radius, x2, y2], fill=fill_color)
     
     def get_text_dimensions(self, text, font):
         """Get accurate text dimensions"""