feat: show park stats

src/app.py

@@ -80,12 +80,18 @@ def sidebar() -> dict | None:
     )
 
 
+def display_stats() -> None:
+    """Display statistics about the images"""
+    st.title("Statistics")
+    st.write("Number of images: ", image_manager.get_image_count())
+    st.write("Number of parks: ", park_manager.get_park_count())
+
 def main() -> None:
     """Main function to run the app"""
     park = sidebar()
     if not park:
-        st.write("Please select a park")
-        return
+        display_stats()
+        st.stop()
 
     images = get_park_images(park["id"])
     predictions = get_image_predictions(images)

src/data_models/image_manager.py

@@ -95,3 +95,16 @@ class ImageManager:
     def close_connection(self):
         """Close the connection."""
         self.session.close()
+
+    def get_image_count(self):
+        """
+        Get the number of images in the `images` table.
+        Returns:
+            int: Number of images.
+        """
+        query = text("SELECT COUNT(*) FROM images")
+        try:
+            result = self.session.execute(query).fetchone()
+            return result[0]
+        except Exception as e:
+            raise Exception(f"An error occurred while getting the image count: {e}")

src/data_models/park_manager.py

@@ -75,3 +75,12 @@ class ParkManager:
     def close_connection(self):
         """Close the connection."""
         self.session.close()
+
+    def get_park_count(self):
+        """Get the number of parks in the database."""
+        query = text("SELECT COUNT(*) FROM parks")
+        try:
+            result = self.session.execute(query).fetchone()
+            return result[0]
+        except Exception as e:
+            raise Exception(f"An error occurred while getting the park count: {e}")