diff --git a/scraper/__main__.py b/scraper/__main__.py index 89024cc..56e99c7 100644 --- a/scraper/__main__.py +++ b/scraper/__main__.py @@ -143,8 +143,10 @@ def main(): USER_PASSWORD = getpass.getpass("Enter Password: ") if HEADLESS_MODE is None: - HEADLESS_MODE - str(input("Headless?[Yes/No]")).lower() - + headless_input = input("Headless? [True/False]: ").strip().lower() + HEADLESS_MODE = 'true' if headless_input == 'true' else 'false' + else: + HEADLESS_MODE = HEADLESS_MODE.strip().lower() print() tweet_type_args = []