Make sure that the randomly chosen line has at least two words

1 files changed, 7 insertions, 2 deletions

diff --git a/bot.py b/bot.py
index ee87ebb..495a22b 100644
--- a/bot.py
+++ b/bot.py
@@ -48,11 +48,16 @@ if args.input is None:
             'Breaking news:',
             'Have I ever told you that',
             'I read on the news that',
-            'I never knew that'
+            'I never knew that',
+            'My dream is',
+            'It\'s terrible that'
         ])
     else:
         with open(args.data, 'r') as f:
-            line = choice(f.readlines()).split()
+            lines = f.readlines()
+            line = choice(lines).split()
+            while len(line) < 2:
+                line = choice(lines).split()
             args.input = line[0] + ' ' + line[1]