from bertopic import BERTopic
import spacy

nlp = spacy.load('de_core_news_lg', exclude=['tagger', 'parser', 'ner', 'attribute_ruler', 'lemmatizer'])

loaded_model = BERTopic.load("../models/bertopic_model", embedding_model=nlp)

# Beispieltext
new_document = """
Workshop zur digitalen Transformation: Lernen Sie, wie Unternehmen erfolgreich digitale Strategien entwickeln und umsetzen. Der Workshop bietet praxisnahe Einblicke in aktuelle digitale Trends und Tools.
"""

# Themen für den neuen Text extrahieren
new_topics, new_probs = loaded_model.transform([new_document])

# Ergebnisse anzeigen
print(f"Themen: {new_topics}")
print(f"Wahrscheinlichkeiten: {new_probs}")

fig = loaded_model.visualize_topics()
fig.show()