script.js

// ✅ Import dependencies
import { patientSignUp, loginAndRedirect } from './firebase-auth.js';
import { db, auth } from './firebase-config.js';
import { onAuthStateChanged } from "https://www.gstatic.com/firebasejs/10.7.0/firebase-auth.js";
import {
  doc, getDoc, addDoc, collection,
  getDocs, updateDoc
} from "https://www.gstatic.com/firebasejs/10.7.0/firebase-firestore.js";

let userId = null;

onAuthStateChanged(auth, async (user) => {
  if (user) {
    userId = user.uid;
    try {
      await loadPatientInfo();
      await loadMedicationLogs();
      await loadPrescriptionHistory();
    } catch (error) {
      console.error("Error loading data:", error);
    }
  } else {
    window.location.href = "page1.html";
  }
});

async function loadPatientInfo() {
  try {
    const docSnap = await getDoc(doc(db, "patients", userId));
    if (docSnap.exists()) {
      const data = docSnap.data();
      document.getElementById("patient-name").textContent = data.name;
      document.getElementById("patient-age").textContent = data.age;
      document.getElementById("patient-gender").textContent = data.gender;
      document.getElementById("patient-doctor").textContent = data.doctor;
      document.getElementById("patient-course").textContent = data.course;
    } else {
      console.log("No patient data found.");
    }
  } catch (error) {
    console.error("Error loading patient info:", error);
  }
}

window.addMedicine = async function () {
  try {
    const name = document.getElementById("medicine-name").value;
    const time = document.getElementById("medicine-time").value;
    const qty = document.getElementById("medicine-qty").value;
    const dayChecks = document.querySelectorAll('#day-checks input[type="checkbox"]:checked');
    const days = Array.from(dayChecks).map(cb => cb.value);

    if (!name || !time || !qty || days.length === 0) {
      alert("Please fill in all fields and select at least one day.");
      return;
    }

    const logData = {
      medicine: name,
      time: time,
      quantity: parseInt(qty),
      date: new Date().toISOString().split('T')[0],
      day: new Date().toLocaleString('en-us', { weekday: 'long' }),
      days: days,
      status: "not taken"
    };

    await addDoc(collection(db, "patients", userId, "logs"), logData);
    alert("Medicine added to log.");
    await loadMedicationLogs();
  } catch (error) {
    console.error("Error adding medicine:", error);
  }
};

async function loadMedicationLogs() {
  try {
    const logsRef = collection(db, "patients", userId, "logs");
    const snapshot = await getDocs(logsRef);
    const logs = snapshot.docs.map(doc => ({ id: doc.id, ...doc.data() }));

    const container = document.getElementById("medication-logs");
    if (!container) return;

    container.innerHTML = "<h2>Medication Logs</h2>";

    const now = new Date();
    const todayUTC = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate()));

    const pastLogs = [], todayLogs = [], upcomingLogs = [];

    logs.forEach(log => {
      const logDateUTC = new Date(`${log.date}T00:00:00Z`);
      if (logDateUTC < todayUTC) pastLogs.push(log);
      else if (logDateUTC.getTime() === todayUTC.getTime()) todayLogs.push(log);
      else upcomingLogs.push(log);
    });

    const sortedPast = [...pastLogs].sort((a, b) => a.date.localeCompare(b.date));
    let streak = 0;
    for (let log of sortedPast) {
      if (log.status === "taken") streak++;
      else streak = 0;
    }

    container.innerHTML += `<p><strong>Current Streak:</strong> ${streak} ${streak === 1 ? 'day' : 'days'}</p>`;

    const renderGroup = (title, logsArray, disable = false) => {
      if (logsArray.length === 0) return;
      const section = document.createElement("div");
      section.innerHTML = `<h3 style="margin-top:1rem;">${title}</h3>`;

      const grouped = {};
      logsArray.forEach(log => {
        if (!grouped[log.date]) grouped[log.date] = [];
        grouped[log.date].push(log);
      });

      Object.keys(grouped).sort().forEach(date => {
        const dateGroup = document.createElement("div");
        dateGroup.innerHTML = `<h4>${date}</h4>`;

        grouped[date].forEach(log => {
          const isChecked = log.status === "taken" ? "checked" : "";
          const isDisabled = disable ? "disabled" : "";
          const logDiv = document.createElement("div");
          logDiv.innerHTML = `
            <p>
              <strong>${log.medicine}</strong> — ${log.time} (${log.day})<br>
              <label>
                <input type="checkbox" ${isChecked} ${isDisabled}
                  onchange="toggleTaken('${log.id}', this.checked)">
                Taken
              </label><br>
              <button onclick="compareWithGemini('${log.medicine}')">💡 Compare Generic</button>
            </p>
          `;
          dateGroup.appendChild(logDiv);
        });

        section.appendChild(dateGroup);
      });

      container.appendChild(section);
    };

    renderGroup("✅ Past Logs", pastLogs);
    renderGroup("🟢 Today's Logs", todayLogs);
    renderGroup("⏳ Upcoming Logs", upcomingLogs, true);

  } catch (error) {
    console.error("Error loading medication logs:", error);
  }
}

async function loadPrescriptionHistory() {
  try {
    const historyRef = collection(db, "patients", userId, "prescriptions");
    const snapshot = await getDocs(historyRef);

    const container = document.getElementById("prescription-history");
    if (!container) return;

    container.innerHTML = snapshot.empty ? "<p>No prescriptions uploaded yet.</p>" : "";

    snapshot.forEach(docSnap => {
      const data = docSnap.data();
      const div = document.createElement("div");
      div.style.border = "1px solid #ccc";
      div.style.margin = "10px 0";
      div.style.padding = "10px";
      div.style.borderRadius = "8px";
      div.innerHTML = `
        <img src="${data.imageURL}" alt="Prescription" style="max-width: 100%; height: auto; border: 1px solid #999;" />
        <p><strong>Extracted Text:</strong><br>${data.text.replace(/\n/g, '<br>')}</p>
        <p><strong>Duration:</strong> ${data.duration} days</p>
        <p><strong>Uploaded At:</strong> ${new Date(data.timestamp).toLocaleString()}</p>
      `;
      container.appendChild(div);
    });
  } catch (error) {
    console.error("Error loading prescription history:", error);
  }
}

window.toggleTaken = async function (logId, isChecked) {
  try {
    const logRef = doc(db, "patients", userId, "logs", logId);
    await updateDoc(logRef, { status: isChecked ? "taken" : "not taken" });
  } catch (error) {
    console.error("Error updating log status:", error);
  }
};

// 🌙 Dark Mode
function toggleDarkMode() {
  document.body.classList.toggle('dark-mode');
}
window.toggleDarkMode = toggleDarkMode;

// 👤 Signup/Login
const loginBtn = document.getElementById('login-btn');
if (loginBtn) {
  loginBtn.addEventListener('click', async () => {
    const email = document.querySelector('input[type="email"]').value;
    const password = document.querySelector('input[type="password"]').value;
    await loginAndRedirect(email, password);
  });
}

const patientForm = document.getElementById('patientSignupForm');
if (patientForm) {
  patientForm.addEventListener('submit', async (e) => {
    e.preventDefault();
    const formData = {
      name: document.getElementById('name').value,
      age: document.getElementById('age').value,
      gender: document.getElementById('gender').value,
      email: document.getElementById('email').value,
      password: document.getElementById('password').value,
      doctor: document.getElementById('doctor-name').value,
      course: document.getElementById('course-duration').value,
    };
    await patientSignUp(formData);
    document.getElementById('successMsg').style.display = 'block';
  });
}

// 🤖 Gemini AI Integration
const GEMINI_API_KEY = "AIzaSyBl_Hb-Z0gofykKlkT31v_qhbn-B-1zU9c";


async function getGenericAlternative(medicineName) {
  const url = `https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key=${GEMINI_API_KEY}`;
  const prompt = `Suggest a cheaper or generic alternative for the medicine "${medicineName}". Include the composition and cost benefit. Respond concisely.`;

  const body = {
    contents: [{ parts: [{ text: prompt }] }]
  };

  const res = await fetch(url, {
    method: "POST",
    headers: { "Content-Type": "application/json" },
    body: JSON.stringify(body)
  });

  const data = await res.json();
  console.log("📦 Gemini API response:", data); // 👈 add this

  try {
    const reply = data.candidates?.[0]?.content?.parts?.[0]?.text;
    return reply || "No suggestion from Gemini.";
  } catch (e) {
    return "❌ Failed to extract suggestion.";
  }
}