added validation functions, mapped advanced phases

src/game.py

@@ -1,49 +1,67 @@
-from game_utils import fetch_prompt, random_animal, random_names
+from game_utils import fetch_prompt, random_animal, random_names, random_index
 from player import Player
 
+# Default Values
+NUMBER_OF_PLAYERS = 5
 
 class Game:
-    def __init__(self):
-        print("Game Created")
+    def __init__(self,
+            human_name: str = None,
+            number_of_players: int = NUMBER_OF_PLAYERS
+        ):
+
+        # Gather Player Names
+        if human_name:
+            ai_names = random_names(number_of_players - 1)
+            self.human_index = random_index(number_of_players)
+        else:
+            ai_names = random_names(number_of_players)
+
+        # Choose Chameleon
+        self.chameleon_index = random_index(number_of_players)
+
+        # Add Players
         self.players = []
+        for i in range(0, number_of_players):
+            if self.human_index == i:
+                name = human_name
+                controller = "human"
+            else:
+                name = ai_names.pop()
+                controller = "ai"
+
+            if self.chameleon_index == i:
+                role = "chameleon"
+            else:
+                role = "herd"
+
+            self.players.append(Player(name, controller, role))
+
+
+        self.player_responses = []
+
+        print("Game Created")
 
-    def add_players(self, players: list[Player]):
-        """Adds players to the game."""
-        self.players = players
 
     def broadcast(self, message):
         """Sends a message to all the players, no response required."""
-        for player in self.players:
-            player.add_message(message)
+        for player_index in range(0, len(self.players)):
+            self.players[player_index].add_message(message)
+            if self.human_index == player_index:
+                print(message)
 
     def format_responses(self) -> str:
         """Formats the responses of the players into a single string."""
-        return "\n".join(self.player_responses)
+        return "\n".join([f" - {response['sender']}: {response['response']}" for response in self.player_responses])
+
+    def get_player_names(self) -> list[str]:
+        """Returns the names of the players."""
+        return [player.name for player in self.players]
 
     def start(self):
         """Starts the game."""
         print("Welcome to Chameleon! This is a social deduction game powered by LLMs.")
 
-        if not self.players:
-            print("Enter your name to begin")
-            human_name = input()
-
-            player_names = random_names(4)
-
-
-            for name in player_names:
-                self.add_players([
-                    Player(name, "ai", "herd")
-                ])
-
-            self.add_players([
-                Player("Player 1", "ai", "herd"),
-                Player("Player 2", "ai", "herd"),
-                Player(human_name, "human", "chameleon"),
-                Player("Player 4", "ai", "herd"),
-                Player("Player 5", "ai", "herd")
-            ])
-
         self.player_responses = []
         herd_animal = random_animal()
 
@@ -59,17 +77,47 @@ class Game:
                     prompt = prompt_template.format(player_responses=self.format_responses())
 
             response = player.collect_input(prompt)
-            self.player_responses.append(response)
+            self.player_responses.append({"sender": player.name, "response": response})
 
         self.player_votes = []
 
-        # Collect Player Votes
-        for player in self.players:
-            prompt_template = fetch_prompt("vote")
-            prompt = prompt_template.format(players=self.players)
-
-            response = player.collect_input(prompt)
-            self.player_responses.append(response)
-
-
-
+        # Show All Player Responses
+        self.broadcast(self.format_responses())
+
+        # Chameleon Decides if they want to guess the animal
+        # TODO: Add Chameleon Guess Decision Logic
+        chameleon_will_guess = False
+
+        if chameleon_will_guess:
+            # Chameleon Guesses Animal
+            # TODO: Add Chameleon Guessing Logic
+            pass
+        else:
+            # All Players Vote for Chameleon
+            for player in self.players:
+                prompt_template = fetch_prompt("vote")
+                prompt = prompt_template.format(players=self.get_player_names())
+
+                response = player.collect_input(prompt)
+                self.player_responses.append(response)
+
+        # Assign Points
+        # Chameleon Wins - 3 Points
+        # Herd Wins by Failed Chameleon Guess - 1 Point (each)
+        # Herd Wins by Correctly Guessing Chameleon - 2 points (each)
+
+    @staticmethod
+    def validate_animal_description(self, description: str) -> bool:
+        """Validates that the description starts with I and is less than 10 words."""
+        if not description.startswith("I"):
+            return False
+
+        if len(description.split(" ")) > 10:
+            return False
+
+        return True
+
+    def validate_vote(self, vote: str) -> bool:
+        """Validates that the vote is for a valid player."""
+        player_names = [player.name.lower() for player in self.players]
+        return vote.lower() in player_names

src/main.py

@@ -2,7 +2,11 @@ from game import Game
 from player import Player
 
 def main():
-    game = Game()
+    print("Please Enter your name:")
+    name = input()
+
+    game = Game(human_name=name)
+
     game.start()