File size: 3,739 Bytes
a61ba58 78b81a5 a61ba58 78b81a5 a61ba58 78b81a5 a61ba58 78b81a5 a61ba58 78b81a5 a61ba58 78b81a5 |
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 88 89 90 91 92 |
from core.constants import GameConfig
from typing import List
from api.models import StoryResponse
class GameState:
def __init__(self):
self.story_beat = GameConfig.STORY_BEAT_INTRO
self.story_history: List[StoryResponse] = []
self.current_time = GameConfig.STARTING_TIME
self.current_location = GameConfig.STARTING_LOCATION
# Universe information
self.universe_style = None
self.universe_genre = None
self.universe_epoch = None
self.universe_story = None
def reset(self):
"""Reset game state while keeping universe information."""
# Save universe info
universe_style = self.universe_style
universe_genre = self.universe_genre
universe_epoch = self.universe_epoch
universe_story = self.universe_story
# Reset game state
self.story_beat = GameConfig.STORY_BEAT_INTRO
self.story_history = []
self.current_time = GameConfig.STARTING_TIME
self.current_location = GameConfig.STARTING_LOCATION
# Restore universe info
self.universe_style = universe_style
self.universe_genre = universe_genre
self.universe_epoch = universe_epoch
self.universe_story = universe_story
def set_universe(self, style: str, genre: str, epoch: str, base_story: str):
"""Configure the game universe."""
self.universe_style = style
self.universe_genre = genre
self.universe_epoch = epoch
self.universe_story = base_story
def has_universe(self) -> bool:
"""Check if universe is configured."""
return all([
self.universe_style is not None,
self.universe_genre is not None,
self.universe_epoch is not None,
self.universe_story is not None
])
def format_history(self) -> str:
"""Format story history for the prompt.
Returns only the last 4 segments of the story (or less if not available)."""
if not self.story_history:
return ""
# Ne prendre que les 3 derniers segments
last_segments = self.story_history[-4:] if len(self.story_history) > 4 else self.story_history
segments = []
for story_response in last_segments:
# Commencer par le choix précédent s'il existe
segment_parts = []
if story_response.previous_choice and story_response.previous_choice != "none":
segment_parts.append(f"[Previous choice: {story_response.previous_choice}]")
# Ajouter le texte de l'histoire
segment_parts.append(story_response.story_text)
# Ajouter les choix disponibles s'ils existent
if story_response.choices:
choices_text = "\nAvailable choices were:"
for choice in story_response.choices:
choices_text += f"\n- {choice.text}"
segment_parts.append(choices_text)
# Joindre toutes les parties avec des sauts de ligne
segments.append("\n".join(segment_parts))
# Ajouter une indication si on a tronqué l'historique
if len(self.story_history) > 4:
segments.insert(0, f"[...{len(self.story_history) - 4} earlier segments omitted...]")
return "\n\n---\n\n".join(segments)
def add_to_history(self, story_response: StoryResponse):
"""Add a story response to history."""
self.story_history.append(story_response)
self.current_time = story_response.time
self.current_location = story_response.location |