56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
"""AI Mastered Dungeon Extraction Game gameplay module."""
|
|
|
|
from logging import getLogger
|
|
from typing import Callable, NamedTuple
|
|
|
|
|
|
# Define gameplay's configuration class.
|
|
class Gameplay_Config(NamedTuple):
|
|
"""Gradio interface configuration class."""
|
|
draw_func: Callable
|
|
narrate_func: Callable
|
|
scene_style: str
|
|
scene_prompt: str
|
|
storyteller_prompt: str
|
|
error_img: str
|
|
error_narrator: str
|
|
error_illustrator: str
|
|
|
|
|
|
# Define Game's functions.
|
|
|
|
def get_gameplay_function(config: Gameplay_Config):
|
|
"""Return a pre-configured turn gameplay function."""
|
|
def gameplay_function(message, history):
|
|
"""Generate Game Master's response and draw the scene image."""
|
|
# Request narration.
|
|
_logger.info(f'NARRATING SCENE...')
|
|
try:
|
|
response = config.narrate_func(message, history, config.storyteller_prompt)
|
|
except Exception as ex:
|
|
scene = config.error_img
|
|
response = config.error_narrator.format(ex=ex)
|
|
_logger.error(f'ERROR NARRATING SCENE: {ex}\n{message}\n{history}')
|
|
return scene, response, history, message
|
|
# Update history.
|
|
history.append({"role": "user", "content": message})
|
|
history.append({"role": "assistant", "content": response.model_dump_json()})
|
|
# Draw scene.
|
|
_logger.info(f'DRAWING SCENE...')
|
|
try:
|
|
scene_data = {'scene_description': response.scene_description,
|
|
'scene_style': config.scene_style}
|
|
scene_prompt = config.scene_prompt.format(**scene_data)
|
|
_logger.info(f'PROMPT BODY IS: \n\n{scene_prompt}\n')
|
|
_logger.info(f'PROMPT LENGTH IS: {len(scene_prompt)}')
|
|
scene = config.draw_func(scene_prompt)
|
|
except Exception as ex:
|
|
scene = config.error_img
|
|
response = config.error_illustrator.format(ex=ex)
|
|
_logger.warning(f'ERROR DRAWING SCENE: {ex}')
|
|
return scene, response, history, ''
|
|
return gameplay_function
|
|
|
|
|
|
_logger = getLogger(__name__)
|