adding game elements

src/game.py

@@ -0,0 +1,67 @@
+from prompts import fetch_prompt
+from player import Player
+
+
+class Game:
+    def __init__(self):
+        print("Game Created")
+        self.players = []
+
+    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)
+
+    def format_responses(self, responses: list[str]) -> str:
+        """Formats the responses of the players into a single string."""
+        return "\n".join(responses)
+
+    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()
+
+            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 = "Giraffe"
+
+        # Collect Player Animal Descriptions
+        for player in self.players:
+            match player.role:
+                case "herd":
+                    prompt_template = fetch_prompt("herd_animal")
+                    prompt = prompt_template.format(animal=herd_animal, player_responses=self.player_responses)
+
+                case "chameleon":
+                    prompt_template = fetch_prompt("chameleon_animal")
+                    prompt = prompt_template.format(player_responses="")
+
+            response = player.collect_input(prompt)
+            self.player_responses.append(response)
+
+        self.player_votes = []
+
+        # Collect Player Votes
+        for player in self.players:
+            prompt_template = fetch_prompt("vote")
+            prompt = prompt_template.format(animal=herd_animal, player_responses=self.player_responses)
+
+            response = player.collect_input(prompt)
+            self.player_responses.append(response)
+
+
+

src/player.py

@@ -0,0 +1,32 @@
+from prompts import fetch_prompt
+
+class Player:
+    def __init__(self, name: str, controller: str, role: str):
+        self.name = name
+        self.controller = controller
+        self.role = role
+        self.messages = []
+
+    def collect_input(self, prompt: str) -> str:
+        """Store the input and output in the messages list. Return the output."""
+        self.messages.append(prompt)
+        output = self.respond(prompt)
+        self.messages.append(output)
+        return output
+
+    def respond(self, prompt: str) -> str:
+        if self.controller == "human":
+            print(prompt)
+            return input()
+
+        elif self.controller == "ai":
+            return "I am an AI and I am responding to the prompt."
+
+    def add_message(self, message: str):
+        """Add a message to the messages list. No response required."""
+        self.messages.append(message)
+
+
+
+
+

src/prompts.py

@@ -0,0 +1,55 @@
+"""
+Prompts used for the game
+"""
+
+
+def fetch_prompt(prompt_name):
+    return prompts[prompt_name]
+
+
+game_rules = '''\
+GAME RULES: You are playing a social deduction game where every player pretends the be the same animal. 
+During the round players go around the room and make an "I"-statement as if they were the animal. 
+All players know what animal they are pretending to be, except one who is known as the Chameleon. The Chameleon and must blend in by providing details about the animal using context from other players. 
+The other players must be careful not to give away too much information with their responses so that Chameleon cannot guess the animal. After all players have spoken, they vote on who they think the Chameleon is. 
+WIN CONDITION:
+- If the other players vote for the Chameleon, they win the game. 
+- If they vote for the wrong player then the Chameleon wins the game. 
+
+'''
+
+herd_animal = """\
+You are a {animal}. In 10 words or less give a description of yourself starting with "I". The description should not give away too much information about the {animal} as you do not want the Chameleon to be able to guess what animal you are. Do not repeat responses from other players.
+Previously Mentioned Descriptions: 
+{player_responses}
+Your Response:
+"""
+
+chameleon_animal = """\
+You are the Chameleon. Initially, you don't know what animal the other players are, your goal is to deduce . Using the context that they have provided, In 10 words or less give a description of yourself starting with "I". If no one else has said anything try to say something generic that could be true of any animals. 
+Previously Mentioned Descriptions:
+{player_responses}
+Your Response:
+"""
+
+
+vote_prompt = """\
+The players have spoken, now it is time to vote. Choose from the players below who you think the Chameleon is.
+{players}
+"""
+
+herd_vote = """\
+Who do you think the Chameleon is?
+"""
+
+chameleon_prompt = """\
+You are the Chameleon, vote for someone else to remove suspicion from yourself.
+"""
+
+
+prompts = {
+    "herd_animal": game_rules + herd_animal,
+    "chameleon_animal": game_rules + chameleon_animal,
+    "vote_prompt": vote_prompt
+}
+