local Players = game:GetService("Players")
local RunService = game:GetService("RunService")
local LocalPlayer = Players.LocalPlayer
local LocalUSTNotes = workspace:WaitForChild("LocalUSTNotes")
local SpawnArea = workspace:WaitForChild("SpawnUSTArea")
local Hitbox = workspace:WaitForChild("UST_Hitbox")

-- Save original values before modifying
local originalCanCollide = SpawnArea.CanCollide
local originalSize = SpawnArea.Size
local originalTransparency = SpawnArea.Transparency
local originalCFrame = SpawnArea.CFrame
local originalSpawnCanQuery = SpawnArea.CanQuery

local originalHitboxSize = Hitbox.Size
local originalHitboxTransparency = Hitbox.Transparency
local originalHitboxCanQuery = Hitbox.CanQuery
local originalHitboxCanCollide = Hitbox.CanCollide

-- Configure the spawn part
SpawnArea.CanCollide = false
SpawnArea.CanQuery = false
SpawnArea.Size = Vector3.new(4, 4, 4)
SpawnArea.Transparency = 0.9

-- Configure the hitbox
Hitbox.Size = Vector3.new(10, 10, 10)
Hitbox.Transparency = 1
Hitbox.CanCollide = false
Hitbox.CanQuery = false

-- Make the spawn area follow the player (twice per second)
local character = LocalPlayer.Character or LocalPlayer.CharacterAdded:Wait()
local rootPart = character:WaitForChild("HumanoidRootPart")

local followActive = true
task.spawn(function()
    while followActive do
        SpawnArea.CFrame = rootPart.CFrame
        task.wait(LocalPlayer:GetNetworkPing()*1.14)
    end
end)

-- Note vacuum logic
local noteConnection = LocalUSTNotes.ChildAdded:Connect(function(child)
    character = LocalPlayer.Character or LocalPlayer.CharacterAdded:Wait()
    rootPart = character:WaitForChild("HumanoidRootPart")
    task.wait()
    if child:IsA("BasePart") then
        child.CFrame = rootPart.CFrame + Vector3.new(0, 3, 0)
    elseif child:IsA("Model") then
        if child.PrimaryPart then
            child:SetPrimaryPartCFrame(rootPart.CFrame + Vector3.new(0, 3, 0))
        else
            child:PivotTo(rootPart.CFrame + Vector3.new(0, 3, 0))
        end
    end
end)

-- UI
local screenGui = Instance.new("ScreenGui")
screenGui.Name = "USTCollectorUI"
screenGui.ResetOnSpawn = false
screenGui.Parent = LocalPlayer.PlayerGui

local frame = Instance.new("Frame")
frame.Size = UDim2.new(0, 200, 0, 70)
frame.Position = UDim2.new(0.5, -100, 0.05, 0)
frame.BackgroundColor3 = Color3.fromRGB(30, 30, 30)
frame.BorderSizePixel = 0
frame.Active = true
frame.Draggable = true
frame.Parent = screenGui

local corner = Instance.new("UICorner")
corner.CornerRadius = UDim.new(0, 8)
corner.Parent = frame

local title = Instance.new("TextLabel")
title.Size = UDim2.new(1, 0, 0.4, 0)
title.Position = UDim2.new(0, 0, 0, 0)
title.BackgroundTransparency = 1
title.Text = "🎵 UST Collector"
title.TextColor3 = Color3.fromRGB(255, 255, 255)
title.TextScaled = true
title.Font = Enum.Font.GothamBold
title.Parent = frame

local stopButton = Instance.new("TextButton")
stopButton.Size = UDim2.new(0.85, 0, 0.45, 0)
stopButton.Position = UDim2.new(0.075, 0, 0.5, 0)
stopButton.BackgroundColor3 = Color3.fromRGB(200, 50, 50)
stopButton.Text = "Stop & Revert"
stopButton.TextColor3 = Color3.fromRGB(255, 255, 255)
stopButton.TextScaled = true
stopButton.Font = Enum.Font.GothamBold
stopButton.BorderSizePixel = 0
stopButton.Parent = frame

local btnCorner = Instance.new("UICorner")
btnCorner.CornerRadius = UDim.new(0, 6)
btnCorner.Parent = stopButton

-- Revert everything on click
stopButton.MouseButton1Click:Connect(function()
    followActive = false
    noteConnection:Disconnect()

    SpawnArea.CanCollide = originalCanCollide
    SpawnArea.CanQuery = originalSpawnCanQuery
    SpawnArea.Size = originalSize
    SpawnArea.Transparency = originalTransparency
    SpawnArea.CFrame = originalCFrame

    Hitbox.Size = originalHitboxSize
    Hitbox.Transparency = originalHitboxTransparency
    Hitbox.CanQuery = originalHitboxCanQuery
    Hitbox.CanCollide = originalHitboxCanCollide

    screenGui:Destroy()
    print("UST Collector stopped and reverted.")
end)