let uploadedImageBase64 = "";

function handleImageUpload(event) {
  const file = event.target.files[0];
  const reader = new FileReader();
  reader.onloadend = () => {
    uploadedImageBase64 = reader.result.split(",")[1];
    document.getElementById("ocrPreview").innerText = "✅ Image loaded.";
  };
  if (file) reader.readAsDataURL(file);
}

function predictText() {
  const text = document.getElementById("newsText").value;
  fetch("http://localhost:5000/predict", {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify({ text })
  })
    .then(res => res.json())
    .then(data => {
      document.getElementById("result").innerHTML =
        `🧾 ${data.label} (${data.confidence}% confidence)`;
    });
}

function predictImage() {
  if (!uploadedImageBase64) {
    alert("Please upload an image first.");
    return;
  }
  fetch("http://localhost:5000/predict-image", {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify({ image: uploadedImageBase64 })
  })
    .then(res => res.json())
    .then(data => {
      if (data.error) {
        document.getElementById("result").innerHTML = "❌ " + data.error;
      } else {
        document.getElementById("result").innerHTML =
          `📜 Extracted Text: ${data.extracted_text}<br>🧾 ${data.label} (${data.confidence}% confidence)`;
      }
    });
}