Update app.py

app.py

@@ -87,30 +87,32 @@ def create_label(background_image, model_data):
     """
     Create the label image by adding text from model_data to the background image.
     """
-    label_img = background_image.convert("RGBA")
-    draw = ImageDraw.Draw(label_img)
+    scale_factor = 2  # Render at 2x resolution for better text clarity
+    high_res_size = (800 * scale_factor, 600 * scale_factor)
 
-    # Load fonts
+    # Resize background to high resolution
+    high_res_image = background_image.resize(high_res_size, Image.Resampling.LANCZOS)
+    draw = ImageDraw.Draw(high_res_image)
+
+    # Load fonts with scaled sizes
     try:
-        # Use Inter_24pt-Bold.ttf for the title
-        title_font = ImageFont.truetype("Inter_24pt-Bold.ttf", size=22)  # Bold for title
-        # Use Inter-Regular.ttf for details and energy
-        details_font = ImageFont.truetype("Inter_18pt-Regular.ttf", size=18)  # Regular for details
-        energy_font = ImageFont.truetype("Inter_18pt-Medium.ttf", size=20)  # Medium for energy
+        title_font = ImageFont.truetype("Inter_24pt-Bold.ttf", size=22 * scale_factor)
+        details_font = ImageFont.truetype("Inter_18pt-Regular.ttf", size=18 * scale_factor)
+        energy_font = ImageFont.truetype("Inter_18pt-Medium.ttf", size=20 * scale_factor)
     except Exception as e:
         st.error(f"Font loading failed: {e}")
-        return label_img
+        return high_res_image
 
-    # Define positions for each text group
-    title_x, title_y = 28, 128
-    details_x, details_y = 375, 210
-    energy_x, energy_y = 350, 390
+    # Define scaled positions for text
+    title_x, title_y = 28 * scale_factor, 128 * scale_factor
+    details_x, details_y = 375 * scale_factor, 210 * scale_factor
+    energy_x, energy_y = 350 * scale_factor, 390 * scale_factor
 
     # Group 1: Title (Bold)
     draw.text((title_x, title_y), str(model_data['model']), font=title_font, fill="black")
-    draw.text((title_x, title_y + 30), str(model_data['provider']), font=title_font, fill="black")
+    draw.text((title_x, title_y + 30 * scale_factor), str(model_data['provider']), font=title_font, fill="black")
 
-    # Group 2: Details (Right-Justified, no prefixes)
+    # Group 2: Details (Right-Justified)
     details_lines = [
         str(model_data['date']),
         str(model_data['task']),
@@ -119,15 +121,17 @@ def create_label(background_image, model_data):
     for i, line in enumerate(details_lines):
         bbox = draw.textbbox((0, 0), line, font=details_font)
         text_width = bbox[2] - bbox[0]
-        draw.text((details_x - text_width, details_y + i * 40), line, font=details_font, fill="black")
+        draw.text((details_x - text_width, details_y + i * 40 * scale_factor), line, font=details_font, fill="black")
 
-    # Group 3: Energy (Bottom-Center, no prefixes)
-    energy_text = str(model_data['energy'])  # Ensure this is a string
+    # Group 3: Energy (Bottom-Center)
+    energy_text = str(model_data['energy'])
     bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
     energy_text_width = bbox[2] - bbox[0]
     draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
 
-    return label_img
+    # Downscale to the target resolution
+    final_image = high_res_image.resize((800, 600), Image.Resampling.LANCZOS)
+    return final_image
 
 if __name__ == "__main__":
     main()