mirror of
https://github.com/Paillat-dev/FABLE.git
synced 2026-01-02 01:06:20 +00:00
5498fb7bac
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
19 lines
691 B
Python
19 lines
691 B
Python
import os
|
|
|
|
from utils.openaicaller import openai
|
|
|
|
with open('prompts/script.txt') as f:
|
|
global_prompt = f.read()
|
|
f.close()
|
|
|
|
async def generate_script(title, description, prompt=global_prompt):
|
|
prompt = prompt.replace("[title]", title)
|
|
prompt = prompt.replace("[description]", description)
|
|
'''response = await openai.ChatCompletion.acreate(
|
|
model="gpt-4",
|
|
messages=[
|
|
{"role":"user","content":prompt}
|
|
],
|
|
)''' # Deprecated. Use openaicaller.py instead
|
|
response = await openai.generate_response(model="gpt-4", messages=[{'role':'user', 'content': prompt}])
|
|
return response['choices'][0]['message']['content'] # type: ignore |