Update app.py

app.py

@@ -4,13 +4,40 @@ from PIL import Image, ImageDraw, ImageFont
 import io
 
 def main():
+    # Inject custom CSS to change the color of selected tasks
+    st.markdown(
+        """
+        <style>
+            /* Change background color of selected items */
+            .stMultiSelect [data-baseweb="tag"] {
+                background-color: #3fa45bff !important; /* Custom green */
+                color: white !important; /* White text */
+                font-weight: bold;
+                border-radius: 5px;
+                padding: 5px 10px;
+            }
+
+            /* Change hover effect */
+            .stMultiSelect [data-baseweb="tag"]:hover {
+                background-color: #358d4d !important;
+            }
+
+            /* Style the dropdown input field */
+            .stMultiSelect input {
+                color: black !important;
+            }
+        </style>
+        """,
+        unsafe_allow_html=True,
+    )
+
     # Sidebar logo and title
     with st.sidebar:
-        col1, col2 = st.columns([1, 5])  # Shrink the logo column and expand the text column
+        col1, col2 = st.columns([1, 5])
 
         with col1:
             logo = Image.open("logo.png")
-            resized_logo = logo.resize((40, 40))  # Resize the logo
+            resized_logo = logo.resize((50, 50))
             st.image(resized_logo)
 
         with col2:
@@ -31,11 +58,8 @@ def main():
                 unsafe_allow_html=True,
             )
 
-    # (Removed the "Generate a Label to Display your AI Energy Score" section)
-
     st.sidebar.markdown("<hr style='border: 1px solid gray; margin: 15px 0;'>", unsafe_allow_html=True)
 
-    # Update instructions header
     st.sidebar.write("### Generate Label:")
 
     # Define the ordered list of tasks.
@@ -52,8 +76,8 @@ def main():
         "Sentence Similarity"
     ]
 
-    # Make the task selection label green and remove redundant text.
-    st.sidebar.markdown('<p style="color: green; font-size: 16px;">Task(s):</p>', unsafe_allow_html=True)
+    # Task selection
+    st.sidebar.write("#### 1. Select task(s) to view models")
     selected_tasks = st.sidebar.multiselect("", options=task_order, default=task_order)
 
     # Mapping from task to CSV file name.
@@ -64,30 +88,27 @@ def main():
         "Image Classification": "image_classification_energyscore.csv",
         "Image Captioning": "image_caption_energyscore.csv",
         "Summarization": "summarization_energyscore.csv",
-        "ASR": "asr_energyscore.csv",
+        "Speech-to-Text (ASR)": "asr_energyscore.csv",
         "Object Detection": "object_detection_energyscore.csv",
         "Question Answering": "question_answering_energyscore.csv",
         "Sentence Similarity": "sentence_similarity_energyscore.csv"
     }
 
-    # Default placeholder model data.
     default_model_data = {
-         'provider': "AI Provider",
-         'model': "Model Name",
-         'full_model': "AI Provider/Model Name",
-         'date': "",
-         'task': "",
-         'hardware': "",
-         'energy': "?",
-         'score': 5
+        'provider': "AI Provider",
+        'model': "Model Name",
+        'full_model': "AI Provider/Model Name",
+        'date': "",
+        'task': "",
+        'hardware': "",
+        'energy': "?",
+        'score': 5
     }
 
     if not selected_tasks:
-        # If no tasks are selected, use generic placeholder.
         model_data = default_model_data
     else:
         dfs = []
-        # Load and process each CSV corresponding to the selected tasks.
         for task in selected_tasks:
             file_name = task_to_file[task]
             try:
@@ -99,17 +120,12 @@ def main():
                 st.sidebar.error(f"Error reading '{file_name}' for task {task}: {e}")
                 continue
 
-            # Save the original full model string and then split the "model" column
             df['full_model'] = df['model']
             df[['provider', 'model']] = df['model'].str.split(pat='/', n=1, expand=True)
-            # Round total_gpu_energy to 3 decimal places and assign to 'energy'
             df['energy'] = df['total_gpu_energy'].round(3)
-            # Use the energy_score column as 'score' (fill missing values with 1 to avoid casting errors)
             df['score'] = df['energy_score'].fillna(1).astype(int)
-            # Hardcode date and hardware
             df['date'] = "February 2025"
             df['hardware'] = "NVIDIA H100-80GB"
-            # Set the task from the file name mapping
             df['task'] = task
 
             dfs.append(df)
@@ -121,7 +137,6 @@ def main():
             if data_df.empty:
                 model_data = default_model_data
             else:
-                # In the scored model dropdown show the full model string.
                 model_options = data_df["full_model"].unique().tolist()
                 selected_model = st.sidebar.selectbox(
                     "Scored Models",
@@ -130,10 +145,9 @@ def main():
                 )
                 model_data = data_df[data_df["full_model"] == selected_model].iloc[0]
 
-    st.sidebar.write("#### 2. Select a model below")
+    st.sidebar.write("#### 2. Select a model to generate label")
     st.sidebar.write("#### 3. Download the label")
     
-    # Select background by score (using generic placeholder score=5 if applicable)
     try:
         score = int(model_data["score"])
         background_path = f"{score}.png"
@@ -145,7 +159,6 @@ def main():
         st.sidebar.error(f"Invalid score '{model_data['score']}'. Score must be an integer.")
         return
 
-    # Keep the final label size at 520×728
     final_size = (520, 728)
     generated_label = create_label_single_pass(background, model_data, final_size)
 
@@ -172,14 +185,9 @@ def main():
 
 
 def create_label_single_pass(background_image, model_data, final_size=(520, 728)):
-    """
-    Resizes the background to 520×728, then draws text onto it.
-    """
-    # 1. Resize background to final_size
     bg_resized = background_image.resize(final_size, Image.Resampling.LANCZOS)
     draw = ImageDraw.Draw(bg_resized)
 
-    # 2. Load fonts at sizes appropriate for a 520×728 label
     try:
         title_font = ImageFont.truetype("Inter_24pt-Bold.ttf", size=27)
         details_font = ImageFont.truetype("Inter_18pt-Regular.ttf", size=23)
@@ -188,37 +196,22 @@ def create_label_single_pass(background_image, model_data, final_size=(520, 728)
         st.error(f"Font loading failed: {e}")
         return bg_resized
 
-    # 3. Place your text.
-    # Flip the order so that the provider (AI Developer) is shown first and the model name second.
     title_x, title_y = 33, 150
     details_x, details_y = 480, 256
     energy_x, energy_y = 480, 472
 
-    # Text 1 (title) – show provider first then model name
     draw.text((title_x, title_y), str(model_data['provider']), font=title_font, fill="black")
     draw.text((title_x, title_y + 38), str(model_data['model']), font=title_font, fill="black")
 
-    # Text 2 (details)
-    details_lines = [
-        str(model_data['date']),
-        str(model_data['task']),
-        str(model_data['hardware'])
-    ]
+    details_lines = [str(model_data['date']), str(model_data['task']), str(model_data['hardware'])]
     for i, line in enumerate(details_lines):
         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 * 47), line, font=details_font, fill="black")
+        draw.text((details_x - bbox[2], details_y + i * 47), line, font=details_font, fill="black")
 
-    # Text 3 (energy)
-    energy_text = str(model_data['energy'])
-    bbox = draw.textbbox((0, 0), energy_text, font=energy_font)
-    energy_text_width = bbox[2] - bbox[0]
-    # Right-align the energy text at energy_x
-    draw.text((energy_x - energy_text_width, energy_y), energy_text, font=energy_font, fill="black")
+    draw.text((energy_x, energy_y), str(model_data['energy']), font=energy_font, fill="black")
 
     return bg_resized
 
 
 if __name__ == "__main__":
-    main()
+    main()