Update app.py

app.py

@@ -7,12 +7,12 @@ def main():
     # Sidebar logo and title
     with st.sidebar:
         col1, col2 = st.columns([1, 5])  # Shrink the logo column and expand the text column
-
+    
         with col1:
             logo = Image.open("logo.png")
             resized_logo = logo.resize((40, 40))  # Resize the logo
             st.image(resized_logo)
-
+    
         with col2:
             st.markdown(
                 """
@@ -77,11 +77,11 @@ def main():
         st.sidebar.error(f"Invalid score '{model_data['score']}'. Score must be an integer.")
         return
 
-    # Keep the final label size at 260x364
-    final_size = (260, 364)
+    # Keep the final label size at 520×728
+    final_size = (520, 728)
     generated_label = create_label_single_pass(background, model_data, final_size)
 
-    st.image(generated_label, caption="Generated Label Preview", width=final_size[0])
+    st.image(generated_label, caption="Generated Label Preview", width=520)
 
     img_buffer = io.BytesIO()
     generated_label.save(img_buffer, format="PNG")
@@ -97,37 +97,33 @@ def main():
     st.sidebar.write("#### 3. Share your label in technical reports, announcements, etc.")
 
 
-def create_label_single_pass(background_image, model_data, final_size=(260, 364)):
+def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
     """
-    Resizes the background to 260×364 with sharper output and draws text onto it.
+    Resizes the background to 520×728, then draws text onto it.
     """
-    # High-resolution canvas scaling factor (e.g., 4x)
-    scale_factor = 4
-    high_res_size = (final_size[0] * scale_factor, final_size[1] * scale_factor)
-
-    # Step 1: Resize background to high-resolution canvas
-    high_res_image = background_image.resize(high_res_size, Image.Resampling.LANCZOS)
+    # 1. Resize background to final_size
+    bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
+    draw = ImageDraw.Draw(bg_resized)
 
-    # Step 2: Draw text on the high-resolution image
-    draw = ImageDraw.Draw(high_res_image)
-
-    # Load fonts with scaled sizes
+    # 2. Load fonts at sizes appropriate for a 520×728 label
     try:
-        title_font = ImageFont.truetype("Inter_24pt-Bold.ttf", size=12 * scale_factor)
-        details_font = ImageFont.truetype("Inter_18pt-Regular.ttf", size=10 * scale_factor)
-        energy_font = ImageFont.truetype("Inter_18pt-Medium.ttf", size=12 * scale_factor)
+        title_font = ImageFont.truetype("Inter_24pt-Bold.ttf", size=27)
+        details_font = ImageFont.truetype("Inter_18pt-Regular.ttf", size=23)
+        energy_font = ImageFont.truetype("Inter_18pt-Medium.ttf", size=24)
     except Exception as e:
         st.error(f"Font loading failed: {e}")
-        return high_res_image
+        return bg_resized
 
-    # Scaled positions for text
-    title_x, title_y = 16 * scale_factor, 75 * scale_factor
-    details_x, details_y = 245 * scale_factor, 130 * scale_factor
-    energy_x, energy_y = 245 * scale_factor, 250 * scale_factor
+    # 3. Place your text. 
+    #    You may need to experiment with x/y coordinates or font sizes
+    #    to make it look right in 520×728.
+    title_x, title_y = 33, 150
+    details_x, details_y = 480, 256
+    energy_x, energy_y = 480, 472
 
     # Text 1 (title)
     draw.text((title_x, title_y), str(model_data['model']), font=title_font, fill="black")
-    draw.text((title_x, title_y + 20 * scale_factor), str(model_data['provider']), font=title_font, fill="black")
+    draw.text((title_x, title_y + 38), str(model_data['provider']), font=title_font, fill="black")
 
     # Text 2 (details)
     details_lines = [
@@ -139,7 +135,7 @@ def create_label_single_pass(background_image, model_data, final_size=(260, 364)
         bbox = draw.textbbox((0, 0), line, font=details_font)
         text_width = bbox[2] - bbox[0]
         # Right-justify the details text at details_x
-        draw.text((details_x - text_width, details_y + i * 25 * scale_factor), line, font=details_font, fill="black")
+        draw.text((details_x - text_width, details_y + i*47), line, font=details_font, fill="black")
 
     # Text 3 (energy)
     energy_text = str(model_data['energy'])
@@ -148,9 +144,8 @@ def create_label_single_pass(background_image, model_data, final_size=(260, 364)
     # Right-align the energy text at energy_x
     draw.text((energy_x - energy_text_width, energy_y), energy_text, font=energy_font, fill="black")
 
-    # Step 3: Downscale to final size
-    final_image = high_res_image.resize(final_size, Image.Resampling.LANCZOS)
-    return final_image
+    return bg_resized
+
 
 if __name__ == "__main__":
     main()