Update app.py

app.py

@@ -1,7 +1,19 @@
 import streamlit as st
 import pandas as pd
 import plotly.express as px
-
+def assign_color(count):
+    if count <= 10:
+        return '1-10'
+    elif count <= 50:
+        return '10-50'
+    elif count <= 100:
+        return '50-100'
+    elif count <= 500:
+        return '100-500'
+    elif count <= 1000:
+        return '500-1000'
+    else:
+        return '>1000'
 # Country mapping dictionary
 country_mapping = {
     'United States': 'USA',