document.addEventListener('DOMContentLoaded', () => {
    const GotitB = document.querySelector(".explainChoix button");
    const explain = document.querySelector(".explainChoix");
    const SummarizeInput = document.querySelector(".SummarizeInput");
    const CaptionInput = document.querySelector(".CaptionInput");
    const fileUpload = document.getElementById('file-upload');
    const imageUpload = document.getElementById('image-upload');
    const fileBtn = document.getElementById('file-btn');
    const imageBtn = document.getElementById('image-btn');
    const convo = document.querySelector('.convo');

    GotitB.addEventListener("click", () => {
        explain.style.opacity = "0";
    });

    document.querySelectorAll('.select-options input[type="radio"]').forEach(radio => {
        radio.addEventListener('change', (e) => {
            if (e.target.checked) {
                const selectedValue = e.target.value;
                if (selectedValue === "Summarize") {
                    SummarizeInput.classList.add("active");
                    SummarizeInput.classList.remove("innactive");
                    CaptionInput.classList.remove("active");
                    CaptionInput.classList.add("innactive");
                } else {
                    SummarizeInput.classList.add("innactive");
                    SummarizeInput.classList.remove("active");
                    CaptionInput.classList.remove("innactive");
                    CaptionInput.classList.add("active");
                }
            }
        });
    });

    fileBtn.addEventListener('click', () => fileUpload.click());
    imageBtn.addEventListener('click', () => imageUpload.click());

    document.querySelectorAll('.sendingQA').forEach(button => {
        button.addEventListener('click', async () => {
            const isSummarizeMode = document.querySelector('input[name="option"]:checked').value === 'Summarize';
            if (isSummarizeMode) {
                await handleSummarize();
            } else {
                await handleCaption();
            }
        });
    });

    convo.addEventListener('dragover', (e) => {
        e.preventDefault();
        convo.classList.add('drag-over');
    });

    convo.addEventListener('dragleave', () => {
        convo.classList.remove('drag-over');
    });

    convo.addEventListener('drop', async (e) => {
        e.preventDefault();
        convo.classList.remove('drag-over');
        
        if (e.dataTransfer.files.length) {
            const file = e.dataTransfer.files[0];
            const isSummarizeMode = document.querySelector('input[name="option"]:checked').value === 'Summarize';
            
            if (isSummarizeMode) {
                fileUpload.files = e.dataTransfer.files;
                await handleSummarize();
            } else {
                imageUpload.files = e.dataTransfer.files;
                await handleCaption();
            }
        }
    });

    async function handleSummarize() {
        const file = fileUpload.files[0];
        if (!file) {
            displayError('Please upload a document first');
            return;
        }
        const length = document.querySelector('input[name="optionS"]:checked')?.value || "medium";

        try {
            convo.innerHTML = '';
            displayFileInfo(file.name, 'document');
            displayThinkingMessage();

            const result = await summarizeDocument(file, length);

            displaySummaryResult(file.name, result.summary, result.audioUrl, result.pdfUrl);
        } catch (error) {
            displayError(error.message || 'Failed to summarize document');
        }
    }

    async function handleCaption() {
        const file = imageUpload.files[0];
        if (!file) {
            displayError('Please upload an image first');
            return;
        }
        try {
            convo.innerHTML = '';
            displayFileInfo(file.name, 'image');
            displayThinkingMessage();

            const caption = await captionImage(file);

            displayCaptionResult(file.name, caption);
        } catch (error) {
            displayError(error.message || 'Failed to generate caption');
        }
    }

    async function summarizeDocument(file, length) {
        const formData = new FormData();
        formData.append('file', file);
        formData.append('length', length);

        const response = await fetch('/summarize/', {
            method: 'POST',
            body: formData
        });

        if (!response.ok) {
            const error = await response.json();
            throw new Error(error.detail || 'Summarization failed');
        }

        return await response.json();
    }

    async function captionImage(file) {
        const formData = new FormData();
        formData.append('file', file);

        const response = await fetch('/imagecaption/', {
            method: 'POST',
            body: formData
        });

        if (!response.ok) {
            const error = await response.json();
            throw new Error(error.detail || 'Captioning failed');
        }

        const result = await response.json();
        return result.caption;
    }

    function displayFileInfo(filename, type) {
        const bubble = document.createElement('div');
        bubble.className = 'bubble right';
        bubble.innerHTML = `
            <div class="label">You</div>
            <div class="text">${type === 'document' ? '📄' : '🖼️'} ${filename}</div>
        `;
        convo.appendChild(bubble);
    }

    function displayThinkingMessage() {
        const bubble = document.createElement('div');
        bubble.className = 'bubble left';
        bubble.innerHTML = `
            <div class="label">Aidan</div>
            <div class="text">
                <div style="display:flex;align-items:center;gap:8px">
                    <span>Processing...</span>
                    <div class="loader"></div>
                </div>
            </div>
        `;
        convo.appendChild(bubble);
        convo.scrollTop = convo.scrollHeight;
    }

    function displaySummaryResult(filename, summary, audioUrl, pdfUrl) {
        convo.removeChild(convo.lastChild);

        const bubble = document.createElement('div');
        bubble.className = 'bubble left';
        bubble.innerHTML = `
            <div class="label">Aidan</div>
            <div class="text">
                <strong>Summary:</strong><br><br>
                ${summary.replace(/\n/g, '<br>')}
                ${audioUrl ? `<br><br><audio controls src="${audioUrl}" style="width: 100%;"></audio>` : ''}
                ${pdfUrl ? `<br><a href="${pdfUrl}" download class="download-link">📥 Download PDF Summary</a>` : ''}
            </div>
        `;
        convo.appendChild(bubble);
        convo.scrollTop = convo.scrollHeight;
    }

    function displayCaptionResult(filename, caption) {
        convo.removeChild(convo.lastChild);

        const bubble = document.createElement('div');
        bubble.className = 'bubble left';
        bubble.innerHTML = `
            <div class="label">Aidan</div>
            <div class="text">
                <strong>Caption:</strong><br><br>
                ${caption}
            </div>
        `;
        convo.appendChild(bubble);
        convo.scrollTop = convo.scrollHeight;
    }

    function displayError(message) {
        const bubble = document.createElement('div');
        bubble.className = 'bubble left';
        bubble.innerHTML = `
            <div class="label">Aidan</div>
            <div class="text" style="color: #ef4444;">
                ⚠️ ${message}
            </div>
        `;
        convo.appendChild(bubble);
        convo.scrollTop = convo.scrollHeight;
    }

    const style = document.createElement('style');
    style.textContent = `
        .loader {
            border: 2px solid #f3f3f3;
            border-top: 2px solid #3b82f6;
            border-radius: 50%;
            width: 16px;
            height: 16px;
            animation: spin 1s linear infinite;
        }
        @keyframes spin {
            0% { transform: rotate(0deg); }
            100% { transform: rotate(360deg); }
        }
        .download-link {
            display: inline-block;
            margin-top: 10px;
            padding: 5px 10px;
            background: #3b82f6;
            color: white;
            border-radius: 5px;
            text-decoration: none;
        }
        .download-link:hover {
            background: #2563eb;
        }
    `;
    document.head.appendChild(style);
});