main.py

import os
import shutil

directory = os.path.join(os.path.expanduser("~"), "Downloads")

extensions = {
    ".jpg" : "Photos",
    ".png" : "Photos",
    ".jpeg": "Photos",
    ".webp" : "Photos",
    ".mp4" : "Videos",
    ".mov" : "Videos",
    ".pdf" : "Documents",
    ".doc" : "Documents",
    ".xlsx" : "Documents",
    ".txt" : "Documents", 
    ".docx" : "Documents",
    ".mp3" : "Music",
    ".wav" : "Music"
}

for filename in os.listdir(directory):
    file_path = os.path.join(directory, filename)

    if os.path.isfile(file_path):
        extension = os.path.splitext(filename)[1].lower()

        if extension in extensions:
            folder_name = extensions[extension]
            
            folder_path = os.path.join(directory, folder_name)
            os.makedirs(folder_path, exist_ok=True)

            destination_path = os.path.join(folder_path, filename)
            shutil.move(file_path, destination_path)

            print(f"Moved{filename} to {folder_name} folder.")
        else:
            print(f"Skipped {filename}. Unknown file extension.")
    else:
        print(f"Skipped {filename}. It is a directory.")

print("File organization completed.")