chev2/shitpost-video-generator
1
0
Fork 0
mirror of https://github.com/chev2/shitpost-video-generator.git synced 2025-10-30 08:12:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve printing formatting

Browse source 
Also change single quotes to double quotes (personal preference)
This commit is contained in:
Chev 2025-05-24 19:28:46 -07:00
parent 5e1be8fa60
commit 42290f6803
Signed by: chev2
GPG key ID: 0B212D6AED495EC9
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
main.py
View file

@ -49,17 +49,18 @@ random_speed_amount = (0.7, 3)
#float. default: 0.3, 2 #float. default: 0.3, 2
contrast_amount = (0.3, 2) contrast_amount = (0.3, 2)
chosenSeed = input("Video seed (or 'any' to use system time): ") chosenSeed = input("Video seed (or 'any' to pick one automatically): ")
while not chosenSeed.isdecimal() and not chosenSeed in ["any", "skip", "default", "time"]: while not chosenSeed.isdecimal() and not chosenSeed in ["any", "skip", "default", "time"]:
chosenSeed = input("Video seed (or 'any' to use system time): ") chosenSeed = input("Video seed (or 'any' to pick one automatically): ")
seed = int(chosenSeed) if chosenSeed.isdecimal() else random.randrange(maxsize) #get a chosen or random seed to use and reference later seed = int(chosenSeed) if chosenSeed.isdecimal() else random.randrange(maxsize) #get a chosen or random seed to use and reference later
print(f'Chose seed: {seed}') print(f"Chose seed: {seed}")
rng = random.Random(seed) rng = random.Random(seed)
print(f'Found {len(videoFiles)} videos and {len(audioFiles)} sounds') print("")
print(f"Found {len(videoFiles)} videos and {len(audioFiles)} sounds")
# Returns a progress bar string, useful for measuring progress in command-line # Returns a progress bar string, useful for measuring progress in command-line
def ProgressBar(current_index, max_index, progress_bar_len:int = 20, include_percent:bool = True, percent_digits:int = 2, def ProgressBar(current_index, max_index, progress_bar_len:int = 20, include_percent:bool = True, percent_digits:int = 2,
@ -169,10 +170,11 @@ randomVideos = rng.sample(videoFiles, min(videoAmount, len(videoFiles)))
#randomVideos = rng.choices(videoFiles, k=min(videoAmount, len(videoFiles))) #randomVideos = rng.choices(videoFiles, k=min(videoAmount, len(videoFiles)))
if videoAmount > len(videoFiles): #if there is a higher chosen amount than total, re-use videos if videoAmount > len(videoFiles): #if there is a higher chosen amount than total, re-use videos
videoAmountToAdd = videoAmount - len(videoFiles) videoAmountToAdd = videoAmount - len(videoFiles)
print(f'Chosen video amount is higher than available video amount - re-using {videoAmountToAdd} videos...') print(f"Chosen video amount is higher than available video amount - re-using {videoAmountToAdd} videos...")
additionalVideos = rng.choices(videoFiles, k=videoAmountToAdd) additionalVideos = rng.choices(videoFiles, k=videoAmountToAdd)
randomVideos += additionalVideos randomVideos += additionalVideos
print("")
print(f"Compiling {videoAmount} videos... ", end="\r") print(f"Compiling {videoAmount} videos... ", end="\r")
for index, video in enumerate(randomVideos): for index, video in enumerate(randomVideos):
@ -201,10 +203,11 @@ randomSounds = rng.sample(audioFiles, min(audioAmount, len(audioFiles)))
if audioAmount > len(audioFiles): if audioAmount > len(audioFiles):
audioAmountToAdd = audioAmount - len(audioFiles) audioAmountToAdd = audioAmount - len(audioFiles)
print(f'Chosen audio amount is higher than available audio amount - re-using {audioAmountToAdd} audio sources...') print(f"Chosen audio amount is higher than available audio amount - re-using {audioAmountToAdd} audio sources...")
additionalAudio = rng.choices(audioFiles, k=audioAmountToAdd) additionalAudio = rng.choices(audioFiles, k=audioAmountToAdd)
randomSounds += additionalAudio randomSounds += additionalAudio
print("")
print(f"Compiling {audioAmount} sounds...", end="\r") print(f"Compiling {audioAmount} sounds...", end="\r")
copiedSoundAmount = 0 copiedSoundAmount = 0
@ -245,7 +248,7 @@ print(f"Compiling {audioAmount} sounds... {ProgressBar(1, 1, progress_bar_len=40
print(f"Finished compiling audio. Added {copiedSoundAmount} duplicate sounds, total {audioAmount+copiedSoundAmount}.") print(f"Finished compiling audio. Added {copiedSoundAmount} duplicate sounds, total {audioAmount+copiedSoundAmount}.")
# The video's filename # The video's filename
finalVideoFilename = f'output/result_seed-{seed}_{videoAmount}{"_effects" if shouldUseEffects else ""}.mp4' finalVideoFilename = f"output/result_seed-{seed}_{videoAmount}{'_effects' if shouldUseEffects else ''}.mp4"
# Create output directory if it doesn't exist # Create output directory if it doesn't exist
if not path.exists("output"): mkdir("output") if not path.exists("output"): mkdir("output")