28 lines
1.0 KiB
Haskell
28 lines
1.0 KiB
Haskell
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
|