test.py

import os
import shutil
from tkinter import Tk, filedialog

def select_folders():
    """
    Prompts you to select folders one by one.
    Returns a list of selected folder paths.
    """
    folders = []
    root = Tk()
    root.withdraw()
    root.attributes('-topmost', True)
    while True:
        print("Select a folder to add. Click Cancel (or close) if done.")
        folder = filedialog.askdirectory(title="Select a Tumor Type Folder")
        if not folder:
            break
        folders.append(folder)
        more = input("Add another folder? (y/n): ").strip().lower()
        if more != 'y':
            break
    return folders

def merge_and_rename_images(folders, dst_folder, prefix='no_tumor_'):
    """
    Merges images from folders into one, renaming them serially.
    """
    if not os.path.exists(dst_folder):
        os.makedirs(dst_folder)
    img_files = []
    for folder in folders:
        for rootdir, dirs, files in os.walk(folder):
            for file in sorted(files):
                if file.lower().endswith(('.jpg', '.jpeg', '.png', '.bmp', '.tif')):
                    img_files.append(os.path.join(rootdir, file))
    for idx, file_path in enumerate(img_files):
        ext = os.path.splitext(file_path)[1]
        new_name = f"{prefix}{idx:04d}{ext}"
        shutil.copy(file_path, os.path.join(dst_folder, new_name))
    print(f"Total {len(img_files)} images merged and renamed in '{dst_folder}'.")

if __name__ == "__main__":
    print("You'll be prompted to select folders one by one with a dialog.")
    folders = select_folders()
    if not folders:
        print("No folders were selected. Exiting.")
    else:
        print("Now select your output folder (merged images will be saved here).")
        root = Tk()
        root.withdraw()
        output = filedialog.askdirectory(title="Select or Create Output Folder")
        if not output:
            print("No output folder selected. Exiting.")
        else:
            merge_and_rename_images(folders, output)