#!/usr/bin/env python3
import zipfile
import os

KML = "mouza_overlay.kml"
IMG = "annotated_map.jpg"
KMZ = "mouza_overlay.kmz"

# ---------- CHECK FILES ----------
if not os.path.exists(KML):
    print("ERROR: KML not found:", KML)
    exit(1)

if not os.path.exists(IMG):
    print("ERROR: Image not found:", IMG)
    exit(1)

# ---------- FIX THE KML (ENSURE CLEAN href) ----------
with open(KML, "r") as f:
    data = f.read()

# Remove possible absolute paths
data = data.replace("/var/www/html/kml_chatgpt/", "")
data = data.replace("\\var\\www\\html\\kml_chatgpt\\", "")

# Rewrite KML
with open(KML, "w") as f:
    f.write(data)

# ---------- BUILD KMZ ----------
with zipfile.ZipFile(KMZ, "w", zipfile.ZIP_DEFLATED) as kmz:
    kmz.write(KML, arcname="mouza_overlay.kml")
    kmz.write(IMG, arcname="annotated_map.jpg")

print("KMZ CREATED SUCCESSFULLY:", KMZ)