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(video.py): add support for script_prompt.txt file to customize script prompts

Browse Source 
fix(script.py): add prompt parameter to generate_script function
feat(main.py): add loop to run main function indefinitely and add option to quit with ctrl+c
This commit is contained in:
Paillat
2023-06-25 19:14:22 +02:00
parent f0dff73ce7
commit 5498fb7bac
3 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
classes/video.py
View File

@@ -35,7 +35,15 @@ class Video:
os.remove(os.path.join( self.path, "script.json"))
if not os.path.exists(os.path.join( self.path, "script.json")):
script = await generate_script(self.idea['title'], self.idea['description'])
script_prompt = None
if os.path.exists(os.path.join(self.parent.path, "script_prompt.txt")):
with open(os.path.join(self.parent.path, "script_prompt.txt"), "r") as f:
script_prompt = f.read()
f.close()
if script_prompt:
script = await generate_script(self.idea['title'], self.idea['description'], script_prompt)
else:
script = await generate_script(self.idea['title'], self.idea['description'])
with open(os.path.join( self.path, "script.json"), "w") as f:
json.dump(json.loads(script), f)
f.close()