local Players = game:GetService("Players")
local ReplicatedStorage = game:GetService("ReplicatedStorage")
local CoreGui = game:GetService("CoreGui")
local lp = Players.LocalPlayer

local success, abilitiesModule = pcall(function()
    return require(ReplicatedStorage:WaitForChild("Modules"):WaitForChild("Abilities"))
end)

if not success or type(abilitiesModule) ~= "table" then
    warn("Failed to load Abilities module")
    return
end

local screenGui = Instance.new("ScreenGui")
screenGui.Name = "NoCooldownGUI"
screenGui.Parent = CoreGui

local button = Instance.new("TextButton")
button.Size = UDim2.new(0, 120, 0, 40)
button.Position = UDim2.new(0, 20, 0, 200)
button.BackgroundColor3 = Color3.fromRGB(40, 40, 40)
button.TextColor3 = Color3.fromRGB(255, 255, 255)
button.Font = Enum.Font.SourceSansBold
button.TextSize = 20
button.Text = "No Cooldown: OFF"
button.Parent = screenGui

local enabled = false
local originalCooldowns = {}

local function updateButton()
    if enabled then
        button.Text = "No Cooldown: ON"
        button.BackgroundColor3 = Color3.fromRGB(0, 170, 0)
    else
        button.Text = "No Cooldown: OFF"
        button.BackgroundColor3 = Color3.fromRGB(170, 0, 0)
    end
end

button.MouseButton1Click:Connect(function()
    enabled = not enabled
    updateButton()

    for abilityName, abilityData in pairs(abilitiesModule) do
        if abilityData.Cooldown then
            if enabled then
                originalCooldowns[abilityName] = originalCooldowns[abilityName] or abilityData.Cooldown
                abilityData.Cooldown = 0
            else
                abilityData.Cooldown = originalCooldowns[abilityName] or abilityData.Cooldown
            end
        end
        -- Optional: handle directional cooldowns if they exist
        for _, dir in pairs({"F_Cooldown","B_Cooldown","R_Cooldown","L_Cooldown"}) do
            if abilityData[dir] then
                if enabled then
                    originalCooldowns[abilityName.."_"..dir] = originalCooldowns[abilityName.."_"..dir] or abilityData[dir]
                    abilityData[dir] = 0
                else
                    abilityData[dir] = originalCooldowns[abilityName.."_"..dir] or abilityData[dir]
                end
            end
        end
    end

    print(enabled and "No Cooldown Enabled" or "No Cooldown Disabled")
end)

updateButton()
print("No Cooldown Script Loaded. Toggle with GUI button.")