Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw, ImageFont
 import io
 
 def main():
-    # 1. Sidebar for Dropdown, Buttons, and Instructions
+    # Sidebar for dropdown, buttons, and instructions
     st.sidebar.title("AI Energy Score Label Generator")
     st.sidebar.write("### Instructions:")
     st.sidebar.write("1. Select a model from the dropdown.")
@@ -12,30 +12,30 @@ def main():
     st.sidebar.write("3. Download the label as a PNG.")
     st.sidebar.markdown("[Learn more about AI Energy Scores](https://example.com)")
 
-    # 2. Read Data from CSV
+    # Read Data from CSV
     try:
         data_df = pd.read_csv("data.csv")
     except FileNotFoundError:
         st.sidebar.error("Could not find 'data.csv'! Please make sure it's present.")
         return
 
-    # 3. Ensure the CSV has required columns
+    # Ensure the CSV has required columns
     required_columns = ["model", "provider", "date", "task", "hardware", "energy", "score"]
     for col in required_columns:
         if col not in data_df.columns:
             st.sidebar.error(f"The CSV file must contain a column named '{col}'.")
             return
 
-    # 4. Create a dropdown list based on unique values in the 'model' column
+    # Dropdown for selecting a model
     model_options = data_df["model"].unique().tolist()
     selected_model = st.sidebar.selectbox("Select a Model:", model_options)
 
-    # 5. Filter the data for the selected model
+    # Filter the data for the selected model
     model_data = data_df[data_df["model"] == selected_model].iloc[0]
 
-    # 6. Dynamically select the background image based on the score
+    # Dynamically select the background image based on the score
     try:
-        score = int(model_data["score"])  # Convert to int to avoid issues
+        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:
@@ -45,13 +45,13 @@ def main():
         st.sidebar.error(f"Invalid score '{model_data['score']}'. Score must be an integer.")
         return
 
-    # 7. Overlay the data on the image
+    # Generate the label with text
     generated_label = create_label(background, model_data)
 
-    # 8. Display the generated label in the main area
+    # Display the label
     st.image(generated_label, caption="Generated Label Preview")
 
-    # 9. Provide a download button in the sidebar
+    # Download button for the label
     img_buffer = io.BytesIO()
     generated_label.save(img_buffer, format="PNG")
     img_buffer.seek(0)
@@ -65,26 +65,25 @@ def main():
 
 def create_label(background_image, model_data):
     """
-    This function takes a background image and a row (model_data) from the CSV,
-    then draws text on the image. Finally, returns the modified image object.
+    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)
 
-    # Load the Inter variable font
+    # Load the Inter variable font (no LAYOUT_RAQM)
     try:
         inter_font_path = "Inter-VariableFont_opsz,wght.ttf"
-        title_font = ImageFont.truetype(inter_font_path, 16, layout_engine=ImageFont.LAYOUT_RAQM)
-        details_font = ImageFont.truetype(inter_font_path, 12, layout_engine=ImageFont.LAYOUT_RAQM)
-        energy_font = ImageFont.truetype(inter_font_path, 14, layout_engine=ImageFont.LAYOUT_RAQM)
+        title_font = ImageFont.truetype(inter_font_path, 16)  # Bold for title
+        details_font = ImageFont.truetype(inter_font_path, 12)  # Medium for details
+        energy_font = ImageFont.truetype(inter_font_path, 14)  # Medium for energy
     except Exception as e:
         st.error(f"Font loading failed: {e}")
         return label_img
 
-    # Define positions for the text groups (easy to tweak!)
-    title_x, title_y = 20, 20  # Position for title group
-    details_x, details_y = 300, 20  # Position for details group
-    energy_x, energy_y = 150, 400  # Position for energy group
+    # 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
 
     # Group 1: Title (Left-Justified)
     draw.text((title_x, title_y), f"Model: {model_data['model']}", font=title_font, fill="black")
@@ -97,11 +96,13 @@ def create_label(background_image, model_data):
         f"Hardware: {model_data['hardware']}"
     ]
     for i, line in enumerate(details_lines):
-        draw.text((details_x, details_y + i * 20), line, font=details_font, fill="black", anchor="ra")
+        text_width, _ = draw.textsize(line, font=details_font)
+        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']}"
-    draw.text((energy_x, energy_y), energy_text, font=energy_font, fill="black")
+    energy_text_width, _ = draw.textsize(energy_text, font=energy_font)
+    draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
 
     return label_img