import spacy
from zshot import PipelineConfig, displacy
from zshot.linker import LinkerRegen, LinkerTARS, LinkerSMXM
from zshot.mentions_extractor import MentionsExtractorSpacy
from zshot.utils.data_models import Entity
from zshot.utils.data_models import Entity, Span
from zshot.mentions_extractor import (MentionsExtractor, MentionsExtractorSpacy, MentionsExtractorFlair,
                                      MentionsExtractorTARS, MentionsExtractorSMXM)
from zshot.mentions_extractor.utils import ExtractorType

nlp = spacy.load('de_core_news_md')
# zero shot definition of entities
nlp_config = PipelineConfig(
    mentions_extractor=MentionsExtractorSpacy(ExtractorType.NER),
    linker=LinkerRegen(),
    entities=[
        Entity(name='Eventtitle',
               description='Der Titel einer Veranstaltung'),
        Entity(name='date',
               description='Das Datum einer Veranstaltung'),
        Entity(name='StartTime',
               description='Die Uhrzeit, an der die Veranstaltung beginnt.'),
        Entity(name='EndTime',
               description='Die Uhrzeit, an der die Veranstaltung endet.'),
        Entity(name='Organizer',
               description='Der Veranstalter einer Veranstaltung.'),
        Entity(name='Price',
               description='Kosten in Euro'),
        ]
)
nlp.add_pipe('zshot', config=nlp_config, last=True)

text = """Workshop Retrodigitalisierung ================================
Thema: Digitalisierte Sammlungen präsentieren – Konzeptionierung, Darstellung und Vermittlung

Der siebte Workshop Retrodigitalisierung findet am 20. und 21. März 2025 bei ZB MED – Informationszentrum Lebenswissenschaften in Köln statt. Er richtet sich an Praktiker:innen, die sich in Bibliotheken mit der Retrodigitalisierung befassen. Wie in den Vorjahren bietet der Workshop ein breites Spektrum an interessanten Vorträgen zur Praxis der Retrodigitalisierung. Dafür brauchen wir Sie und Ihre Einreichungen!

Im Fokus des nächsten Workshops steht die zeitgemäße Präsentation digitalisierter Sammlungen. Das Programm widmet sich insbesondere den Themen Konzeptionierung, Darstellung und Vermittlung von digitalisierten Sammlungen und Beständen über die Präsentationsplattformen der Einrichtungen rund um die Nutzung von Digitalisaten.

Der Call for Presentations läuft noch bis zum 18. Oktober 2024. Wir freuen uns auf Ihren Beitrag!

Der Workshop Retrodigitalisierung wird gemeinsam von den drei deutschen Zentralen Fachbibliotheken TIB – Leibniz-Informationszentrum Technik und Naturwissenschaften, ZB MED – Informationszentrum Lebenswissenschaften und ZBW – Leibniz-Informationszentrum Wirtschaft sowie der Staatsbibliothek zu Berlin – Preußischer Kulturbesitz durchgeführt.

Wann? 20. März bis 21. März 2025

Wo? ZB MED in Köln"""

doc = nlp(text)
displacy.serve(doc, style='ent')