Диалог моделей

coursework/part2/app/Main.hs

@@ -10,6 +10,18 @@ main =
     let sentences = splitText content in
     let dict = buildDictionary sentences in
     saveDictionary "dict.txt" dict >>
-    putStrLn "Введите слово или пару слов:" >>
+    putStrLn "Введите слово или пару слов для генерации фразы:" >>
     getLine >>= \input ->
-    processInput dict input
+    processInput dict input >>
+
+    putStrLn "Введите имя второго файла:" >>
+    getLine >>= \fileName2 ->
+    readFile fileName2 >>= \content2 ->
+    let dict2 = buildDictionary (splitText content2) in
+    saveDictionary "dict2.txt" dict2 >>
+    putStrLn "Введите начальное слово или пару слов для старта диалога:" >>
+    getLine >>= \input2 ->
+    putStrLn "Введите количество сообщений M:" >>
+    getLine >>= \ms ->
+    let m = read ms :: Int in
+    twoModelsDialog dict dict2 input2 m