middle_earth_adventure/game_core.py

# 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)