src/services/google_cloud_image_upload.py

from google.cloud import storage
from PIL import Image
import os
import uuid
import tempfile

def get_credentials():
    credentials_json_string = os.getenv("GOOGLE_APPLICATION_CREDENTIALS_JSON")

    # create a temp file with the credentials
    with tempfile.NamedTemporaryFile(mode="w+", delete=False, suffix=".json") as temp_file:
        temp_file.write(credentials_json_string)
        temp_file_path = temp_file.name
    return temp_file_path

class GoogleCloudImageUploadService:
    BUCKET_NAME = "picchat-assets"
    MAX_DIMENSION = 1024

    def __init__(self):
        # Using API key here as per your original code. Note that for production,
        # service account credentials are generally recommended.
        # get credentials from env
        credentials_json = get_credentials()
        self.storage_client = storage.Client.from_service_account_json(credentials_json)

    def upload_image_to_gcs(self, source_file_name):
        """
        Uploads an image to the specified Google Cloud Storage bucket.
        Supports both JPEG and PNG formats.
        """
        try:
            bucket = self.storage_client.bucket(self.BUCKET_NAME)
            blob_name = str(uuid.uuid4())
            blob = bucket.blob(blob_name)

            # Open and optionally resize the image, then save to a temporary file.
            with Image.open(source_file_name) as image:
                # Determine the original format. If it's not JPEG or PNG, default to JPEG.
                original_format = image.format.upper() if image.format in ['JPEG', 'PNG'] else "JPEG"

                # Resize if needed.
                if image.width > self.MAX_DIMENSION or image.height > self.MAX_DIMENSION:
                    image.thumbnail((self.MAX_DIMENSION, self.MAX_DIMENSION))
                
                # Choose the file extension based on the image format.
                suffix = ".jpg" if original_format == "JPEG" else ".png"
                
                # Create a temporary file with the appropriate suffix.
                with tempfile.NamedTemporaryFile(delete=False, suffix=suffix) as temp_file:
                    temp_filename = temp_file.name
                    image.save(temp_filename, format=original_format)

            try:
                # Set content type based on the image format.
                content_type = "image/jpeg" if original_format == "JPEG" else "image/png"
                blob.upload_from_filename(temp_filename, content_type=content_type)
                blob.make_public()
            finally:
                # Remove the temporary file.
                os.remove(temp_filename)

            print(f"File {source_file_name} uploaded to {blob_name} in bucket {self.BUCKET_NAME}.")
            return blob.public_url
        except Exception as e:
            print(f"An error occurred: {e}")
            return None

if __name__ == "__main__":
    image = "./assets/lakeview.jpg"  # Replace with your JPEG or PNG image path.
    upload_service = GoogleCloudImageUploadService()
    url = upload_service.upload_image_to_gcs(image)
    print(url)