fix console output #2

generate_long_video.py

@@ -1,7 +1,11 @@
 from pathlib import Path
 from utils import settings
+from video_generator import VideoGenerator
 import sys
 
+abs_directory = Path().absolute()
+config = settings.check_toml(f"{abs_directory}/utils/.config.template.toml", f"{abs_directory}/config.toml")
+
 num_of_vids_to_be_conc = input("How many videos would you like to be concatenated into a single video? ")
 try:
     num_of_vids_to_be_conc = int(num_of_vids_to_be_conc)
@@ -16,13 +20,8 @@ else:
     if remove_past_results == "y": remove_past_results = True
     if remove_past_results == "n": remove_past_results = False
 
-abs_directory = Path().absolute()
-config = settings.check_toml(f"{abs_directory}/utils/.config.template.toml", f"{abs_directory}/config.toml")
-
 config["settings"]["resolution_w"] = 1920
 config["settings"]["resolution_h"] = 1080
 
-from video_generator import VideoGenerator
-
 video_gen = VideoGenerator(config)
 video_gen.generate_long_video(num_of_vids_to_be_conc, remove_past_results)

generate_shorts.py

@@ -1,14 +1,18 @@
 from pathlib import Path
 from utils import settings
+from video_generator import VideoGenerator
 import sys
 
-num_of_shorts = input("How many shorts would you like to be generated? ")
+abs_directory = Path().absolute()
+config = settings.check_toml(f"{abs_directory}/utils/.config.template.toml", f"{abs_directory}/config.toml")
+
+num_of_shorts = input("\nHow many shorts would you like to be generated? ")
 try:
     num_of_shorts = int(num_of_shorts)
 except:
     print("Input could not be converted to integer, please enter a number!")
     sys.exit(400)
-    
+
 remove_past_results = input("Would you like to remove past results? (y/n) ")
 
 if remove_past_results not in ["y", "n"]: print("Please enter a valid option (y/n)!"); sys.exit(400)
@@ -16,14 +20,8 @@ else:
     if remove_past_results == "y": remove_past_results = True
     if remove_past_results == "n": remove_past_results = False
 
-
-abs_directory = Path().absolute()
-config = settings.check_toml(f"{abs_directory}/utils/.config.template.toml", f"{abs_directory}/config.toml")
-
 config["settings"]["resolution_w"] = 1080
 config["settings"]["resolution_h"] = 1920
 
-from video_generator import VideoGenerator
-
 video_gen = VideoGenerator(config)
 video_gen.generate_shorts(num_of_shorts, remove_past_results)