File size: 2,188 Bytes
5f685fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 |
from shortGPT.gpt import gpt_utils
import random
import json
def generateRedditPostMetadata(title):
name = generateUsername()
if title and title[0] == '"':
title = title.replace('"', '')
n_months = random.randint(1,11)
header = f"{name} - {n_months} months ago"
n_comments = random.random() * 10 + 2
n_upvotes = n_comments*(1.2+ random.random()*2.5)
return title, header, f"{n_comments:.1f}k", f"{n_upvotes:.1f}k"
def getInterestingRedditQuestion():
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/reddit_generate_question.yaml')
return gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1.08)
def createRedditScript(question):
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/reddit_generate_script.yaml')
chat = chat.replace("<<QUESTION>>", question)
result = "Reddit, " + question +" "+gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1.08)
return result
def getRealisticness(text):
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/reddit_filter_realistic.yaml')
chat = chat.replace("<<INPUT>>", text)
while True:
try:
result = gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1)
return json.loads(result)['score']
except Exception as e:
print("Error in getRealisticness", e.args[0])
def getQuestionFromThread(text):
if ((text.find("Reddit, ") < 15) and (10 < text.find("?") < 100)):
question = text.split("?")[0].replace("Reddit, ", "").strip().capitalize()
else:
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/reddit_filter_realistic.yaml')
chat = chat.replace("<<STORY>>", text)
question = gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system).replace("\n", "")
question = question.replace('"', '').replace("?", "")
return question
def generateUsername():
chat, system = gpt_utils.load_local_yaml_prompt('prompt_templates/reddit_username.yaml')
return gpt_utils.gpt3Turbo_completion(chat_prompt=chat, system=system, temp=1.2).replace("u/", "")
|