Update app.py

app.py

@@ -2,23 +2,6 @@ import os
 import streamlit as st
 import folium
 from streamlit_folium import folium_static
-import pandas as pd
-from streamlit.runtime.state import SessionStateProxy
-
-@st.cache_resource
-def load_states():
-    return pd.read_csv("https://raw.githubusercontent.com/plotly/datasets/master/2011_us_ag_exports.csv")
-
-@st.cache_resource
-def load_cities():
-    return pd.read_csv("https://raw.githubusercontent.com/plotly/datasets/master/2014_us_cities.csv")
-
-def get_lat_lon(city, state):
-    cities_df = load_cities()
-    city_row = cities_df[(cities_df["City"] == city) & (cities_df["State"] == state)]
-    if not city_row.empty:
-        return city_row["lat"].values[0], city_row["lon"].values[0]
-    return None
 
 # Set the page configuration
 st.set_page_config(page_title="Google Maps 3D", layout="wide")
@@ -26,13 +9,21 @@ st.set_page_config(page_title="Google Maps 3D", layout="wide")
 # Set your Google Maps API key
 api_key = os.getenv("GM_TOKEN")
 
-# Load states data
-states_df = load_states()
-state_codes = sorted(states_df["code"].unique())
+# Define state codes
+state_codes = ["AL", "AK", "AZ", "AR", "CA", "CO", "CT", "DE", "FL", "GA", "HI", "ID", "IL", "IN", "IA", "KS", "KY", "LA", "ME", "MD", "MA", "MI", "MN", "MS", "MO", "MT", "NE", "NV", "NH", "NJ", "NM", "NY", "NC", "ND", "OH", "OK", "OR", "PA", "RI", "SC", "SD", "TN", "TX", "UT", "VT", "VA", "WA", "WV", "WI", "WY"]
+
+# Define a dictionary of city coordinates
+city_coords = {
+    "Mound, MN": [44.9327, -93.6663],
+    "San Francisco, CA": [37.7749, -122.4194],
+    "New York, NY": [40.7128, -74.0060],
+    "Chicago, IL": [41.8781, -87.6298],
+    # Add more cities and their coordinates as needed
+}
 
 # Initialize session state
 if "city" not in st.session_state:
-    st.session_state["city"] = "Mound"
+    st.session_state["city"] = "Mound, MN"
 if "state" not in st.session_state:
     st.session_state["state"] = "MN"
 
@@ -48,16 +39,16 @@ st.session_state["state"] = state
 
 # Get the latitude and longitude of the selected city and state
 if search_button:
-    lat_lon = get_lat_lon(city, state)
-    if lat_lon:
-        map_center = list(lat_lon)
+    city_state = f"{city}, {state}"
+    if city_state in city_coords:
+        map_center = city_coords[city_state]
         map_zoom = 12
     else:
         st.warning(f"Could not find the city '{city}' in the state '{state}'.")
         map_center = [37.7749, -122.4194]  # Example coordinates for San Francisco
         map_zoom = 12
 else:
-    map_center = get_lat_lon("Mound", "MN")
+    map_center = city_coords["Mound, MN"]
     map_zoom = 12
 
 # Create a Folium map instance