Update app.py

app.py

@@ -4,12 +4,31 @@ from PIL import Image, ImageDraw, ImageFont
 import io
 
 def main():
-    # Sidebar for dropdown, buttons, and instructions
+    # Sidebar logo and title
     st.sidebar.image("logo.png", use_container_width=True)  # Display the logo at the top
     st.sidebar.title("Label Generator")
+
+    # Instructions with integrated dropdown
     st.sidebar.write("### Instructions:")
-    st.sidebar.write("1. Select a model from the dropdown.")
-    st.sidebar.write("2. Download the label.")
+    st.sidebar.write("1. Select a model from the dropdown below:")
+    model_options = data_df["model"].unique().tolist()  # Get model options
+    selected_model = st.sidebar.selectbox(
+        "Select a Model", 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. Review the label preview and download your label below:")
+    img_buffer = io.BytesIO()
+    generated_label.save(img_buffer, format="PNG")
+    img_buffer.seek(0)
+    st.sidebar.download_button(
+        label="Download Label as PNG",
+        data=img_buffer,
+        file_name="AIEnergyScore.png",
+        mime="image/png"
+    )
+
+    # Step 3 instructions
     st.sidebar.write("3. Share your label in technical reports, announcements, etc.")
     st.sidebar.markdown("[AI Energy Score Leaderboard](https://huggingface.co/spaces/AIEnergyScore/Leaderboard)")