diff --git a/main.py b/main.py index 4c9e575..b992b11 100644 --- a/main.py +++ b/main.py @@ -10,6 +10,7 @@ import scripts.screens.all_screens +os.makedirs('./saved_images', exist_ok=True) clock = pygame.time.Clock() scripts.screens.all_screens.start_screen.screen_switches() diff --git a/scripts/screens/save_image.py b/scripts/screens/save_image.py index ce561f9..caeda38 100644 --- a/scripts/screens/save_image.py +++ b/scripts/screens/save_image.py @@ -103,7 +103,6 @@ def save_image(self): file_number = "" i = 0 - os.makedirs('./saved_images', exist_ok=True) while True: if os.path.isfile(f"{'./saved_images'}/{file_name + file_number}.png"): i += 1