Update app.py

app.py

@@ -32,14 +32,14 @@ def main():
 
     # Dropdown for selecting a model
     st.sidebar.write("### Instructions:")
-    st.sidebar.write("1. Select a model below")
+    st.sidebar.write("# 1. Select a model below")
     model_options = data_df["model"].unique().tolist()  # Get model options
     selected_model = st.sidebar.selectbox(
         "Scored Models", model_options, help="Start typing to search for a model"
     )  # Searchable dropdown
 
     # Add step 2 instructions and move the Download button
-    st.sidebar.write("2. Download the label")
+    st.sidebar.write("# 2. Download the label")
 
     # Filter the data for the selected model
     model_data = data_df[data_df["model"] == selected_model].iloc[0]
@@ -81,16 +81,19 @@ def main():
     )
 
     # Step 3 instructions
-    st.sidebar.write("3. Share your label in technical reports, announcements, etc.")
+    st.sidebar.write("# 3. Share your label in technical reports, announcements, etc.")
 
 def create_label(background_image, model_data):
     """
     Create the label image by adding text from model_data to the background image.
+    Render at high resolution and scale down for sharper output.
     """
-    scale_factor = 2  # Render at 2x resolution for better text clarity
-    high_res_size = (800 * scale_factor, 600 * scale_factor)
+    # Scale factor for high resolution rendering
+    scale_factor = 2
+    original_size = background_image.size
+    high_res_size = (original_size[0] * scale_factor, original_size[1] * scale_factor)
 
-    # Resize background to high resolution
+    # Resize background proportionally for high resolution
     high_res_image = background_image.resize(high_res_size, Image.Resampling.LANCZOS)
     draw = ImageDraw.Draw(high_res_image)
 
@@ -129,8 +132,8 @@ def create_label(background_image, model_data):
     energy_text_width = bbox[2] - bbox[0]
     draw.text((energy_x - energy_text_width // 2, energy_y), energy_text, font=energy_font, fill="black")
 
-    # Downscale to the target resolution
-    final_image = high_res_image.resize((800, 600), Image.Resampling.LANCZOS)
+    # Downscale to original size
+    final_image = high_res_image.resize(original_size, Image.Resampling.LANCZOS)
     return final_image
 
 if __name__ == "__main__":