Get input dir from cmdline

1 files changed, 2 insertions, 0 deletions

diff --git a/main.py b/main.py
index a9300c0..3a59b9c 100644
--- a/main.py
+++ b/main.py
@@ -7,6 +7,8 @@ from converters.links_to_gemini import convert_links_to_gemini, convert_file_lin
 import shutil
 
 dir = "input"
+if len(sys.argv) > 1:
+    dir=sys.argv[1]
 
 def wipe_old():
     for root, dirs, files in os.walk("output"):