feat(video.py): add support for script_prompt.txt file to customize script prompts

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

classes/video.py

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

generators/script.py

@@ -6,8 +6,7 @@ with open('prompts/script.txt') as f:
     global_prompt = f.read()
     f.close()
 
-async def generate_script(title, description):
-    prompt = global_prompt
+async def generate_script(title, description, prompt=global_prompt):
     prompt = prompt.replace("[title]", title)
     prompt = prompt.replace("[description]", description)
     '''response = await openai.ChatCompletion.acreate(

main.py

@@ -65,7 +65,12 @@ async def main():
     printm("Done!")
     printm("Here is the video:")
     printm(video)
+    input("Press enter to continue...")
 if __name__ == "__main__":
-    loop = asyncio.get_event_loop()
-    loop.run_until_complete(main())
-    loop.close()
+    while True:
+        asyncio.run(main())
+        try:
+            input("Press enter to continue or type ctrl+c to quit : ")
+            clear_screen()
+        except KeyboardInterrupt:
+            break