import nltk
from nltk.corpus import PlaintextCorpusReader
from nltk.text import Text

# Pfad zum Verzeichnis mit den Dateien
corpus_root = 'F:\\Varlaam_Iasaf\\Ruthenisch-Text-Proccess\\txt_norm'
corpus_root = 'F:\\AKSL\\CCMH\\txt'

# Erstellen des Corpus Readers für alle Dateien im Verzeichnis
wordlists = PlaintextCorpusReader(corpus_root, '.*')

# Liste der Dateinamen im Verzeichnis
file_names = wordlists.fileids()

# Liste der interessanten Wörter, deren Kontext Sie vergleichen möchten
words_to_search = ["тое", "на", "и"]

# Durchgehen jeder Datei und Anzeigen des Kontexts der interessanten Wörter
for file in file_names:
    print(f"Kontext in Datei: {file}")
    
    # Erstellen eines Text-Objekts für die aktuelle Datei
    text = Text(wordlists.words(file))
    
    for word in words_to_search:
        print(f"Kontext für '{word}':")
        text.concordance(word, lines=5)  # Anzeigen der ersten 5 Zeilen des Kontextes
        print("\n")