module Main (main) where

import Lib

main :: IO ()
main =
    putStrLn "Введите имя файла:" >>
    getLine >>= \fileName ->
    readFile fileName >>= \content ->
    let sentences = splitText content in
    let dict = buildDictionary sentences in
    saveDictionary "dict.txt" dict >>
    putStrLn "Введите слово или пару слов для генерации фразы:" >>
    getLine >>= \input ->
    processInput dict input >>

    putStrLn "Введите имя второго файла:" >>
    getLine >>= \fileName2 ->
    readFile fileName2 >>= \content2 ->
    let dict2 = buildDictionary (splitText content2) in
    saveDictionary "dict2.txt" dict2 >>
    putStrLn "Введите начальное слово или пару слов для старта диалога:" >>
    getLine >>= \input2 ->
    putStrLn "Введите количество сообщений от каждой модели:" >>
    getLine >>= \ms ->
    let m = read ms :: Int in
    twoModelsDialog dict dict2 input2 m