add --file and --output support

Author: mateofumis

sepunycoder

@@ -1,12 +1,12 @@
 #!/usr/bin/python3
 import time
-import os
+import argparse
+
 def translte_to_cyrillic(text):
     punycode_table = {
- 'A': 'A', 'B': 'B', 'C': 'C', 'D': 'D', 'E': 'E', 'F': 'F', 'G': 'G', 'H': 'H', 'I': 'І', 'J': 'Ј', 'K': 'К', 'L': 'L', 'M': 'М', 'N': 'N', 'O': 'О', 'P': 'Р', 'Q': 'Ԛ', 'R': 'R', 'S': 'Ѕ', 'T': 'T', 'U': 'U', 'V': 'V', 'W': 'W', 'X': 'Х', 'Y': 'Y', 'Z': 'Z',
- 'a': 'а', 'b': 'b', 'c': 'с', 'd': 'd', 'e': 'е', 'f': 'f', 'g': 'g', 'h': 'h', 'i': 'і', 'j': 'ј', 'k': 'k', 'l': 'l', 'm': 'm', 'n': 'n', 'o': 'о', 'p': 'р', 'q': 'ԛ', 'r': 'r', 's': 'ѕ', 't': 't', 'u': 'u', 'v': 'v', 'w': 'w', 'x': 'х', 'y': 'у', 'z': 'z'
-}
-
+        'A': 'A', 'B': 'B', 'C': 'C', 'D': 'D', 'E': 'E', 'F': 'F', 'G': 'G', 'H': 'H', 'I': 'І', 'J': 'Ј', 'K': 'К', 'L': 'L', 'M': 'М', 'N': 'N', 'O': 'О', 'P': 'Р', 'Q': 'Ԛ', 'R': 'R', 'S': 'Ѕ', 'T': 'T', 'U': 'U', 'V': 'V', 'W': 'W', 'X': 'Х', 'Y': 'Y', 'Z': 'Z',
+        'a': 'а', 'b': 'b', 'c': 'с', 'd': 'd', 'e': 'е', 'f': 'f', 'g': 'g', 'h': 'h', 'i': 'і', 'j': 'ј', 'k': 'k', 'l': 'l', 'm': 'm', 'n': 'n', 'o': 'о', 'p': 'р', 'q': 'ԛ', 'r': 'r', 's': 'ѕ', 't': 't', 'u': 'u', 'v': 'v', 'w': 'w', 'x': 'х', 'y': 'у', 'z': 'z'
+    }
     result = ''
     for char in text:
         if char.lower() in punycode_table:
@@ -16,21 +16,12 @@ def translte_to_cyrillic(text):
                 result += punycode_table[char.lower()]
         else:
             result += char
-
     return result
 
-# Colors
 green_color = "\033[1;32m"
 end_color = "\033[0m"
-red_color = "\033[1;31m"
-blue_color = "\033[1;34m"
-yellow_color = "\033[1;33m"
-purple_color = "\033[1;35m"
-turquoise_color = "\033[1;36m"
-gray_color = "\033[1;37m"
 
-# Print Welcome Banner
-banner=r'''{}
+banner = r'''{}
   ______ _______ ______                                  _
  / _____|_______|_____ \                                | |
 ( (____  _____   _____) )   _ ____  _   _  ____ ___   __| |_____  ____
@@ -45,14 +36,25 @@ for line in lines:
     print(line)
     time.sleep(0.08)
 
-# Get user Input for Text to Translate
-user_input = input("🔓 Enter the text to translate to Cyrillic: ")
-user_input_https = input("🔗 Do you want to add \"https://\" protocol? (y/n): ")
-output_text = translte_to_cyrillic(user_input)
-if user_input_https == "y":
-    output_text = "https://" + output_text
-    print("👀 Translated text:", output_text)
-elif user_input_https == "n":
-    print("👀 Translated text:", output_text)
+parser = argparse.ArgumentParser(description="Translate text to Cyrillic punycode.")
+parser.add_argument('--file', type=str, help="Input file containing the text to translate")
+parser.add_argument('--output', type=str, help="Output file to save the translated text")
+args = parser.parse_args()
+
+if args.file:
+    with open(args.file, 'r') as file:
+        user_input = file.read()
+    output_text = translte_to_cyrillic(user_input)
+else:
+    user_input = input("🔓 Enter the text to translate to Cyrillic: ")
+    user_input_https = input("🔗 Do you want to add \"https://\" protocol? (y/n): ")
+    output_text = translte_to_cyrillic(user_input)
+    if user_input_https == "y":
+        output_text = "https://" + output_text
+
+if args.output:
+    with open(args.output, 'w') as file:
+        file.write(output_text)
+    print(f"👀 Translated text saved to {args.output}")
 else:
     print("👀 Translated text:", output_text)