update

client/src/components/StoryChoices.jsx

@@ -31,6 +31,8 @@ export function StoryChoices({
   disabled = false,
   isLastStep = false,
   isGameOver = false,
+  isDeath = false,
+  isVictory = false,
   containerRef,
 }) {
   const navigate = useNavigate();
@@ -53,13 +55,13 @@ export function StoryChoices({
         <Typography
           variant="h3"
           sx={{
-            color: "white",
+            color: isVictory ? "#4CAF50" : "#f44336",
             textAlign: "center",
             mb: 2,
             textTransform: "uppercase",
           }}
         >
-          The End
+          {isVictory ? "VICTORY" : "DEFEAT"}
         </Typography>
         <Button
           variant="outlined"

client/src/layouts/ComicLayout.jsx

@@ -125,6 +125,8 @@ function ComicPage({
               layout.segments[layout.segments.length - 1]?.isDeath ||
               layout.segments[layout.segments.length - 1]?.isVictory
             }
+            isDeath={layout.segments[layout.segments.length - 1]?.isDeath}
+            isVictory={layout.segments[layout.segments.length - 1]?.isVictory}
           />
         </Box>
       )}

server/core/constants.py

@@ -6,6 +6,10 @@ class GameConfig:
     # Story constraints
     MIN_PANELS = 1
     MAX_PANELS = 4
+
+    MIN_SEGMENTS_BEFORE_END = 2
+    MAX_SEGMENTS_BEFORE_END = 4
+    WINNING_STORY_CHANCE = 0.2
     
     # Story progression
     STORY_BEAT_INTRO = 0

server/core/generators/image_prompt_generator.py

@@ -70,9 +70,6 @@ class ImagePromptGenerator(BaseGenerator):
         IMPORTANT RULES FOR IMAGE PROMPTS:
         - If you are prompting only one panel, it must be an important panel. Dont use only one panel often. It should be a key moment in the story.
         - If you are prompting more than one panel, they must be distinct and meaningful.
-        - For death scenes: Focus on the dramatic and emotional impact, not the gore or violence
-        - For victory scenes: Emphasize triumph, relief, and accomplishment
-        - For victory and death scenes, you MUST use 1 panel only
 
         RESPONSE FORMAT:
         You must return a valid JSON object that matches this Pydantic schema:
@@ -102,6 +99,13 @@ Story text: {story_text}
 
 Generate panel descriptions that capture the key moments of this scene.
 
+
+
+- For death scenes: Focus on the dramatic and emotional impact, not the gore or violence
+- For victory scenes: Emphasize triumph, relief, and accomplishment
+- For victory and death scenes, you MUST use 1 panel only
+
+
 Story state: {is_end}
 """
 
@@ -162,7 +166,7 @@ Story state: {is_end}
         metadata = f"[{time} - {location}] "
         return f"{self.artist_style} -- {metadata}{prompt}"
 
-    async def generate(self, story_text: str, time: str, location: str, is_death: bool = False, is_victory: bool = False) -> ImagePromptResponse:
+    async def generate(self, story_text: str, time: str, location: str, is_death: bool = False, is_victory: bool = False, turn_before_end: int = 0, is_winning_story: bool = False) -> ImagePromptResponse:
         """Generate image prompts based on story text.
         
         Args:
@@ -178,9 +182,10 @@ Story state: {is_end}
 
         is_end=""
         if is_death:
-            is_end = "this is a death to represent"
+            is_end = "this is a death. just one panel, MANDATORY."
         elif is_victory:
-            is_end = "this is a victory to represent"
+            is_end = "this is a victory. just one panel, MANDATORY."
+
 
         response = await super().generate(
             story_text=story_text,

server/core/generators/metadata_generator.py

@@ -85,7 +85,7 @@ Current game state:
         except ValueError as e:
             raise ValueError(str(e))
 
-    async def generate(self, story_text: str, current_time: str, current_location: str, story_beat: int, error_feedback: str = "") -> StoryMetadataResponse:
+    async def generate(self, story_text: str, current_time: str, current_location: str, story_beat: int, error_feedback: str = "", turn_before_end: int = 0, is_winning_story: bool = False) -> StoryMetadataResponse:
         """Surcharge de generate pour inclure le error_feedback par défaut."""
 
         is_end = "This should be close to the end of the story." if story_beat >= 5 else ""
@@ -95,5 +95,7 @@ Current game state:
             current_location=current_location,
             story_beat=story_beat,
             error_feedback=error_feedback,
-            is_end=is_end
+            is_end=is_end,
+            turn_before_end=turn_before_end,
+            is_winning_story=is_winning_story
         ) 

server/core/generators/story_segment_generator.py

@@ -57,7 +57,7 @@ Rules: {FORMATTING_RULES}
 
 You must return a JSON object with the following format:
 {{
-    "story_text": "Your story segment here (15 words)"
+    "story_text": "Your story segment here"
 }}
 """
 
@@ -72,10 +72,6 @@ Current game state :
 - Story beat: {story_beat}
 
 {is_end}
-You must return a JSON object with the following format:
-{{
-    "story_text": "Your story segment here (15 words)"
-}}
 """
         return ChatPromptTemplate(
             messages=[
@@ -109,10 +105,11 @@ You must return a JSON object with the following format:
                 "Example: {'story_text': 'Your story segment here'}"
             )
 
-    async def generate(self, story_beat: int, current_time: str, current_location: str, previous_choice: str, story_history: str = "") -> StorySegmentResponse:
+    async def generate(self, story_beat: int, current_time: str, current_location: str, previous_choice: str, story_history: str = "", turn_before_end: int = 0, is_winning_story: bool = False) -> StorySegmentResponse:
         """Generate the next story segment."""
 
-        is_end = "Generate the END of the story. Reminder: 30 words." if story_beat >= random.randint(5, 10) else "Generate the next segment of the story. REMINDER: 15 words."
+        what_to_represent =" this is a victory !" if is_winning_story else "this is a death !"
+        is_end = f"Generate the END of the story. {what_to_represent} in 30 words. THIS IS MANDATORY." if story_beat == turn_before_end  else "Generate the next segment of the story in 15 words."
         
         return await super().generate(
             HERO_DESCRIPTION=HERO_DESCRIPTION,

server/core/story_generator.py

@@ -6,6 +6,8 @@ from core.generators.story_segment_generator import StorySegmentGenerator
 from core.generators.image_prompt_generator import ImagePromptGenerator
 from core.generators.metadata_generator import MetadataGenerator
 from core.game_state import GameState
+import random
+from core.constants import GameConfig
 
 class StoryGenerator:
     _instance = None
@@ -22,6 +24,8 @@ class StoryGenerator:
             print("Initializing StoryGenerator singleton")
             self.api_key = api_key
             self.model_name = model_name
+            self.turn_before_end = random.randint(GameConfig.MIN_SEGMENTS_BEFORE_END, GameConfig.MAX_SEGMENTS_BEFORE_END)
+            self.is_winning_story = random.random() < GameConfig.WINNING_STORY_CHANCE
             self.mistral_client = MistralClient(api_key=api_key, model_name=model_name)
             self.image_prompt_generator = None  # Will be initialized with the first universe style
             self.metadata_generator = MetadataGenerator(self.mistral_client)
@@ -76,7 +80,9 @@ class StoryGenerator:
             current_time=game_state.current_time,
             current_location=game_state.current_location,
             previous_choice=previous_choice,
-            story_history=story_history
+            story_history=story_history,
+            turn_before_end=self.turn_before_end,
+            is_winning_story=self.is_winning_story
         )
 
         # print(f"Generated story text: {segment_response}")
@@ -86,7 +92,9 @@ class StoryGenerator:
             story_text=segment_response.story_text,
             current_time=game_state.current_time,
             current_location=game_state.current_location,
-            story_beat=game_state.story_beat
+            story_beat=game_state.story_beat,
+            turn_before_end=self.turn_before_end,
+            is_winning_story=self.is_winning_story
         )
         # print(f"Generated metadata_response: {metadata_response}")
         
@@ -96,7 +104,9 @@ class StoryGenerator:
             time=metadata_response.time,
             location=metadata_response.location,
             is_death=metadata_response.is_death,
-            is_victory=metadata_response.is_victory
+            is_victory=metadata_response.is_victory,
+            turn_before_end=self.turn_before_end,
+            is_winning_story=self.is_winning_story
         )
         # print(f"Generated image prompts: {prompts_response}")