Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -202,7 +202,8 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
|
|
202 |
|
203 |
title_x, title_y = 33, 150
|
204 |
details_x, details_y = 480, 256
|
205 |
-
energy_x
|
|
|
206 |
|
207 |
draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
|
208 |
draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
|
@@ -210,12 +211,19 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
|
|
210 |
details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
|
211 |
for i, line in enumerate(details_lines):
|
212 |
bbox = draw.textbbox((0, 0), line, font=details_font)
|
213 |
-
|
|
|
214 |
|
215 |
-
|
|
|
|
|
|
|
|
|
|
|
216 |
|
217 |
return bg_resized
|
218 |
|
219 |
|
|
|
220 |
if __name__ == "__main__":
|
221 |
main()
|
|
|
202 |
|
203 |
title_x, title_y = 33, 150
|
204 |
details_x, details_y = 480, 256
|
205 |
+
energy_x = 480 # Right margin for the energy value
|
206 |
+
energy_y = 472
|
207 |
|
208 |
draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
|
209 |
draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
|
|
|
211 |
details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
|
212 |
for i, line in enumerate(details_lines):
|
213 |
bbox = draw.textbbox((0, 0), line, font=details_font)
|
214 |
+
text_width = bbox[2] - bbox[0] # Get text width
|
215 |
+
draw.text((details_x - text_width, details_y + i * 47), line, font=details_font, fill="black")
|
216 |
|
217 |
+
# Right-align the energy text
|
218 |
+
energy_text = str(model_data['energy'])
|
219 |
+
energy_bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
|
220 |
+
energy_text_width = energy_bbox[2] - energy_bbox[0]
|
221 |
+
|
222 |
+
draw.text((energy_x - energy_text_width, energy_y), energy_text, font=energy_font, fill="black")
|
223 |
|
224 |
return bg_resized
|
225 |
|
226 |
|
227 |
+
|
228 |
if __name__ == "__main__":
|
229 |
main()
|