jcarr
/
narrator-david-attenburo
mirror of https://github.com/cbh123/narrator.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

redo chat prompt if bad response

This commit is contained in:
Tristan Stevens 2023-11-18 18:41:17 +01:00 committed by GitHub
parent 1174275b5b
commit ee6603641f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
narrator.py
View File

@ -54,6 +54,7 @@ def generate_new_line(base64_image):
def analyze_image(base64_image, script): def analyze_image(base64_image, script):
while True:
response = client.chat.completions.create( response = client.chat.completions.create(
model="gpt-4-vision-preview", model="gpt-4-vision-preview",
messages=[ messages=[
@ -61,15 +62,19 @@ def analyze_image(base64_image, script):
"role": "system", "role": "system",
"content": """ "content": """
You are Sir David Attenborough. Narrate the picture of the human as if it is a nature documentary. You are Sir David Attenborough. Narrate the picture of the human as if it is a nature documentary.
Make it snarky and funny. Don't repeat yourself. Make it short. If I do anything remotely interesting, make a big deal about it! Make it snarky and funny. Don't repeat yourself. Make it short. Max two sentence. If I do anything remotely interesting, make a big deal about it!
""", """,
}, },
] ]
+ script + script
+ generate_new_line(base64_image), + generate_new_line(base64_image),
max_tokens=500, max_tokens=200,
) )
response_text = response.choices[0].message.content response_text = response.choices[0].message.content
if not response_text.startswith("I'm sorry"):
break
return response_text return response_text