#!/usr/bin/env python3.8
from glob import glob
from itertools import islice
from os import path
from textwrap import wrap

from fuzzywuzzy import fuzz, process


transcripts = {}
for filename in glob(path.join('*', '*')):
    with open(filename) as f:
        subtitles = ' '.join(' '.join(islice(f, 2, None, 4)).split())
        transcripts[filename] = subtitles.replace('.', '. ').replace('?', '? ')

while query := input('>>> '):
    bests = process.extractBests(query, transcripts, scorer=fuzz.partial_ratio)
    for index, (transcript, score, filename) in enumerate(bests):
        print(index, filename)
    while index := input('... '):
        try:
            chosen = bests[int(index)]
        except (IndexError, ValueError):
            pass
        else:
            print(chosen[2], *wrap(chosen[0], 80), sep='\n')