async function generateImage() {
    const prompt = document.getElementById('prompt').value;
    const response = await fetch('http://your-server-ip:5000/generate', {
        method: 'POST',
        headers: {
            'Content-Type': 'application/json',
        },
        body: JSON.stringify({ prompt: prompt }),
    });
    const data = await response.json();
    const img = document.getElementById('generated-image');
    img.src = `data:image/png;base64,${data.image}`;
}