mirror of
https://github.com/Paillat-dev/FABLE.git
synced 2026-01-02 01:06:20 +00:00
7fe78c3796
feat(main.py): add translated title and description to meta_FR.txt file feat(main.py): use the new video description variable when uploading the video
66 lines
2.8 KiB
Python
66 lines
2.8 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, translate
|
|
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)
|
|
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()
|
|
for i in range(len(ideas)):
|
|
print(str(i) + ". " + ideas[i]['title'])
|
|
idea = int(input("Which idea do you want to generate a script for? (enter the number): "))
|
|
idea = ideas[idea]
|
|
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}"
|
|
with open(path + "/meta.txt", 'w', encoding='utf-8') as f:
|
|
f.write(description)
|
|
f.close()
|
|
with open(path + "/meta_FR.txt", 'w', encoding='utf-8') as f:
|
|
transtitle = translate('FR', idea['title']) #use the non formatted title
|
|
transdesc = translate('FR', idea['description'])
|
|
final = f"Titre: {transtitle}\nDescription: {transdesc}\nCrédits musicaux: {credits}"
|
|
f.write(final)
|
|
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()) |