main.py

@@ -1213,11 +1213,14 @@ while run:
         selected_album = render_albums()
 
     if current_mode == "select":
+        try:
             if image_display is None or current_album != selected_album:
                 selected_image = ImageRender(selected_album)
                 image_display = True
 
             selected_image.render(screen)
+        except (AttributeError, NameError):
+            pass
     
     event_list = pygame.event.get()
     
@@ -1231,6 +1234,7 @@ while run:
             if not non_movable_area.collidepoint(pygame.mouse.get_pos()):
                 can_move = True
             if current_mode == "select":
+                try:
                     clicked_image_rect = ImageRender.image_save(selected_album, current_img)
                     clicked_image = clicked_image_rect.collidepoint(event.pos)
                     if clicked_image:
@@ -1260,6 +1264,8 @@ while run:
                     if to_right_button.is_hovered(event.pos):
                         selected_image.update_index(1)
                         last_image_switch_time = current_time
+                except NameError:
+                    quick_game()
 
             elif current_mode == "game":
                 if non_movable_area.collidepoint(event.pos) and not timer_running: