update

client/src/App.jsx

@@ -65,6 +65,8 @@ function App() {
             text: response.data.story_text,
             isChoice: false,
             isDeath: response.data.is_death,
+            isVictory: response.data.is_victory,
+            radiationLevel: response.data.radiation_level,
             imageUrl: imageUrl,
           },
         ]);
@@ -75,6 +77,8 @@ function App() {
             text: response.data.story_text,
             isChoice: false,
             isDeath: response.data.is_death,
+            isVictory: response.data.is_victory,
+            radiationLevel: response.data.radiation_level,
             imageUrl: imageUrl,
           },
         ]);
@@ -130,14 +134,45 @@ function App() {
           <Typography variant="h4" component="h1">
             Echoes of Influence
           </Typography>
-          <Button
-            variant="outlined"
-            startIcon={<RestartAltIcon />}
-            onClick={() => handleStoryAction("restart")}
-            disabled={isLoading}
-          >
-            Restart
-          </Button>
+          <Box sx={{ display: "flex", alignItems: "center", gap: 2 }}>
+            <Box
+              sx={{
+                display: "flex",
+                alignItems: "center",
+                bgcolor: "warning.main",
+                color: "white",
+                px: 2,
+                py: 1,
+                borderRadius: 1,
+                "& .radiation-value": {
+                  color:
+                    storySegments.length > 0 &&
+                    storySegments[storySegments.length - 1].radiationLevel >= 7
+                      ? "error.light"
+                      : "inherit",
+                },
+              }}
+            >
+              <Typography variant="body1" component="span">
+                Radiation:{" "}
+                <span className="radiation-value">
+                  {storySegments.length > 0
+                    ? `${
+                        storySegments[storySegments.length - 1].radiationLevel
+                      }/10`
+                    : "0/10"}
+                </span>
+              </Typography>
+            </Box>
+            <Button
+              variant="outlined"
+              startIcon={<RestartAltIcon />}
+              onClick={() => handleStoryAction("restart")}
+              disabled={isLoading}
+            >
+              Restart
+            </Button>
+          </Box>
         </Box>
 
         {isLoading && <LinearProgress sx={{ mb: 2 }} />}
@@ -160,11 +195,13 @@ function App() {
                   maxWidth: "80%",
                   bgcolor: segment.isDeath
                     ? "error.light"
+                    : segment.isVictory
+                    ? "success.light"
                     : segment.isChoice
                     ? "primary.light"
                     : "grey.100",
                   color:
-                    segment.isDeath || segment.isChoice
+                    segment.isDeath || segment.isVictory || segment.isChoice
                       ? "white"
                       : "text.primary",
                 }}
@@ -173,6 +210,8 @@ function App() {
                   primary={
                     segment.isDeath
                       ? "DEATH"
+                      : segment.isVictory
+                      ? "VICTOIRE"
                       : segment.isChoice
                       ? "Your Choice"
                       : "Story"

server/game/game_logic.py

@@ -3,6 +3,7 @@ from typing import List
 from langchain_mistralai.chat_models import ChatMistralAI
 from langchain.output_parsers import PydanticOutputParser, OutputFixingParser
 from langchain.prompts import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessagePromptTemplate
+from langchain.schema import HumanMessage, SystemMessage
 
 # Game constants
 MAX_RADIATION = 10
@@ -18,11 +19,28 @@ class GameState:
 
 # Story output structure
 class StorySegment(BaseModel):
-    story_text: str = Field(description="The next segment of the story")
+    story_text: str = Field(description="The next segment of the story. Like 20 words.")
     choices: List[str] = Field(description="Exactly two possible choices for the player", min_items=2, max_items=2)
-    is_death: bool = Field(description="Whether this segment ends in Sarah's death", default=False)
+    is_victory: bool = Field(description="Whether this segment ends in Sarah's victory", default=False)
     radiation_increase: int = Field(description="How much radiation this segment adds (0-3)", ge=0, le=3, default=1)
 
+# Prompt templates
+SYSTEM_ART_PROMPT = """You are an expert in image generation prompts.
+Transform the story into a short and precise prompt.
+
+Strict format:
+"color comic panel, style of Hergé, [main scene in 5-7 words], french comic panel"
+
+Example:
+"color comic panel, style of Hergé, detective running through dark alley, french comic panel"
+
+Rules:
+- Maximum 20 words to describe the scene
+- No superfluous adjectives
+- Capture only the main action"""
+
+HUMAN_ART_PROMPT = "Transform into a short prompt: {story_text}"
+
 class StoryGenerator:
     def __init__(self, api_key: str):
         self.parser = PydanticOutputParser(pydantic_object=StorySegment)
@@ -44,45 +62,51 @@ class StoryGenerator:
         self.prompt = self._create_prompt()
         
     def _create_prompt(self) -> ChatPromptTemplate:
-        system_template = """You are narrating an EXTREMELY lethal dystopian story. Your goal is to kill Sarah in creative and brutal ways unless players make PERFECT choices. This is a horror survival game where death is the most common outcome.
+        system_template = """You are narrating a brutal dystopian story where Sarah must survive in a radioactive wasteland. This is a comic book story.
 
-IMPORTANT: The first story beat (story_beat = 0) MUST be an introduction that sets up the horror atmosphere but CANNOT kill Sarah. After that, death should be frequent.
+IMPORTANT: The first story beat (story_beat = 0) MUST be an introduction that sets up the horror atmosphere.
 
 RADIATION SYSTEM:
-- Each segment must specify a radiation_increase value (0-3)
-- 0: Safe area or good protection
-- 1: Standard background radiation
-- 2: Dangerous exposure
-- 3: Critical radiation levels
+You must set a radiation_increase value for each segment based on the environment and situation:
+- 0: Completely safe area (rare, only in bunkers or heavily shielded areas)
+- 1: Standard exposure (most common, for regular exploration)
+- 2: Elevated risk (when near radiation sources or in contaminated areas)
+- 3: Critical exposure (very rare, only in extremely dangerous situations)
+
+IMPORTANT RULES FOR RADIATION:
+- DO NOT mention radiation values in the choices
+- Most segments should have radiation_increase = 1
+- Use 2 or 3 only in specific dangerous areas
+- Use 0 only in safe shelters
 - Current radiation level: {radiation_level}/10
-- If radiation reaches 10, Sarah dies horribly
+- Death occurs automatically when radiation reaches 10
 
 Core story elements:
 - Sarah is deeply traumatized by the AI uprising that killed most of humanity
 - She abandoned her sister during the Great Collapse, leaving her to die
-- She's on a suicide mission, but a quick death is not redemption
-- The radiation is EXTREMELY lethal - even minor exposure causes severe damage
-- Most choices should lead to death (except in introduction)
-- The environment actively tries to kill Sarah (raiders, AI, radiation, traps)
+- She's on a mission of redemption in this hostile world
+- The radiation is an invisible, constant threat
+- The environment is full of dangers (raiders, AI, traps)
+- Focus on survival horror and tension
 
 Each response MUST contain:
-1. A detailed story segment that puts Sarah in mortal danger (except in introduction), describing:
-   - The horrific environment
-   - The immediate threats to her life
-   - Her deteriorating physical state (based on radiation_level)
-   - Her mental state and previous choices
+1. A detailed story segment that:
+   - Describes the horrific environment
+   - Shows immediate dangers
+   - Details Sarah's physical state (based on radiation_level)
+   - Reflects her mental state and previous choices
 
 2. Exactly two VERY CONCISE choices (max 10 words each):
    Examples of good choices:
-   - "Rush through radiation zone (+3 radiation)" vs "Take long way (+1 radiation)"
-   - "Trust the survivor" vs "Shoot on sight"
+   - "Explore the abandoned hospital" vs "Search the residential area"
+   - "Trust the survivor" vs "Keep your distance"
    - "Use the old AI system" vs "Find a manual solution"
    
    Each choice must:
    - Be direct and brief
-   - Clearly show radiation risk when relevant
+   - Never mention radiation numbers
    - Feel meaningful
-   - After introduction: both should feel dangerous
+   - Present different risk levels
 
 {format_instructions}"""
 
@@ -90,7 +114,7 @@ Each response MUST contain:
 Current radiation level: {radiation_level}/10
 Previous choice: {previous_choice}
 
-Generate the next story segment and choices. If this is story_beat 0, create an atmospheric introduction that sets up the horror but doesn't kill Sarah. Otherwise, create a brutal and potentially lethal segment."""
+Generate the next story segment and choices. If this is story_beat 0, create an atmospheric introduction that sets up the horror but doesn't kill Sarah (radiation_increase MUST be 0). Otherwise, create a brutal and potentially lethal segment."""
 
         return ChatPromptTemplate(
             messages=[
@@ -100,27 +124,45 @@ Generate the next story segment and choices. If this is story_beat 0, create an
             partial_variables={"format_instructions": self.parser.get_format_instructions()}
         )
 
-    def generate_story_segment(self, game_state: GameState, previous_choice: str = "none") -> StorySegment:
-        # Get the formatted messages
+    def generate_story_segment(self, game_state: GameState, previous_choice: str) -> StorySegment:
         messages = self.prompt.format_messages(
             story_beat=game_state.story_beat,
             radiation_level=game_state.radiation_level,
             previous_choice=previous_choice
         )
-
-        # Get response from the model
+        
         response = self.chat_model.invoke(messages)
         
-        # Parse the response with retry mechanism
         try:
-            parsed_response = self.parser.parse(response.content)
-        except Exception as parsing_error:
-            print(f"First parsing attempt failed, trying to fix output: {str(parsing_error)}")
-            parsed_response = self.fixing_parser.parse(response.content)
-            
-        return parsed_response
+            segment = self.parser.parse(response.content)
+            # Force radiation_increase to 0 for the first story beat
+            if game_state.story_beat == 0:
+                segment.radiation_increase = 0
+            return segment
+        except Exception as e:
+            print(f"Error parsing response: {str(e)}")
+            print("Attempting to fix output...")
+            segment = self.fixing_parser.parse(response.content)
+            # Force radiation_increase to 0 for the first story beat
+            if game_state.story_beat == 0:
+                segment.radiation_increase = 0
+            return segment
+
+    async def transform_story_to_art_prompt(self, story_text: str) -> str:
+        try:
+            messages = [
+                SystemMessage(content=SYSTEM_ART_PROMPT),
+                HumanMessage(content=HUMAN_ART_PROMPT.format(story_text=story_text))
+            ]
+
+            response = self.chat_model.invoke(messages)
+            return response.content
+
+        except Exception as e:
+            print(f"Error transforming prompt: {str(e)}")
+            return story_text
 
     def process_radiation_death(self, segment: StorySegment) -> StorySegment:
         segment.is_death = True
-        segment.story_text += "\n\nFINAL RADIATION DEATH: Sarah's body finally gives in to the overwhelming radiation. Her cells break down as she collapses, mind filled with regret about her sister. The medical supplies she carried will never reach their destination. Her mission ends here, another victim of the wasteland's invisible killer."
+        segment.story_text += "\n\nThe end... ?"
         return segment 

server/server.py

@@ -8,6 +8,7 @@ from dotenv import load_dotenv
 import requests
 import base64
 import time
+import random
 
 # Choose import based on environment
 if os.getenv("DOCKER_ENV"):
@@ -60,6 +61,7 @@ class StoryResponse(BaseModel):
     story_text: str
     choices: List[Choice]
     is_death: bool = False
+    is_victory: bool = False
     radiation_level: int
 
 class ChatMessage(BaseModel):
@@ -111,17 +113,37 @@ async def chat_endpoint(chat_message: ChatMessage):
         print("Updated radiation level:", game_state.radiation_level)
         
         # Check for radiation death
-        if game_state.radiation_level >= MAX_RADIATION:
-            story_segment = story_generator.process_radiation_death(story_segment)
-            print("Processed radiation death")
+        is_death = game_state.radiation_level >= MAX_RADIATION
+        if is_death:
+            story_segment.story_text += f"""
+
+MORT PAR RADIATION: Le corps de Sarah ne peut plus supporter ce niveau de radiation ({game_state.radiation_level}/10). 
+Ses cellules se désagrègent alors qu'elle s'effondre, l'esprit rempli de regrets concernant sa sœur. 
+Les fournitures médicales qu'elle transportait n'atteindront jamais leur destination. 
+Sa mission s'arrête ici, une autre victime du tueur invisible des terres désolées."""
+            story_segment.choices = []
         
-        # Only increment story beat if not dead
-        if not story_segment.is_death:
+        # Check for victory condition
+        if not is_death and game_state.story_beat >= 5:
+            # Chance de victoire augmente avec le nombre de steps
+            victory_chance = (game_state.story_beat - 4) * 0.2  # 20% de chance par step après le 5ème
+            if random.random() < victory_chance:
+                story_segment.is_victory = True
+                story_segment.story_text = f"""Sarah l'a fait ! Elle a trouvé un bunker sécurisé avec des survivants. 
+                À l'intérieur, elle découvre une communauté organisée qui a réussi à maintenir un semblant de civilisation. 
+                Ils ont même un système de décontamination ! Son niveau de radiation : {game_state.radiation_level}/10.
+                Elle peut enfin se reposer et peut-être un jour, reconstruire un monde meilleur.
+                
+                VICTOIRE !"""
+                story_segment.choices = []
+        
+        # Only increment story beat if not dead and not victory
+        if not is_death and not story_segment.is_victory:
             game_state.story_beat += 1
             print("Incremented story beat to:", game_state.story_beat)
 
         # Convert to response format
-        choices = [] if story_segment.is_death else [
+        choices = [] if is_death or story_segment.is_victory else [
             Choice(id=i, text=choice.strip())
             for i, choice in enumerate(story_segment.choices, 1)
         ]
@@ -129,7 +151,8 @@ async def chat_endpoint(chat_message: ChatMessage):
         response = StoryResponse(
             story_text=story_segment.story_text,
             choices=choices,
-            is_death=story_segment.is_death,
+            is_death=is_death,
+            is_victory=story_segment.is_victory,
             radiation_level=game_state.radiation_level
         )
         print("Sending response:", response)
@@ -141,40 +164,6 @@ async def chat_endpoint(chat_message: ChatMessage):
         print("Traceback:", traceback.format_exc())
         raise HTTPException(status_code=500, detail=str(e))
 
-async def transform_story_to_art_prompt(story_text: str) -> str:
-    try:
-        from langchain_mistralai.chat_models import ChatMistralAI
-        from langchain.schema import HumanMessage, SystemMessage
-
-        chat = ChatMistralAI(
-            api_key=mistral_api_key,
-            model="mistral-small"
-        )
-
-        messages = [
-            SystemMessage(content="""Tu es un expert en prompts pour la génération d'images. 
-            Transforme l'histoire en un prompt court et précis.
-            
-            Format strict:
-            "color comic panel, style of Hergé, [scène principale en 5-7 mots], french comic panel"
-            
-            Exemple:
-            "color comic panel, style of Hergé, detective running through dark alley, french comic panel"
-            
-            Règles:
-            - Maximum 20 mots pour décrire la scène
-            - Pas d'adjectifs superflus
-            - Capture l'action principale uniquement"""),
-            HumanMessage(content=f"Transforme en prompt court: {story_text}")
-        ]
-
-        response = chat.invoke(messages)
-        return response.content
-
-    except Exception as e:
-        print(f"Error transforming prompt: {str(e)}")
-        return story_text
-
 @app.post("/api/generate-image", response_model=ImageGenerationResponse)
 async def generate_image(request: ImageGenerationRequest):
     try:
@@ -184,12 +173,12 @@ async def generate_image(request: ImageGenerationRequest):
                 error="HF_API_KEY is not configured in .env file"
             )
 
-        # Transformer le prompt en prompt artistique
+        # Transform the prompt into an artistic prompt
         original_prompt = request.prompt
-        # Enlever le préfixe pour la transformation
+        # Remove prefix for transformation
         story_text = original_prompt.replace("moebius style scene: ", "").strip()
-        art_prompt = await transform_story_to_art_prompt(story_text)
-        # Réappliquer le préfixe
+        art_prompt = await story_generator.transform_story_to_art_prompt(story_text)
+        # Reapply prefix
         final_prompt = f"moebius style scene: {art_prompt}"
         print("Original prompt:", original_prompt)
         print("Transformed art prompt:", final_prompt)