feat: app download image from bucket

src/app.py

@@ -8,12 +8,13 @@ from data_models.sql_connection import get_db_connection
 from data_models.park_manager import ParkManager
 from data_models.image_manager import ImageManager
 from data_models.openai_manager import OpenAIManager
-
+from cloud_storage import GoogleCloudStorage
 
 engine, session = get_db_connection()
 park_manager = ParkManager()
 image_manager = ImageManager()
 openai_manager = OpenAIManager()
+gcs = GoogleCloudStorage("ialab-fr-7047e56bef0b.json")
 
 
 def get_image_predictions(images: list[dict]) -> dict:
@@ -31,7 +32,7 @@ def get_image_predictions(images: list[dict]) -> dict:
     return predictions
 
 
-def image_gallery(images: list[dict], predictions:dict) -> None:
+def image_gallery(images: list[dict], predictions: dict) -> None:
     """Display a gallery of images in a streamlit app
 
     Args:
@@ -42,9 +43,9 @@ def image_gallery(images: list[dict], predictions:dict) -> None:
     columns = st.columns(3)
     for index, image in enumerate(images):
         with columns[index % 3]:
-            resized_image = resize_image(image["name"], target_w=1080, target_h=720)
+            image_blob = gcs.download_blob("suad_park", image["name"])
             st.image(
-                np.array(resized_image), width=200, caption=f"Image id #{image['id']}"
+                image_blob, width=200, caption=f"Image id #{image['id']}"
             )
             if not image["id"] in predictions or predictions[image["id"]] is None:
                 st.write("No predictions available")

src/cloud_storage/google_cloud_storage.py

@@ -32,3 +32,18 @@ class GoogleCloudStorage(BaseStorage):
 
         print(f"File uploaded to gs://{bucket_name}/{bucket_file_path}")
         return f"gs://{bucket_name}/{bucket_file_path}"
+
+    def download_blob(self, bucket_name: str, source_blob_name: str) -> bytes:
+        """Downloads a blob from the bucket
+
+        Args:
+            bucket_name (str): Bucket name
+            source_blob_name (str): Name of the blob to download
+
+        Returns:
+            bytes: Content of the blob
+        """
+        bucket = self.client.get_bucket(bucket_name)
+        source_blob_name = source_blob_name.replace(f"gs://{bucket_name}/", "")
+        blob = bucket.blob(source_blob_name)
+        return blob.download_as_bytes()