FABLE/main.py

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())