document.addEventListener("DOMContentLoaded", function () {
    const convo = document.querySelector(".convo");
    const sendBtn = document.querySelector(".sendingQA");
    const userInput = document.querySelector(".qt input");
    const imageIcon = document.querySelector(".fa-image");
    const fileIcon = document.querySelector(".fa-file");

    let selectedFile = null;
    let filePreviewDiv = null;
    let autoScroll = true;

    convo.addEventListener("scroll", () => {
        const threshold = 150;
        autoScroll = convo.scrollHeight - convo.scrollTop - convo.clientHeight < threshold;
    });

    function scrollToBottom() {
        if (autoScroll) {
            convo.scrollTo({ top: convo.scrollHeight, behavior: "smooth" });
        }
    }

    function createMessageBubble(text, sender, audioUrl = null, fileName = null) {
        const bubble = document.createElement("div");
        bubble.classList.add("bubble", sender === "You" ? "right" : "left");

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

        const textDiv = document.createElement("div");
        textDiv.className = "text";
        textDiv.innerText = text;

        bubble.appendChild(label);
        bubble.appendChild(textDiv);

        if (fileName) {
            const fileLine = document.createElement("div");
            fileLine.style.fontSize = "0.85rem";
            fileLine.style.color = "#555";
            fileLine.innerHTML = `📎 <strong>File:</strong> ${fileName}`;
            bubble.appendChild(fileLine);
        }

        if (audioUrl) {
            const audio = document.createElement("audio");
            audio.controls = true;
            audio.autoplay = true;
            audio.src = audioUrl;
            audio.style.marginTop = "10px";
            bubble.appendChild(audio);
        }

        convo.appendChild(bubble);
        scrollToBottom();
        return bubble;
    }

  function showFilePreview(filename, icon = "📎") {
    if (filePreviewDiv) filePreviewDiv.remove();

    filePreviewDiv = document.createElement("div");
    filePreviewDiv.className = "file-preview-bubble";
    filePreviewDiv.innerHTML = `${icon} <strong>Selected:</strong> ${filename}`;

    convo.appendChild(filePreviewDiv); 
    scrollToBottom();
}

    sendBtn.addEventListener("click", async () => {
        const question = userInput.value.trim();
        if (!question) return;

        if (!selectedFile) {
            alert("Please upload a file before asking a question.");
            return;
        }

        if (filePreviewDiv) filePreviewDiv.remove();  // Remove preview bubble

        createMessageBubble(question, "You", null, selectedFile.name);

        const thinkingBubble = createMessageBubble("Waiat let me think 🤔", "Chris");

        const formData = new FormData();
        formData.append("question", question);
        formData.append("file", selectedFile);

        try {
            const response = await fetch("/predict", {
                method: "POST",
                body: formData
            });

            const result = await response.json();

            thinkingBubble.querySelector(".text").innerText = result.answer || "No response received.";

            if (result.audio) {
                const audio = document.createElement("audio");
                audio.controls = true;
                audio.autoplay = true;
                audio.src = result.audio;
                audio.style.marginTop = "10px";
                thinkingBubble.appendChild(audio);
            }

        } catch (err) {
            thinkingBubble.querySelector(".text").innerText = "⚠️ Chris had trouble connecting.";
        }

        userInput.value = "";
        selectedFile = null;
    });

    imageIcon.addEventListener("click", () => {
        const input = document.createElement("input");
        input.type = "file";
        input.accept = "image/*";
        input.onchange = () => {
            selectedFile = input.files[0];
            if (selectedFile) showFilePreview(selectedFile.name, "🖼️");
        };
        input.click();
    });

    fileIcon.addEventListener("click", () => {
        const input = document.createElement("input");
        input.type = "file";
        input.accept = ".pdf,.docx,.pptx,.xlsx";
        input.onchange = () => {
            selectedFile = input.files[0];
            if (selectedFile) showFilePreview(selectedFile.name, "📄");
        };
        input.click();
    });
});