Spaces:
Runtime error
Runtime error
File size: 4,001 Bytes
0974473 520c904 0974473 520c904 0974473 520c904 0974473 520c904 0974473 520c904 0974473 520c904 |
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 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
# imports
import random
import time
from openai import OpenAI
from dotenv import load_dotenv
from middle_earth_adventure.constants import FIGHT_OPTIONS, PLAYER_FEELINGS, RIDDLE_OPTIONS, ROMANCE_OPTIONS
from middle_earth_adventure.prompts import FINISH_PROMPT, SYSTEM_PROMPT, START_PROMPT, CONTINUE_PROMPT
from middle_earth_adventure.utils import pick_rand_items
from middle_earth_adventure.schemas import Player
load_dotenv()
class GameCore:
def __init__(self, api_key:str, text_model:str, tts_model: str) -> None:
self.client = OpenAI(api_key=api_key)
self.text_model = text_model
self.tts_model = tts_model
self.message_history = []
# Start Adventure
async def start_adventure(self, player: Player):
if player is None: return None
self.message_history = []
messages=[
{"role": "system", "content": SYSTEM_PROMPT},
{"role": "user", "content": START_PROMPT.format(name=player.name, sex=player.sex, type=player.type,
skill1=player.skills[0], skill2=player.skills[1])}
]
ai_response = self.client.chat.completions.create(messages=messages, model=self.text_model)
ai_response = ai_response.choices[0].to_dict()["message"]
self.message_history += messages
self.message_history.append(ai_response)
return ai_response['content']
# Continue Adventure
async def continue_adventure(self, selection: str, player: Player):
if player is None: return None
variants, feeling = self.get_game_variations()
print('variants=', str(variants))
message= {"role": "user", "content": CONTINUE_PROMPT.format(selection=selection, feeling=feeling, variants=variants)}
ai_response = self.client.chat.completions.create(messages=[*self.message_history, message], model=self.text_model)
ai_response = ai_response.choices[0].to_dict()["message"]
self.message_history.append(message)
self.message_history.append(ai_response)
return ai_response['content']
async def finish_adventure(self, player, selection):
if player is None: return None
message= {"role": "user", "content": FINISH_PROMPT.format(selection=selection)}
ai_response = self.client.chat.completions.create(messages=[*self.message_history, message],
model=self.text_model,
frequency_penalty=1.0,
temperature=1.7,
max_tokens=100,
)
ai_response = ai_response.choices[0].to_dict()["message"]
self.message_history.append(message)
self.message_history.append(ai_response)
return ai_response['content']
async def narrate_adventure_out_loud(self, text: str, narrator_voice: str):
mp3_narration = self.client.audio.speech.create(model=self.tts_model, voice=narrator_voice, input=text)
mp3_narration = mp3_narration.content
# audio_base64 = base64.b64encode(mp3_narration).decode('utf-8')
return mp3_narration
async def generate_picture_of_the_adventure(self, prompt, model):
if model == "dall-e-2": size = "512x512"
if model == "dall-e-3": size = "1024x1024"
image = self.client.images.generate(model=model, prompt=prompt, response_format='url',
size=size, quality="standard", n=1)
return (image.data[0].url)
def get_game_variations(self):
story_variants = ["(give me a {opt} fight)".format(opt=" and ".join(random.sample(FIGHT_OPTIONS, 2))),
"(give me a {opt} riddle)".format(opt=" and ".join(random.sample(RIDDLE_OPTIONS, 2))),
"(give me an {opt} character encounter)".format(opt=" and ".join(random.sample(ROMANCE_OPTIONS, 2))),
"(add an epic building)"]
feeling = random.choice(PLAYER_FEELINGS)
return random.choice(story_variants), random.choice(feeling)
|