From a8ed7c582157baf7a4a544dad7b15930fba20665 Mon Sep 17 00:00:00 2001
From: Emil Mirzayev <25964049+emilmirzayev@users.noreply.github.com>
Date: Sat, 12 Oct 2024 20:30:26 +0100
Subject: [PATCH] Update async_image_generator.py

---
 async_image_generator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/async_image_generator.py b/async_image_generator.py
index c60e635..93867e4 100644
--- a/async_image_generator.py
+++ b/async_image_generator.py
@@ -82,7 +82,7 @@ def save_image(image_bytes, folder, filename, resize=False):
 
 def get_next_image_number(folder):
     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
 
 
@@ -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):
             if image_bytes:
                 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)