Update app.py

app.py

@@ -2,6 +2,20 @@ 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 = {
     'United States': 'USA',
     'United Kingdom': 'GBR',
@@ -143,13 +157,17 @@ def create_country_map(df, title):
     # Map country names to ISO codes
     country_counts['country'] = country_counts['country'].map(country_mapping)
     
+    # Assign color ranges based on counts
+    country_counts['color'] = country_counts['count'].apply(assign_color)
+    
     fig = px.choropleth(country_counts, 
                         locations="country",
-                        color="count",
+                        color="color",
                         hover_name="country",
                         title=title,
-                        projection="natural earth")
-    
+                        projection="natural earth",
+                        color_discrete_sequence=['#7FFF00', '#FFD700', '#FFA500', '#FF4500', '#DC143C', '#8B0000'],
+                        category_orders={"color": ['1-10', '10-50', '50-100', '100-500', '500-1000', '>1000']})
     return fig
 
 # Streamlit app