Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -8,25 +8,12 @@ import uuid
|
|
8 |
|
9 |
client = Client("ysharma/BiRefNet_for_text_writing")
|
10 |
|
11 |
-
def
|
12 |
-
"""Helper function to draw text with stroke
|
13 |
-
# Draw
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
# Draw the shadow text with stroke effect
|
19 |
-
shadow_x = x + shadow_offset[0]
|
20 |
-
shadow_y = y + shadow_offset[1]
|
21 |
-
|
22 |
-
for adj_x in range(-stroke_width, stroke_width + 1):
|
23 |
-
for adj_y in range(-stroke_width, stroke_width + 1):
|
24 |
-
draw.text(
|
25 |
-
(shadow_x + adj_x, shadow_y + adj_y),
|
26 |
-
text,
|
27 |
-
font=font,
|
28 |
-
fill=shadow_color
|
29 |
-
)
|
30 |
|
31 |
def remove_background(image):
|
32 |
# Save the image to a specific location
|
@@ -53,8 +40,7 @@ def add_text_to_image(
|
|
53 |
opacity,
|
54 |
x_position,
|
55 |
y_position,
|
56 |
-
thickness
|
57 |
-
use_shadow
|
58 |
):
|
59 |
"""
|
60 |
Add text to an image with customizable properties
|
@@ -106,20 +92,15 @@ def add_text_to_image(
|
|
106 |
# Create final color with opacity
|
107 |
text_color = (*rgb_color, int(opacity))
|
108 |
|
109 |
-
#
|
110 |
-
|
111 |
-
|
112 |
-
# Draw the text with effects
|
113 |
-
add_text_with_effects(
|
114 |
draw,
|
115 |
text,
|
116 |
actual_x,
|
117 |
actual_y,
|
118 |
font,
|
119 |
text_color,
|
120 |
-
int(thickness)
|
121 |
-
shadow=use_shadow,
|
122 |
-
shadow_offset=shadow_offset
|
123 |
)
|
124 |
|
125 |
# Combine the original image with the text overlay
|
@@ -162,7 +143,6 @@ def create_interface():
|
|
162 |
label="X Position (%)")
|
163 |
y_position = gr.Slider(minimum=0, maximum=100, value=50, step=1,
|
164 |
label="Y Position (%)")
|
165 |
-
shadow_checkbox = gr.Checkbox(label="Add Shadow Effect", value=False)
|
166 |
|
167 |
with gr.Column():
|
168 |
# Output image
|
@@ -182,8 +162,7 @@ def create_interface():
|
|
182 |
opacity_slider,
|
183 |
x_position,
|
184 |
y_position,
|
185 |
-
thickness
|
186 |
-
shadow_checkbox
|
187 |
],
|
188 |
outputs=output_image
|
189 |
)
|
@@ -199,8 +178,7 @@ def create_interface():
|
|
199 |
150,
|
200 |
50,
|
201 |
21,
|
202 |
-
9
|
203 |
-
True
|
204 |
],
|
205 |
[
|
206 |
"pear.jpg",
|
@@ -210,8 +188,7 @@ def create_interface():
|
|
210 |
100,
|
211 |
50,
|
212 |
2,
|
213 |
-
5
|
214 |
-
True
|
215 |
],
|
216 |
[
|
217 |
"sample_text_image.jpeg",
|
@@ -221,8 +198,7 @@ def create_interface():
|
|
221 |
150,
|
222 |
50,
|
223 |
2,
|
224 |
-
8
|
225 |
-
True
|
226 |
],
|
227 |
],
|
228 |
inputs=[
|
@@ -233,8 +209,7 @@ def create_interface():
|
|
233 |
opacity_slider,
|
234 |
x_position,
|
235 |
y_position,
|
236 |
-
thickness
|
237 |
-
shadow_checkbox,
|
238 |
],
|
239 |
outputs=output_image,
|
240 |
fn=add_text_to_image,
|
|
|
8 |
|
9 |
client = Client("ysharma/BiRefNet_for_text_writing")
|
10 |
|
11 |
+
def add_text_with_stroke(draw, text, x, y, font, text_color, stroke_width):
|
12 |
+
"""Helper function to draw text with stroke"""
|
13 |
+
# Draw the stroke/outline
|
14 |
+
for adj_x in range(-stroke_width, stroke_width + 1):
|
15 |
+
for adj_y in range(-stroke_width, stroke_width + 1):
|
16 |
+
draw.text((x + adj_x, y + adj_y), text, font=font, fill=text_color)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
|
18 |
def remove_background(image):
|
19 |
# Save the image to a specific location
|
|
|
40 |
opacity,
|
41 |
x_position,
|
42 |
y_position,
|
43 |
+
thickness
|
|
|
44 |
):
|
45 |
"""
|
46 |
Add text to an image with customizable properties
|
|
|
92 |
# Create final color with opacity
|
93 |
text_color = (*rgb_color, int(opacity))
|
94 |
|
95 |
+
# Draw the text with stroke for thickness
|
96 |
+
add_text_with_stroke(
|
|
|
|
|
|
|
97 |
draw,
|
98 |
text,
|
99 |
actual_x,
|
100 |
actual_y,
|
101 |
font,
|
102 |
text_color,
|
103 |
+
int(thickness)
|
|
|
|
|
104 |
)
|
105 |
|
106 |
# Combine the original image with the text overlay
|
|
|
143 |
label="X Position (%)")
|
144 |
y_position = gr.Slider(minimum=0, maximum=100, value=50, step=1,
|
145 |
label="Y Position (%)")
|
|
|
146 |
|
147 |
with gr.Column():
|
148 |
# Output image
|
|
|
162 |
opacity_slider,
|
163 |
x_position,
|
164 |
y_position,
|
165 |
+
thickness
|
|
|
166 |
],
|
167 |
outputs=output_image
|
168 |
)
|
|
|
178 |
150,
|
179 |
50,
|
180 |
21,
|
181 |
+
9
|
|
|
182 |
],
|
183 |
[
|
184 |
"pear.jpg",
|
|
|
188 |
100,
|
189 |
50,
|
190 |
2,
|
191 |
+
5
|
|
|
192 |
],
|
193 |
[
|
194 |
"sample_text_image.jpeg",
|
|
|
198 |
150,
|
199 |
50,
|
200 |
2,
|
201 |
+
8
|
|
|
202 |
],
|
203 |
],
|
204 |
inputs=[
|
|
|
209 |
opacity_slider,
|
210 |
x_position,
|
211 |
y_position,
|
212 |
+
thickness
|
|
|
213 |
],
|
214 |
outputs=output_image,
|
215 |
fn=add_text_to_image,
|