Update app.py

app.py

@@ -8,6 +8,13 @@ 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 remove_background(image):
     # Save the image to a specific location
     filename = f"image_{uuid.uuid4()}.png"  # Generates a universally unique identifier (UUID) for the filename
@@ -25,7 +32,6 @@ def superimpose(image_with_text, overlay_image):
     # image_with_text.save("output_image.png")
     return image_with_text
 
-
 def add_text_to_image(
     input_image,
     text,
@@ -33,7 +39,8 @@ def add_text_to_image(
     color,
     opacity,
     x_position,
-    y_position
+    y_position,
+    thickness
 ):
     """
     Add text to an image with customizable properties
@@ -82,12 +89,18 @@ def add_text_to_image(
     actual_x = int((image.width - text_width) * (x_position / 100))
     actual_y = int((image.height - text_height) * (y_position / 100))
 
-    # Draw the main text
-    draw.text(
-        (actual_x, actual_y),
-        text,
-        font=font,
-        fill=(*rgb_color, int(opacity))
+    # Create final color with opacity
+    text_color = (*rgb_color, int(opacity))
+    
+    # Draw the text with stroke for thickness
+    add_text_with_stroke(
+        draw, 
+        text, 
+        actual_x, 
+        actual_y, 
+        font, 
+        text_color, 
+        int(thickness)
     )
 
     # Combine the original image with the text overlay
@@ -117,6 +130,8 @@ def create_interface():
                 text_input = gr.Textbox(label="Enter Text", placeholder="Type your text here...")
                 font_size = gr.Slider(minimum=10, maximum=800, value=400, step=10, 
                                     label="Font Size")
+                thickness = gr.Slider(minimum=0, maximum=20, value=0, step=1,
+                                    label="Text Thickness")
                 color_dropdown = gr.Dropdown(
                     choices=["White", "Black", "Red", "Green", "Blue", "Yellow", "Purple"],
                     value="White",
@@ -146,7 +161,8 @@ def create_interface():
                 color_dropdown,
                 opacity_slider,
                 x_position,
-                y_position
+                y_position,
+                thickness
             ],
             outputs=output_image
         )
@@ -161,7 +177,8 @@ def create_interface():
                     "Purple",
                     150,
                     50,
-                    21
+                    21,
+                    9
                 ],
                 [
                     "pear.jpg",
@@ -170,7 +187,8 @@ def create_interface():
                     "Black",
                     100,
                     50,
-2
+                    2,
+                    5
                 ],
                 [
                     "sample_text_image.jpeg",
@@ -179,7 +197,8 @@ def create_interface():
                     "Black",
                     150,
                     50,
-                    2
+                    2,
+                    8
                 ],
             ],
             inputs=[
@@ -189,7 +208,8 @@ def create_interface():
                 color_dropdown,
                 opacity_slider,
                 x_position,
-                y_position
+                y_position,
+                thickness
             ],
             outputs=output_image,
             fn=add_text_to_image,