add query cache

app.py

@@ -30,11 +30,14 @@ def connect_to_db():
         st.error(f"An error occurred while connecting to the database: {e}")
         return None
 
-conn = connect_to_db()
+# conn = connect_to_db()
 
+# Perform query.
+# Uses st.cache_data to only rerun when the query changes or after 10 min.
+@st.cache_data(ttl=600)
 # Function to fetch previously saved data points
 def fetch_saved_data():
-    # conn = connect_to_db()
+    conn = connect_to_db()
     if conn:
         cursor = conn.cursor()
         cursor.execute('SELECT "X", "Y", description, gl_certainty FROM public.gettinglost_tracking')
@@ -44,7 +47,10 @@ def fetch_saved_data():
         return data
     return []
 
+# Perform query.
+# Uses st.cache_data to only rerun when the query changes or after 10 min.
 # Function to save data to the database
+@st.cache_data(ttl=600)
 def save_data(lat, lon, description, rating):
     conn = connect_to_db()
     if conn:
@@ -53,7 +59,7 @@ def save_data(lat, lon, description, rating):
                        (lat, lon, lon, lat, description, rating))
         conn.commit()
         cursor.close()
-        conn.close()
+        # conn.close()
         st.sidebar.success("Data recorded successfully!")
     else:
         st.sidebar.error("Failed to save data.")
@@ -67,6 +73,7 @@ draw = Draw(draw_options={'polyline': False, "marker": True, 'rectangle': False,
 m.add_child(draw)
 
 # Display previously saved data points on the map
+# conn = connect_to_db()
 for point in fetch_saved_data():
     lat, lon, desc, rating = point
     folium.Marker([lat, lon], popup=f"{desc} (Certainty: {rating})").add_to(m)