GenerationService

local ReplicatedStorage = game:GetService("ReplicatedStorage")
local GenerationService = game:GetService("GenerationService")

-- Create a RemoteEvent for client-server communication
local meshGenerationEvent = Instance.new("RemoteEvent")
meshGenerationEvent.Name = "MeshGenerationEvent"
meshGenerationEvent.Parent = ReplicatedStorage

-- Event to send generation results back to clients
local generationResultEvent = Instance.new("RemoteEvent")
generationResultEvent.Name = "GenerationResultEvent"
generationResultEvent.Parent = ReplicatedStorage

-- Handle mesh generation requests from clients
meshGenerationEvent.OnServerEvent:Connect(function(player, prompt)
    print("Generating mesh for player", player.Name, "with prompt:", prompt)

    -- Generate the mesh on the server
    local success, generationId, contextId = pcall(function()
        return GenerationService:GenerateMeshAsync(
            {["Prompt"] = prompt},
            player,
            {},
            function(intermediateType, intermediateGenerationId, intermediateContextId)
                -- Send intermediate result to the requesting client
                generationResultEvent:FireClient(
                    player,
                    intermediateGenerationId,
                    true -- isIntermediate
                )
            end
        )
    end)

    if success then
        -- Send the final generation ID to the client
        generationResultEvent:FireClient(player, generationId, false)
        print("Successfully generated mesh with ID:", generationId)
    else
        -- Notify the client of failure
        generationResultEvent:FireClient(player, nil, false)
        warn("Failed to generate mesh:", generationId)
    end
end)

local ReplicatedStorage = game:GetService("ReplicatedStorage")
local GenerationService = game:GetService("GenerationService")
local Players = game:GetService("Players")

-- Connect to the RemoteEvents
local meshGenerationEvent = ReplicatedStorage:WaitForChild("MeshGenerationEvent")
local generationResultEvent = ReplicatedStorage:WaitForChild("GenerationResultEvent")

-- Local player
local player = Players.LocalPlayer

-- Store generated meshes
local generatedMeshes = {}

-- Function to request mesh generation
local function requestMeshGeneration(prompt)
    print("Requesting mesh generation for prompt: " .. prompt)
    meshGenerationEvent:FireServer(prompt)
end

-- Handle generation results from the server
generationResultEvent.OnClientEvent:Connect(function(generationId, isIntermediate)
    if not generationId then
        print("Generation failed")
        return
    end

    print(isIntermediate and "Loading intermediate result..." or "Loading final result...")

    local mesh = GenerationService:LoadGeneratedMeshAsync(generationId)

    if mesh then
        -- Clean up previous mesh if it exists
        if generatedMeshes[isIntermediate and "intermediate" or "final"] then
            generatedMeshes[isIntermediate and "intermediate" or "final"]:Destroy()
        end

        mesh.Parent = workspace

        -- Position the mesh in front of the player
        local character = player.Character
        if character and character:FindFirstChild("HumanoidRootPart") then
            local rootPart = character.HumanoidRootPart
            mesh:PivotTo(rootPart.CFrame * CFrame.new(0, 0, -10))
        end

        -- Store the mesh for cleanup
        generatedMeshes[isIntermediate and "intermediate" or "final"] = mesh
        print(isIntermediate and "Loaded intermediate result" or "Loaded final result")
    else
        print("Failed to load mesh")
    end
end)

-- Example usage
requestMeshGeneration("toy robot")