from src.nlp.playground.ner import GlinerHandler
from src.utils.apis.googlemaps_api import GoogleMapsAPI


class AddressExtractor(GoogleMapsAPI):
    def __init__(self):
        super().__init__()
        self.gliner_handler = GlinerHandler()


    def extract_address(self, data):
        entities = self.gliner_handler.extract_entities(data, ["STRASSE",  "STADT"])
        street = house_number = postal_code = city = None
        for entity in entities:
            if entity["label"] == "STRASSE":
                if street is None:
                    street = entity["text"]
            if entity["label"] == "STADT":
                if city is None:
                    city = entity["text"]


        address_text = " ".join([
            street if street else "",
            house_number if house_number else "",
            postal_code if postal_code else "",
            city if city else ""
        ])

        return self.address_validation(address_text)