Update app.py

Increase chart size and fix color scale issue for filter selection that includes rank

app.py

@@ -66,21 +66,31 @@ def display_filter(col_names):
     return filter_selection
 
 
+# noinspection PyShadowingBuiltins
 def display_chart(df_tr, filter):
     """
     Displays various charts of the data.
     """
-    # Create chart
+    # Determine filter value in order to set proper color scale for chart
+    if 'rank' in filter.lower():
+        color_scale = 'plasma_r'
+    else:
+        color_scale = 'plasma'
+#te
+        # Create chart
     fig = px.choropleth(df_tr,
                         locations='State',
                         color=filter,
                         locationmode='USA-states',  # Set to plot as US States
                         title='2019 U.S. Energy {filter}'.format(filter=filter),
+                        color_continuous_scale=color_scale,
                         )
 
     # Add title and set USA as boundary for map
-    fig.update_layout(title_x=0.45,
+    fig.update_layout(title_x=0.4,
                       geo_scope='usa',  # Plot only the USA instead of globe
+                      height=800,
+                      width=1000
                       )
 
     # Display chart using Streamlit