Update app.py

app.py

@@ -8,12 +8,25 @@ import uuid
 
 client = Client("ysharma/BiRefNet_for_text_writing")
 
-def add_text_with_stroke(draw, text, x, y, font, text_color, stroke_width):
-    """Helper function to draw text with stroke"""
-    # Draw the stroke/outline
-    for adj_x in range(-stroke_width, stroke_width + 1):
-        for adj_y in range(-stroke_width, stroke_width + 1):
-            draw.text((x + adj_x, y + adj_y), text, font=font, fill=text_color)
+def add_text_with_effects(draw, text, x, y, font, text_color, stroke_width, shadow=False, shadow_offset=(5, 5)):
+    """Helper function to draw text with stroke and optional shadow"""
+    # Draw shadow first if enabled
+    if shadow:
+        # Create shadow color (darker version of text color or black)
+        shadow_color = (0, 0, 0, text_color[3])  # Black with same opacity as text
+        
+        # Draw the shadow text with stroke effect
+        shadow_x = x + shadow_offset[0]
+        shadow_y = y + shadow_offset[1]
+        
+        for adj_x in range(-stroke_width, stroke_width + 1):
+            for adj_y in range(-stroke_width, stroke_width + 1):
+                draw.text(
+                    (shadow_x + adj_x, shadow_y + adj_y),
+                    text,
+                    font=font,
+                    fill=shadow_color
+                )
 
 def remove_background(image):
     # Save the image to a specific location
@@ -221,7 +234,7 @@ def create_interface():
                 x_position,
                 y_position,
                 thickness,
-                shadow_checkbox
+                shadow_checkbox,
             ],
             outputs=output_image,
             fn=add_text_to_image,