Spaces:
Runtime error
Runtime error
File size: 2,172 Bytes
2350624 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
from appwrite.services.storage import Storage
from appwrite.client import Client
from appwrite.query import Query
from io import BytesIO
from PIL import Image
import configparser
import os
def getImages(nImages: int) -> dict[str, list[Image.Image]]:
"""
Retrieves images from the configured Appwrite S3 bucket.
Args:
nImages (int): The maximum number of images to retrieve from the bucket.
Returns:
dict[str, list[Image.Image]]: A dictionary where each key is a category (str) and each value is a list of PIL images (list[Image.Image]) belonging to that category.
"""
# configuring the appwrite client
client = Client()
(client
.set_endpoint(os.environ["APPWRITE_ENDPOINT"])
.set_project(os.environ["APPWRITE_PROJECT_ID"])
.set_key(os.environ["APPWRITE_API_KEY"])
.set_self_signed()
.set_session("")
)
# retrieving names of all files from the storage bucket
storage = Storage(client)
allFiles = storage.list_files(bucket_id = os.environ["APPWRITE_BUCKET_ID"], queries = [Query.limit(nImages)])
allFiles = [file["$id"] for file in allFiles["files"]]
extractedData = {
"chokers": [x for x in allFiles if x.startswith("CH")],
"shortNecklaces": [x for x in allFiles if x.startswith("SN")],
"longNecklaces": [x for x in allFiles if x.startswith("LN")],
"models": [x for x in allFiles if x.startswith("MD")]
}
# getting PIL images out of the files
extractedData = {
x: [
Image.open(
BytesIO(
storage.get_file_view(
bucket_id = os.environ["APPWRITE_BUCKET_ID"],
file_id = y
)
)
) for y in extractedData[x]
] for x in extractedData
}
return extractedData
def getConfig(path: str):
"""
Load configuration from a specified file.
Args:
path (str): The path to the configuration file.
Returns:
ConfigParser: The loaded configuration object.
"""
config = configparser.ConfigParser()
config.read(path)
return config |