Fix change directory output_dir option on gen-main

diff --git a/nhentai/command.py b/nhentai/command.py
index 47fbc3c3fe9350e72169b057e75ddff048d77404..d0fce3dba577fb760b5343c1f4084c3dbf5c729c 100644 (file)
--- a/nhentai/command.py
+++ b/nhentai/command.py
@@ -94,8 +94,11 @@ def main():
             elif options.is_pdf:
                 generate_pdf(options.output_dir, doujinshi, options.rm_origin_dir)
 
-        if options.main_viewer:
+        if options.main_viewer and options.output_dir:
             generate_main_html(options.output_dir)
+        elif options.main_viewer:
+            generate_main_html()
+
 
         if not platform.system() == 'Windows':
             logger.log(15, '🍻 All done.')