Karl/comfy_fm24_newgens
1
0
Fork 0
mirror of https://github.com/karl0ss/comfy_fm24_newgens.git synced 2025-04-29 04:13:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

Update async_image_generator.py

Browse Source
This commit is contained in:
Emil Mirzayev 2024-10-12 20:30:26 +01:00 committed by GitHub
parent cf9285a709
commit a8ed7c5821
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
async_image_generator.py
View File

@ -82,7 +82,7 @@ def save_image(image_bytes, folder, filename, resize=False):
def get_next_image_number(folder): def get_next_image_number(folder):
os.makedirs(folder, exist_ok=True) os.makedirs(folder, exist_ok=True)
existing_files = [f for f in os.listdir(folder) if f.endswith('.jpeg')] existing_files = [f for f in os.listdir(folder) if f.endswith('.png')]
return len(existing_files) + 1 return len(existing_files) + 1
@ -138,7 +138,7 @@ async def generate_images_for_country_group(session, country_group, config, n_pe
for i, image_bytes in enumerate(image_bytes_list): for i, image_bytes in enumerate(image_bytes_list):
if image_bytes: if image_bytes:
next_number = get_next_image_number(folder_name) next_number = get_next_image_number(folder_name)
file_name = f"{country_group}{next_number}.jpeg" file_name = f"{country_group}{next_number}.png"
save_image(image_bytes, folder_name, file_name, resize) save_image(image_bytes, folder_name, file_name, resize)