Update app.py

app.py

@@ -34,16 +34,22 @@ def main():
     model_data = data_df[data_df["model"] == selected_model].iloc[0]
 
     # Dynamically select the background image based on the score
-    try:
-        score = int(model_data["score"])  # Convert to int
-        background_path = f"{score}.png"  # E.g., "1.png", "2.png"
-        background = Image.open(background_path).convert("RGBA")
-    except FileNotFoundError:
-        st.sidebar.error(f"Could not find background image '{score}.png'. Using default background.")
-        background = Image.open("default_background.png").convert("RGBA")
-    except ValueError:
-        st.sidebar.error(f"Invalid score '{model_data['score']}'. Score must be an integer.")
-        return
+try:
+    score = int(model_data["score"])  # Convert to int
+    background_path = f"{score}.png"  # E.g., "1.png", "2.png"
+    background = Image.open(background_path).convert("RGBA")
+
+    # Proportional scaling to fit within the target size
+    target_size = (800, 600)  # Maximum width and height
+    background.thumbnail(target_size, Image.ANTIALIAS)
+
+except FileNotFoundError:
+    st.sidebar.error(f"Could not find background image '{score}.png'. Using default background.")
+    background = Image.open("default_background.png").convert("RGBA")
+    background.thumbnail(target_size, Image.ANTIALIAS)  # Resize default image proportionally
+except ValueError:
+    st.sidebar.error(f"Invalid score '{model_data['score']}'. Score must be an integer.")
+    return
 
     # Generate the label with text
     generated_label = create_label(background, model_data)
@@ -81,28 +87,27 @@ def create_label(background_image, model_data):
         return label_img
 
     # Define positions for each text group
-    title_x, title_y = 20, 20  # Top-left corner for title
-    details_x, details_y = label_img.width - 20, 20  # Top-right corner for details
-    energy_x, energy_y = label_img.width // 2, label_img.height - 50  # Center-bottom for energy
+    title_x, title_y = 100, 120
+    details_x, details_y = 200, 340
+    energy_x, energy_y = 600, 600
 
-    # Group 1: Title (Left-Justified)
-    draw.text((title_x, title_y), f"Model: {model_data['model']}", font=title_font, fill="black")
-    draw.text((title_x, title_y + 25), f"Provider: {model_data['provider']}", font=title_font, fill="black")
+    # Group 1: Title (Left-Justified, no prefixes)
+    draw.text((title_x, title_y), model_data['model'], font=title_font, fill="black")
+    draw.text((title_x, title_y + 25), model_data['provider'], font=title_font, fill="black")
 
-    # Group 2: Details (Right-Justified)
+    # Group 2: Details (Right-Justified, no prefixes)
     details_lines = [
-        f"Date: {model_data['date']}",
-        f"Task: {model_data['task']}",
-        f"Hardware: {model_data['hardware']}"
+        model_data['date'],
+        model_data['task'],
+        model_data['hardware']
     ]
     for i, line in enumerate(details_lines):
-        # Use textbbox to calculate text width
         bbox = draw.textbbox((0, 0), line, font=details_font)
-        text_width = bbox[2] - bbox[0]  # Right - Left
+        text_width = bbox[2] - bbox[0]
         draw.text((details_x - text_width, details_y + i * 20), line, font=details_font, fill="black")
 
-    # Group 3: Energy (Bottom-Center)
-    energy_text = f"Energy: {model_data['energy']}"
+    # Group 3: Energy (Bottom-Center, no prefixes)
+    energy_text = 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")