import requests
from typing import Dict, Any
from core.config import GEN_AI_URL

URL = GEN_AI_URL


def send_metadata(payload: Dict[str, Any]) -> Dict[str, Any]:
    headers = {
        "Content-Type": "application/json",
        "API_KEY": "testsatupeta"
    }

    try:
        response = requests.post(
            f"{URL}",
            json=payload,
            headers=headers,
        )

        # response.raise_for_status()
        return response.json()

    except requests.exceptions.RequestException as e:
        return {
            "success": False,
            "error": str(e)
        }


if __name__ == "__main__":
    # Contoh payload
    payload = {
        "nama_file_peta": "peta bencana.pdf",
        "nama_opd": "Badan Penanggulangan Bencana Daerah (BPBD)",
        "tipe_data_spasial": "Multipolygon",
        "struktur_atribut_data": {},
        "metadata": {
            "judul": "",
            "abstrak": "",
            "tujuan": "",
            "keyword": [],
            "kategori": [],
            "kategori_mapset": ""
        }
    }

    result = send_metadata(payload)
    print(result)