diff --git a/coursework/part2/src/Lib.hs b/coursework/part2/src/Lib.hs index 816ace4..f71e9ef 100644 --- a/coursework/part2/src/Lib.hs +++ b/coursework/part2/src/Lib.hs @@ -85,19 +85,21 @@ dialogStep dict prevPhrase = Just key -> newStdGen >>= \gen -> let p = generatePhrase dict key gen - in putStrLn (unwords p) >> return p + in putStrLn ("(" ++ key ++ ") " ++ unwords p) >> return p twoModelsDialog :: Map String [String] -> Map String [String] -> String -> Int -> IO () twoModelsDialog dict1 dict2 start m = newStdGen >>= \gen -> let first = generatePhrase dict1 start gen - in putStrLn (unwords first) >> + in putStrLn ("Модель 1: (" ++ start ++ ") " ++ unwords first) >> loop dict1 dict2 first m where loop d1 d2 prev 0 = return () loop d1 d2 prev i = + putStr "Модель 2: " >> dialogStep d2 prev >>= \resp -> if null resp then return () else + putStr "Модель 1: " >> dialogStep d1 resp >>= \resp2 -> if null resp2 then return () else loop d1 d2 resp2 (i-1) \ No newline at end of file