Spaces:
Runtime error
Runtime error
| # 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) | |