mirror of
https://github.com/Paillat-dev/FABLE.git
synced 2026-01-02 09:16:20 +00:00
79 lines
3.4 KiB
Python
79 lines
3.4 KiB
Python
import os
|
|
import json
|
|
import asyncio
|
|
import logging
|
|
|
|
from generators.ideas import generate_ideas
|
|
from generators.script import generate_script
|
|
from generators.montage import mount, prepare
|
|
from generators.miniature import generate_miniature
|
|
from generators.uploader import upload_video
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
async def main():
|
|
if not os.path.exists('videos'): os.makedirs('videos')
|
|
with open('env/subjects.txt', 'r', encoding='utf-8') as f:
|
|
subjects = f.read().splitlines()
|
|
f.close()
|
|
for i in range(len(subjects)):
|
|
print(str(i) + ". " + subjects[i])
|
|
subject = int(input("Which subject do you want to generate ideas for? (enter the number): "))
|
|
subject = subjects[subject]
|
|
subjectdirpath = "videos/" + subject[:25].replace(" ", "_").replace(":", "")
|
|
if not os.path.exists(subjectdirpath):
|
|
os.makedirs(subjectdirpath)
|
|
input("It looks like it is the first time you are generating ideas for this subject. The requiered folder has been created. Press enter to continue.")
|
|
input("Please put all the requiered google credentials files in that folder. Press enter to continue.")
|
|
input("Please put a file called bcg.png in that folder. It will be used as the background of the thumbnails. Press enter to continue.")
|
|
if input("Do you want to generate new ideas? (y/N): ") == "y":
|
|
await generate_ideas(subjectdirpath, subject)
|
|
with open(subjectdirpath + '/ideas.json', 'r', encoding='utf-8') as f:
|
|
ideas = json.load(f)
|
|
f.close()
|
|
existing = []
|
|
new = []
|
|
for i in ideas:
|
|
if os.path.exists(subjectdirpath + "/" + i['title'][:25].replace(" ", "_").replace(":", "") + "/script.json"):
|
|
existing.append(i)
|
|
else:
|
|
new.append(i)
|
|
print("Existing ideas:")
|
|
for i in range(len(existing)):
|
|
print(str(i) + ". " + existing[i]['title'])
|
|
print("New ideas:")
|
|
for i in range(len(new)):
|
|
print(str(i + len(existing)) + ". " + new[i]['title'])
|
|
idea = int(input("Which idea do you want to generate a script for? (enter the number): "))
|
|
if idea < len(existing):
|
|
idea = existing[idea]
|
|
else:
|
|
idea = new[idea - len(existing)]
|
|
title = idea['title']
|
|
title = title[:25]
|
|
i = 0
|
|
path = subjectdirpath + "/" + title
|
|
path = path.replace(" ", "_").replace(":", "")
|
|
if not os.path.exists(path + "/script.json"):
|
|
script = await generate_script(idea['title'], idea['description'])
|
|
if os.path.exists(path) and os.path.exists(path + "/script.json"):
|
|
if input("There is already a script for this idea. Do you want to overwrite it? (y/n)") != "y":
|
|
print("Exiting...")
|
|
exit(1)
|
|
if not os.path.exists(path): os.makedirs(path)
|
|
with open(path + "/script.json", 'w', encoding='utf-8') as f:
|
|
f.write(script)
|
|
f.close()
|
|
script = prepare(path)
|
|
credits = mount(path, script)
|
|
description = f"{idea['description']}\n\nMusic credits: {credits}"
|
|
if credits != None:
|
|
with open(path + "/meta.txt", 'w', encoding='utf-8') as f:
|
|
f.write(description)
|
|
f.close()
|
|
generate_miniature(path, title=idea['title'], description=idea['description'])
|
|
upload_video(path, idea['title'], description, 28, "", "private", subjectdirpath)
|
|
print(f"Your video is ready! You can find it in {path}.")
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main()) |