script.js

const firebaseConfig = {
  apiKey: "AIzaSyCxAhTuPuNelY0IiWYvguR7sFxJ2dXAphQ",
  authDomain: "future-self-mirror-server.firebaseapp.com",
  projectId: "future-self-mirror-server"
};

firebase.initializeApp(firebaseConfig);
const db = firebase.firestore();

function createParticles() {
  const container = document.getElementById('particles');
  for (let i = 0; i < 50; i++) {
    const particle = document.createElement('div');
    particle.className = 'particle';
    particle.style.left = Math.random() * 100 + '%';
    particle.style.animationDelay = Math.random() * 15 + 's';
    particle.style.animationDuration = (Math.random() * 10 + 15) + 's';
    container.appendChild(particle);
  }
}

createParticles();

function typeText(el, text, speed = 22) {
  return new Promise(resolve => {
    el.textContent = "";
    el.classList.add("typing");
    let i = 0;

    const timer = setInterval(() => {
      el.textContent += text[i++];
      if (i >= text.length) {
        clearInterval(timer);
        el.classList.remove("typing");
        resolve();
      }
    }, speed);
  });
}

function analyzeHabit(text) {
  const patterns = {
    mental: {
      keywords: ["overthink", "stress", "anxiety", "negative", "worry", "ruminate", "panic", "fear", "doubt"],
      subTypes: {
        cognitive: ["overthink", "ruminate", "analyze"],
        emotional: ["anxiety", "worry", "panic", "fear"],
        selfTalk: ["negative", "doubt", "critic"]
      }
    },
    behavioral: {
      keywords: ["phone", "scroll", "instagram", "youtube", "gaming", "netflix", "social media", "tiktok", "shopping", "gambling"],
      subTypes: {
        digital: ["phone", "scroll", "instagram", "youtube", "tiktok", "gaming"],
        consumption: ["netflix", "binge", "watch"],
        compulsive: ["shopping", "gambling", "betting"]
      }
    },
    productivity: {
      keywords: ["procrastinate", "delay", "lazy", "avoid", "distract", "waste time", "unproductive"],
      subTypes: {
        avoidance: ["procrastinate", "delay", "avoid"],
        distraction: ["distract", "wander", "unfocused"],
        inertia: ["lazy", "unmotivated", "stuck"]
      }
    },
    physical: {
      keywords: ["junk", "unhealthy", "no exercise", "sedentary", "smoking", "drinking", "eating", "sugar", "caffeine"],
      subTypes: {
        nutrition: ["junk", "sugar", "eating", "food"],
        substance: ["smoking", "drinking", "alcohol", "caffeine"],
        movement: ["sedentary", "no exercise", "inactive"]
      }
    },
    social: {
      keywords: ["isolate", "avoid people", "lonely", "withdraw", "anti-social"],
      subTypes: {
        withdrawal: ["isolate", "withdraw", "hide"],
        avoidance: ["avoid people", "anti-social"]
      }
    },
    sleep: {
      keywords: ["late", "insomnia", "sleep", "night", "tired"],
      subTypes: {
        timing: ["late", "night owl"],
        quality: ["insomnia", "restless"]
      }
    }
  };

  let primaryType = "general";
  let subType = null;
  let maxScore = 0;

  for (let type in patterns) {
    const score = patterns[type].keywords.filter(w => text.includes(w)).length;
    if (score > maxScore) {
      maxScore = score;
      primaryType = type;
      
      for (let sub in patterns[type].subTypes) {
        if (patterns[type].subTypes[sub].some(w => text.includes(w))) {
          subType = sub;
          break;
        }
      }
    }
  }

  return { primaryType, subType };
}

function analyzeGoal(text) {
  const patterns = {
    relief: {
      keywords: ["calm", "meditate", "peace", "relax", "breathe", "mindful", "zen"],
      intensity: ["deep peace", "total calm"]
    },
    discipline: {
      keywords: ["routine", "consistent", "discipline", "structure", "schedule", "habit", "regular"],
      intensity: ["strict", "rigorous", "unwavering"]
    },
    growth: {
      keywords: ["learn", "study", "skill", "career", "improve", "master", "develop", "grow"],
      intensity: ["excel", "master", "expert"]
    },
    fitness: {
      keywords: ["exercise", "gym", "workout", "run", "fit", "healthy", "strong", "athletic"],
      intensity: ["transform", "athlete", "peak"]
    },
    creative: {
      keywords: ["create", "write", "paint", "music", "art", "express"],
      intensity: ["daily practice", "prolific"]
    },
    social: {
      keywords: ["connect", "friends", "relationship", "communicate", "people"],
      intensity: ["thriving", "deep connections"]
    },
    financial: {
      keywords: ["save", "money", "budget", "invest", "financial"],
      intensity: ["wealth", "financial freedom"]
    }
  };

  let goalType = "general";
  let ambitionLevel = "moderate";
  let maxScore = 0;

  for (let type in patterns) {
    const score = patterns[type].keywords.filter(w => text.includes(w)).length;
    if (score > maxScore) {
      maxScore = score;
      goalType = type;
      
      if (patterns[type].intensity.some(w => text.includes(w))) {
        ambitionLevel = "high";
      }
    }
  }

  return { goalType, ambitionLevel };
}

function analyzeSeverity(habitText, goalText) {
  let score = 0;
  
  if (["always", "every day", "daily", "constantly"].some(w => habitText.includes(w))) score += 3;
  if (["can't stop", "addicted", "compulsive", "uncontrollable"].some(w => habitText.includes(w))) score += 4;
  if (["hours", "all day", "entire day"].some(w => habitText.includes(w))) score += 2;
  if (["multiple times", "repeatedly"].some(w => habitText.includes(w))) score += 2;
  
  if (["sometimes", "occasionally", "rarely"].some(w => habitText.includes(w))) score -= 2;
  if (["want to", "ready to"].some(w => goalText.includes(w))) score -= 1;
  if (["trying", "attempting"].some(w => goalText.includes(w))) score += 1;

  if (score >= 5) return { level: "severe", attempts: 0 };
  if (score >= 3) return { level: "moderate", attempts: habitText.includes("tried") ? 1 : 0 };
  return { level: "mild", attempts: habitText.includes("started") ? 1 : 0 };
}

function analyzeContext(habitText, goalText) {
  const context = {
    emotional: false,
    environmental: false,
    social: false,
    timeConstrained: false,
    identityDriven: false
  };

  if (["feel", "emotion", "stress", "anxiety", "sad", "angry"].some(w => habitText.includes(w))) {
    context.emotional = true;
  }

  if (["home", "work", "room", "desk", "bed"].some(w => habitText.includes(w))) {
    context.environmental = true;
  }

  if (["friends", "people", "social", "alone", "others"].some(w => habitText.includes(w) || goalText.includes(w))) {
    context.social = true;
  }

  if (["morning", "evening", "night", "weekend", "schedule"].some(w => goalText.includes(w))) {
    context.timeConstrained = true;
  }

  if (["become", "be someone", "type of person", "who I am"].some(w => goalText.includes(w))) {
    context.identityDriven = true;
  }

  return context;
}

function buildSmartPrediction({ name, habitAnalysis, goalAnalysis, severityData, context }) {
  const { primaryType: habitType, subType } = habitAnalysis;
  const { goalType, ambitionLevel } = goalAnalysis;
  const { level: severity, attempts } = severityData;

  const frictionPatterns = {
    severe: {
      mental: "intense cognitive loops and emotional resistance",
      behavioral: "powerful dopamine-driven urges and withdrawal symptoms",
      productivity: "deep-rooted avoidance patterns and fear of failure",
      physical: "strong physiological cravings and body-mind disconnection",
      social: "profound isolation patterns and social anxiety",
      sleep: "circadian disruption and hyperarousal states",
      general: "deeply entrenched automatic behaviors"
    },
    moderate: {
      mental: "recurring thought patterns and emotional discomfort",
      behavioral: "habitual impulses and mild withdrawal effects",
      productivity: "moderate resistance and distraction tendencies",
      physical: "regular cravings and comfort-seeking behaviors",
      social: "selective avoidance and social hesitation",
      sleep: "irregular sleep-wake patterns",
      general: "familiar comfort zones and routine resistance"
    },
    mild: {
      mental: "occasional mental drift and minor stress responses",
      behavioral: "situational temptations and brief impulses",
      productivity: "intermittent procrastination and low motivation",
      physical: "mild cravings triggered by specific contexts",
      social: "sporadic withdrawal and minor social discomfort",
      sleep: "occasional late nights",
      general: "minor habit inertia"
    }
  };

  const adaptationStrategies = {
    mental: {
      cognitive: "metacognitive awareness and thought defusion techniques",
      emotional: "emotional regulation and distress tolerance skills",
      selfTalk: "cognitive reframing and self-compassion practices"
    },
    behavioral: {
      digital: "environmental design, friction tools, and replacement routines",
      consumption: "scheduled exposure limits and alternative rewards",
      compulsive: "urge surfing and delay strategies"
    },
    productivity: {
      avoidance: "exposure therapy to difficult tasks and micro-commitment strategies",
      distraction: "attention training and single-tasking protocols",
      inertia: "activation energy reduction and momentum-building rituals"
    },
    physical: {
      nutrition: "meal planning, environment control, and nutritional awareness",
      substance: "gradual reduction protocols and substitute behaviors",
      movement: "identity-based exercise and incremental activity increase"
    },
    social: {
      withdrawal: "graduated exposure to social situations and connection rituals",
      avoidance: "social skills practice and relationship investment"
    },
    sleep: {
      timing: "chronotherapy and consistent wake times",
      quality: "sleep hygiene optimization and stress management"
    },
    general: "pattern interruption and environmental restructuring"
  };

  const outcomeProjections = {
    relief: {
      high: "profound inner peace and emotional mastery",
      moderate: "consistent calm and stress resilience",
      low: "baseline relaxation and reduced anxiety"
    },
    discipline: {
      high: "effortless consistency and systems-driven living",
      moderate: "reliable routines and structured days",
      low: "improved follow-through on commitments"
    },
    growth: {
      high: "accelerated skill mastery and career transformation",
      moderate: "steady competence building and expanded capabilities",
      low: "noticeable improvement in targeted areas"
    },
    fitness: {
      high: "physical transformation and athletic identity",
      moderate: "consistent fitness and improved health markers",
      low: "regular movement and energy increase"
    },
    creative: {
      high: "prolific output and artistic identity",
      moderate: "regular creative practice and skill development",
      low: "occasional creative expression"
    },
    social: {
      high: "thriving social life and deep connections",
      moderate: "quality relationships and social confidence",
      low: "reduced isolation and improved connections"
    },
    financial: {
      high: "financial freedom and wealth accumulation",
      moderate: "financial stability and growing savings",
      low: "better money management and reduced stress"
    },
    general: {
      high: "personal transformation and life optimization",
      moderate: "sustainable life improvements and balance",
      low: "positive changes and increased wellbeing"
    }
  };

  const predictions = [];
  
  let phase1 = `${name} begins confronting ${frictionPatterns[severity][habitType]}`;
  if (context.emotional) {
    phase1 += ", particularly when emotional triggers arise";
  }
  if (context.environmental) {
    phase1 += ". Environmental cues initially provoke strong habit impulses";
  }
  if (attempts > 0) {
    phase1 += `. Unlike previous attempts, awareness of failure patterns creates strategic advantage`;
  }
  phase1 += ".";
  predictions.push(phase1);

  const strategy = subType 
    ? adaptationStrategies[habitType][subType] 
    : adaptationStrategies[habitType];
  
  let phase2 = `Over weeks 3-8, ${strategy} becomes the primary change mechanism`;
  if (context.timeConstrained) {
    phase2 += ", integrated into specific daily time blocks";
  }
  if (context.social) {
    phase2 += ", supported by social accountability structures";
  }
  phase2 += ". Willpower gradually becomes unnecessary.";
  predictions.push(phase2);

  let phase3;
  if (severity === "severe") {
    phase3 = `Months 3-4 mark the identity inflection point. ${name} no longer identifies with the old habit. Behavioral choices align with the emerging self-concept rather than requiring active resistance.`;
  } else if (severity === "moderate") {
    phase3 = `By month 3, the habit loses its grip on daily decisions. ${name} experiences increasing moments where the new behavior feels natural, not forced.`;
  } else {
    phase3 = `Within 8-10 weeks, the old habit becomes an occasional temptation rather than a default response. Decision-making shifts from effortful to intuitive.`;
  }
  predictions.push(phase3);

  const outcome = outcomeProjections[goalType] 
    ? outcomeProjections[goalType][ambitionLevel] 
    : outcomeProjections.general[ambitionLevel];
  
  let phase4 = `By month 6, ${outcome} becomes the lived reality`;
  if (context.identityDriven) {
    phase4 += `. The transformation reflects ${name}'s core identity rather than external achievements`;
  }
  phase4 += `. What once required discipline now operates automatically—a new baseline has been established.`;
  predictions.push(phase4);

  return predictions;
}

function buildSmartTimeline({ habitAnalysis, goalAnalysis, severityData, context }) {
  const { primaryType: habitType } = habitAnalysis;
  const { level: severity } = severityData;

  const timelineTemplates = {
    severe: {
      mental: [
        "Intense awareness of thought patterns emerges",
        "Emotional regulation skills develop under pressure",
        "Thought-behavior gap widens noticeably",
        "New neural pathways strengthen through repetition",
        "Old patterns lose their automatic quality",
        "Mental clarity becomes the new baseline"
      ],
      behavioral: [
        "Withdrawal symptoms peak and begin declining",
        "Environmental controls prevent automatic triggers",
        "Urge intensity decreases by 40-50%",
        "Alternative rewards become genuinely satisfying",
        "Identity shifts from 'recovering' to 'changed'",
        "Old behavior feels foreign and unappealing"
      ],
      productivity: [
        "Task avoidance confronted with micro-commitments",
        "Small wins accumulate, reducing fear response",
        "Momentum builds through consistent small actions",
        "Procrastination loses its emotional charge",
        "Proactive behavior becomes more frequent than reactive",
        "High-value work becomes energizing rather than draining"
      ],
      physical: [
        "Physical cravings intensify before declining",
        "Body resets to natural hunger/energy rhythms",
        "Health improvements become noticeable feedback",
        "Physical sensations guide decisions automatically",
        "Energy and vitality replace previous numbness",
        "Physical identity transforms completely"
      ],
      default: [
        "Strong resistance and discomfort surface",
        "Coping mechanisms develop and strengthen",
        "Behavioral control increases significantly",
        "New patterns feel increasingly natural",
        "Old habit loses its psychological grip",
        "Transformation solidifies as permanent identity"
      ]
    },
    moderate: {
      mental: [
        "Mental patterns become visible through observation",
        "Moments of choice expand between trigger and response",
        "Stress response system begins recalibrating",
        "Cognitive flexibility increases noticeably",
        "Emotional reactivity decreases substantially",
        "Mental peace becomes accessible on-demand"
      ],
      behavioral: [
        "Temptations arise but feel manageable",
        "Replacement behaviors begin satisfying similar needs",
        "Slip-ups become learning opportunities, not failures",
        "Consistency builds through small daily decisions",
        "Behavior aligns with values more often than not",
        "New identity as disciplined person fully integrates"
      ],
      productivity: [
        "Resistance noticed and worked with rather than against",
        "Task initiation becomes easier through reduced friction",
        "Productive periods lengthen incrementally each week",
        "Focus capacity strengthens through deliberate practice",
        "Procrastination becomes the exception, not the rule",
        "Work satisfaction and output quality both increase"
      ],
      physical: [
        "Body awareness increases, cravings become readable",
        "Healthier choices feel more rewarding than expected",
        "Physical feedback loop reinforces positive behaviors",
        "Energy levels stabilize and improve",
        "Body confidence grows with visible changes",
        "Physical wellbeing becomes a core life priority"
      ],
      default: [
        "Initial resistance gives way to curiosity",
        "Behavioral adjustments tested and refined",
        "Consistency patterns emerge and stabilize",
        "Confidence in change process deepens",
        "Old habit feels less compelling",
        "New identity locks in permanently"
      ]
    },
    mild: {
      mental: [
        "Awareness of patterns develops gradually",
        "Small mindset shifts accumulate",
        "Mental flexibility improves through practice",
        "Emotional responses become more regulated",
        "Clarity increases in daily decision-making",
        "Mental wellbeing feels substantially improved"
      ],
      behavioral: [
        "Behavioral experiments begin with low stakes",
        "Positive feedback reinforces new choices",
        "Habit cues lose their automatic quality",
        "New routines establish and feel comfortable",
        "Behavioral identity shifts incrementally",
        "Desired behavior operates on autopilot"
      ],
      productivity: [
        "Small productivity wins build early momentum",
        "Task approach anxiety diminishes",
        "Work sessions become more focused",
        "Completion rate increases across projects",
        "Productive identity strengthens",
        "High performance feels natural and sustainable"
      ],
      physical: [
        "Initial changes feel manageable and positive",
        "Body responds positively to new patterns",
        "Physical routines establish without strain",
        "Health markers improve steadily",
        "Physical capacity expands noticeably",
        "Healthy lifestyle becomes effortless default"
      ],
      default: [
        "Awareness and intention set the foundation",
        "Behavioral experiments yield positive results",
        "Consistency builds through gentle persistence",
        "Confidence in process grows steadily",
        "Old habit fades into background",
        "New lifestyle fully integrated"
      ]
    }
  };

  const template = timelineTemplates[severity][habitType] || timelineTemplates[severity].default;
  
  return template.map((item, index) => {
    let enhanced = item;
    
    if (index === 1 && context.environmental) {
      enhanced += " with strategic environment redesign";
    }
    if (index === 2 && context.social) {
      enhanced += " through social support and accountability";
    }
    if (index === 4 && context.identityDriven) {
      enhanced += " as core identity transformation deepens";
    }
    
    return enhanced;
  });
}

function smoothScrollToBottom() {
  setTimeout(() => {
    window.scrollTo({
      top: document.body.scrollHeight,
      behavior: "smooth"
    });
  }, 100);
}

async function generateFuture() {
  const name = document.getElementById("name").value.trim();
  const bad = document.getElementById("currentHabit").value.toLowerCase();
  const good = document.getElementById("newHabit").value.toLowerCase();

  if (!name || !bad || !good) {
    alert("Please fill in all fields to reveal your future.");
    return;
  }

  const generateBtn = document.getElementById("generateBtn");
  generateBtn.disabled = true;
  generateBtn.innerHTML = '<span>✨ Generating...</span>';
  
  const habitAnalysis = analyzeHabit(bad);
  const goalAnalysis = analyzeGoal(good);
  const severityData = analyzeSeverity(bad, good);
  const context = analyzeContext(bad, good);

  const result = document.getElementById("result");
  result.classList.remove("hidden");
  
  smoothScrollToBottom();

  const predictions = buildSmartPrediction({
    name,
    habitAnalysis,
    goalAnalysis,
    severityData,
    context
  });

  const list = document.getElementById("futureList");
  list.innerHTML = "";

  const progressBar = document.getElementById("progressBar");
  let progress = 0;

  for (let i = 0; i < predictions.length; i++) {
    const li = document.createElement("li");
    list.appendChild(li);

    smoothScrollToBottom();
    await typeText(li, predictions[i]);
    
    progress = ((i + 1) / (predictions.length + 6)) * 100;
    progressBar.style.width = progress + "%";
    
    await new Promise(r => setTimeout(r, 300));
    smoothScrollToBottom();
  }

  const timelineData = buildSmartTimeline({
    habitAnalysis,
    goalAnalysis,
    severityData,
    context
  });

  const timeline = document.getElementById("timeline");
  timeline.innerHTML = "";

  for (let i = 0; i < 6; i++) {
    const div = document.createElement("div");
    div.className = "month";

    const h4 = document.createElement("h4");
    const p = document.createElement("p");

    div.append(h4, p);
    timeline.appendChild(div);

    smoothScrollToBottom();
    await typeText(h4, `Month ${i + 1}`, 30);
    await typeText(p, timelineData[i], 18);
    
    progress = ((predictions.length + i + 1) / (predictions.length + 6)) * 100;
    progressBar.style.width = progress + "%";
    
    await new Promise(r => setTimeout(r, 200));
    smoothScrollToBottom();
  }

  document.getElementById("newBtn").classList.remove("hidden");
  generateBtn.disabled = false;
  generateBtn.innerHTML = '<span>✨ Reveal My Future</span>';

  try {
    await db.collection("futureSelf").add({
      name,
      currentHabit: bad,
      newHabit: good,
      habitType: habitAnalysis.primaryType,
      habitSubType: habitAnalysis.subType,
      goalType: goalAnalysis.goalType,
      ambitionLevel: goalAnalysis.ambitionLevel,
      severity: severityData.level,
      previousAttempts: severityData.attempts,
      context: context,
      predictions: predictions,
      timeline: timelineData,
      createdAt: firebase.firestore.FieldValue.serverTimestamp()
    });
  } catch (error) {
    console.error("Firebase save error:", error);
  }
}

function resetPrediction() {
  document.getElementById("name").value = "";
  document.getElementById("currentHabit").value = "";
  document.getElementById("newHabit").value = "";

  document.getElementById("result").classList.add("hidden");
  document.getElementById("newBtn").classList.add("hidden");

  document.getElementById("futureList").innerHTML = "";
  document.getElementById("timeline").innerHTML = "";
  document.getElementById("progressBar").style.width = "0%";

  window.scrollTo({
    top: 0,
    behavior: "smooth"
  });
}