cache data

app.py

@@ -12,12 +12,19 @@ file_url = st.query_params.get("file_url", None)
 print(f"{file_url=}")
 
 if file_url:
-    if file_url.startswith("https://drive.google.com/file/d/"):
-        ID = file_url.replace("https://drive.google.com/file/d/", "").split("/")[0]
-        file_url = f"https://drive.google.com/uc?id={ID}"
+    if ("file_url" in st.session_state) and (st.session_state.file_url == file_url):
+        st.toast("Using cached data")
+        input_gdf = st.session_state.input_gdf
+    else:
+        if file_url.startswith("https://drive.google.com/file/d/"):
+            ID = file_url.replace("https://drive.google.com/file/d/", "").split("/")[0]
+            file_url = f"https://drive.google.com/uc?id={ID}"
 
-    input_gdf = gpd.read_file(file_url)
-    input_gdf = input_gdf.to_crs(epsg=7761)  # Gujarat zone
+        input_gdf = gpd.read_file(file_url)
+        input_gdf = input_gdf.to_crs(epsg=7761)  # Gujarat zone
+        st.session_state.file_url = file_url
+        st.session_state.input_gdf = input_gdf
+        st.toast("Data loaded and cached")
 
     def format_fn(x):
         return input_gdf.drop(columns=["geometry"]).loc[x].to_dict()