local Players = game:GetService("Players")
local CoreGui = game:GetService("CoreGui")
local AvatarEditor = game:GetService("AvatarEditorService")
local TweenService = game:GetService("TweenService")
local UIS = game:GetService("UserInputService")

if _G.AvatarRigPromptUI then return end
_G.AvatarRigPromptUI = true

local RigUI = Instance.new("ScreenGui")
RigUI.Name = "AvatarRigPromptUI"
RigUI.ResetOnSpawn = false

local MainFrame = Instance.new("Frame")
MainFrame.Size = UDim2.new(0, 200, 0, 100)
MainFrame.Position = UDim2.new(0.5, -100, 0, 20)
MainFrame.BackgroundColor3 = Color3.fromRGB(25, 25, 35)

local UIStroke = Instance.new("UIStroke")
UIStroke.Thickness = 2
UIStroke.Color = Color3.fromRGB(100, 200, 255)
UIStroke.Parent = MainFrame

local dragActive, dragStart, frameStart
MainFrame.InputBegan:Connect(function(input)
    if input.UserInputType == Enum.UserInputType.MouseButton1 then
        dragActive = true
        dragStart = input.Position
        frameStart = MainFrame.Position
    end
end)

UIS.InputChanged:Connect(function(input)
    if dragActive and input.UserInputType == Enum.UserInputType.MouseMovement then
        local delta = input.Position - dragStart
        MainFrame.Position = UDim2.new(
            frameStart.X.Scale,
            frameStart.X.Offset + delta.X,
            frameStart.Y.Scale,
            frameStart.Y.Offset + delta.Y
        )
    end
end)

UIS.InputEnded:Connect(function(input)
    if input.UserInputType == Enum.UserInputType.MouseButton1 then
        dragActive = false
    end
end)

local function ExecuteRigChange(rigType)
    local char = Players.LocalPlayer.Character
    if not char then return end
    
    local humanoid = char:FindFirstChildOfClass("Humanoid")
    if not humanoid then return end

    local desc = humanoid.HumanoidDescription:Clone()
    
    AvatarEditor:PromptSaveAvatar(desc, Enum.HumanoidRigType[rigType])
    local result = AvatarEditor.PromptSaveAvatarCompleted:Wait()
    
    if result == Enum.AvatarPromptResult.Success then
        humanoid.Health = 0
        humanoid:ChangeState(Enum.HumanoidStateType.Dead)
        
        Players.LocalPlayer.CharacterAdded:Wait()
        task.wait(1)
        
        local newHumanoid = Players.LocalPlayer.Character:FindFirstChildOfClass("Humanoid")
        if newHumanoid then
            newHumanoid:ApplyDescription(desc)
            newHumanoid:BuildRigFromDescription(desc, Enum.HumanoidRigType[rigType])
        end
    end
end

local R6Button = Instance.new("TextButton")
R6Button.Text = "R6 MODE"
R6Button.Size = UDim2.new(0.45, 0, 0.4, 0)
R6Button.Position = UDim2.new(0.025, 0, 0.05, 0)
R6Button.Font = Enum.Font.GothamBold
R6Button.TextColor3 = Color3.new(1,1,1)
R6Button.BackgroundColor3 = Color3.fromRGB(50, 150, 200)

R6Button.MouseButton1Click:Connect(function()
    TweenService:Create(R6Button, TweenInfo.new(0.2), {BackgroundColor3 = Color3.fromRGB(20, 100, 150)}):Play()
    ExecuteRigChange("R6")
    task.wait(0.2)
    TweenService:Create(R6Button, TweenInfo.new(0.2), {BackgroundColor3 = Color3.fromRGB(50, 150, 200)}):Play()
end)

local R15Button = R6Button:Clone()
R15Button.Text = "R15 MODE"
R15Button.Position = UDim2.new(0.525, 0, 0.05, 0)
R15Button.BackgroundColor3 = Color3.fromRGB(200, 100, 50)

R15Button.MouseButton1Click:Connect(function()
    TweenService:Create(R15Button, TweenInfo.new(0.2), {BackgroundColor3 = Color3.fromRGB(150, 50, 20)}):Play()
    ExecuteRigChange("R15")
    task.wait(0.2)
    TweenService:Create(R15Button, TweenInfo.new(0.2), {BackgroundColor3 = Color3.fromRGB(200, 100, 50)}):Play()
end)

R6Button.Parent = MainFrame
R15Button.Parent = MainFrame
MainFrame.Parent = RigUI
RigUI.Parent = CoreGui