document.addEventListener('DOMContentLoaded', () => {
    const convo = document.querySelector(".convo");
    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 sendButtons = document.querySelectorAll('.sendingQA');

    let selectedFile = null;
    let filePreviewBubble = null;

    // File upload handlers
    fileBtn.addEventListener('click', () => fileUpload.click());
    imageBtn.addEventListener('click', () => imageUpload.click());

    fileUpload.addEventListener('change', (e) => {
        if (e.target.files.length) {
            selectedFile = e.target.files[0];
            displayFilePreview(selectedFile);
        }
    });

    imageUpload.addEventListener('change', (e) => {
        if (e.target.files.length) {
            selectedFile = e.target.files[0];
            displayFilePreview(selectedFile);
        }
    });

    function displayFilePreview(file) {
        if (filePreviewBubble) filePreviewBubble.remove();
        
        filePreviewBubble = document.createElement('div');
        filePreviewBubble.className = 'file-preview-bubble bubble right';
        filePreviewBubble.innerHTML = `
            <div class="label">You</div>
            <div class="text">📎 Selected: ${file.name}</div>
        `;
        convo.appendChild(filePreviewBubble);
        convo.scrollTop = convo.scrollHeight;
    }

    function createMessageBubble(text, sender = "You", audioSrc = null) {
        const bubble = document.createElement('div');
        bubble.className = `bubble ${sender === "You" ? "right" : "left"}`;

        const label = document.createElement('div');
        label.className = "label";
        label.innerText = sender;

        const message = document.createElement('div');
        message.className = "text";
        
        if (sender === "You") {
            message.innerText = text;
        } else {
            message.innerHTML = text;
            
            if (audioSrc) {
                const audioContainer = document.createElement('div');
                audioContainer.style.display = 'flex';
                audioContainer.style.alignItems = 'center';
                audioContainer.style.gap = '10px';
                audioContainer.style.marginTop = '8px';
                
                const audio = new Audio(audioSrc);
                const audioIcon = document.createElement('i');
                audioIcon.className = 'fa-solid fa-volume-high audio-toggle';
                audioIcon.style.cursor = 'pointer';
                
                audioIcon.addEventListener('click', () => {
                    if (audio.paused) {
                        audio.play();
                        audioIcon.classList.remove('fa-volume-xmark');
                        audioIcon.classList.add('fa-volume-high');
                    } else {
                        audio.pause();
                        audioIcon.classList.remove('fa-volume-high');
                        audioIcon.classList.add('fa-volume-xmark');
                    }
                });
                
                audioContainer.appendChild(audioIcon);
                audioContainer.appendChild(document.createTextNode('Listen'));
                message.appendChild(audioContainer);
            }
        }

        bubble.appendChild(label);
        bubble.appendChild(message);
        convo.appendChild(bubble);
        convo.scrollTop = convo.scrollHeight;
        return bubble;
    }

    async function handleSend() {
        if (!selectedFile) {
            alert("Please upload a file first");
            return;
        }

        const isSummarizeMode = document.querySelector('input[name="option"]:checked').value === 'Summarize';
        
        // User message
        createMessageBubble(
            isSummarizeMode ? "Please summarize this document" : "Please caption this image",
            "You"
        );

        // Thinking message
        const thinkingBubble = createMessageBubble(
            isSummarizeMode ? "Summarizing your document..." : "Generating caption for your image...",
            "Aidan"
        );

        try {
            const formData = new FormData();
            formData.append('file', selectedFile);
            
            if (isSummarizeMode) {
                const length = document.querySelector('input[name="optionS"]:checked').value;
                formData.append('length', length);
            }

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

            const result = await response.json();
            
            // Replace thinking message with actual result
            thinkingBubble.remove();
            
            if (isSummarizeMode) {
                createMessageBubble(
                    `<strong>Summary:</strong><br><br>${result.summary.replace(/\n/g, '<br>')}` +
                    (result.pdf_url ? `<br><br><a href="${result.pdf_url}" download target="_blank">Download PDF</a>` : ''),
                    "Aidan",
                    result.audio_url
                );
            } else {
                createMessageBubble(
                    `<strong>Caption:</strong><br><br>${result.answer}`,
                    "Aidan",
                    result.audio
                );
            }
        } catch (error) {
            thinkingBubble.remove();
            createMessageBubble(
                "⚠️ Sorry, I encountered an error processing your request",
                "Aidan"
            );
            console.error(error);
        } finally {
            selectedFile = null;
            fileUpload.value = '';
            imageUpload.value = '';
            if (filePreviewBubble) {
                filePreviewBubble.remove();
                filePreviewBubble = null;
            }
        }
    }

    // Attach send handlers
    sendButtons.forEach(button => {
        button.addEventListener('click', handleSend);
    });
});