Paillat-dev/FABLE
1
0
Fork 0
mirror of https://github.com/Paillat-dev/FABLE.git synced 2026-01-02 01:06:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(main.py): add support for existing ideas in the idea selection process

Browse Source
This commit is contained in:
Paillat
2023-05-28 11:26:13 +02:00
parent ef684ffff5
commit 7c6989161c
1 changed files with 18 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
main.py
View File

@@ -26,15 +26,29 @@ async def main():
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("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 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.") 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": if input("Do you want to generate new ideas? (y/N): ") == "y":
await generate_ideas(subjectdirpath, subject) await generate_ideas(subjectdirpath, subject)
with open(subjectdirpath + '/ideas.json', 'r', encoding='utf-8') as f: with open(subjectdirpath + '/ideas.json', 'r', encoding='utf-8') as f:
ideas = json.load(f) ideas = json.load(f)
f.close() f.close()
for i in range(len(ideas)): existing = []
print(str(i) + ". " + ideas[i]['title']) 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): ")) idea = int(input("Which idea do you want to generate a script for? (enter the number): "))
idea = ideas[idea] if idea < len(existing):
idea = existing[idea]
else:
idea = new[idea - len(existing)]
title = idea['title'] title = idea['title']
title = title[:25] title = title[:25]
i = 0 i = 0