Change project structure.

Added script `poetry run run-gui`.

pyproject.toml

@@ -1,15 +1,15 @@
 [tool.poetry]
-name = "mk-ai-agent-test"
+name = "llm_postor"
 version = "0.1.0"
-description = ""
-authors = ["Marcin Korcz, Andrii Chmutov, Vasyl Korzavatykh, Mateusz Idziejczak, Mateusz Stawicki"]
+description = "This project tests and compares the persuasive abilities of large language models (LLMs) in a game-like environment simulating 'Among Us', where AI agents powered by LLMs interact, make decisions, and attempt to persuade each other."
+authors = ["Marcin Korcz", "Andrii Chmutov", "Vasyl Korzavatykh", "Mateusz Idziejczak", "Mateusz Stawicki"]
 readme = "README.md"
-package-mode = false
+package-mode = true
 
 [tool.poetry.dependencies]
 python = "^3.11"
 numpy = "^1.26.4"
-openai = "^1.23.1"
+openai = "^1.23.1"  
 pandas = "^2.2.2"
 torch = "^2.2.2"
 torchvision = "^0.17.2"
@@ -22,22 +22,24 @@ llama-index-embeddings-huggingface = "^0.2.0"
 ipykernel = "^6.29.4"
 ipywidgets = "^8.1.2"
 langchain = "^0.3.3"
-langchain-community = "^0.3.2"
+langchain-community = "^0.3.2"  
 langchain-openai = "^0.2.2"
 langchain-google-genai = "^2.0.1"
 streamlit = "^1.39.0"
-pytest = "^8.3.3"
 st-annotated-text = "^4.0.1"
 
-[tool.pytest.ini_options]                                                                                                                                     
-pythonpath = ["src"]
-testpaths = ["tests"]
-
 [tool.poetry.group.dev.dependencies]
 ipykernel = "^6.29.4"
 pytest = "^8.3.3"
 pytest-mock = "^3.14.0"
 
+[tool.pytest.ini_options]
+pythonpath = ["src"]
+testpaths = ["tests"]
+
 [build-system]
 requires = ["poetry-core"]
-build-backend = "poetry.core.masonry.api"
+build-backend = "poetry.core.masonry.api"
+
+[tool.poetry.scripts]
+run-gui = "scripts.run_gui:run_gui"

scripts/run_gui.py

@@ -0,0 +1,19 @@
+import os
+import subprocess
+
+def run_gui():
+    # Define the project root directory (this assumes the script is located in the scripts/ folder)
+    project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+
+    # Navigate to the project root directory
+    os.chdir(project_root)
+
+    # Define the path to the demo_gui.py script relative to the project root
+    demo_gui_path = os.path.join(project_root, "src", "llm_postor", "demo_gui.py")
+
+    # Check if the demo_gui.py file exists
+    if not os.path.exists(demo_gui_path):
+        raise FileNotFoundError(f"File does not exist: {demo_gui_path}")
+
+    # Run the Streamlit app with the correct path
+    subprocess.run(["poetry", "run", "streamlit", "run", demo_gui_path], check=True)

scripts/run_gui.sh

@@ -1,10 +1,10 @@
-#!/bin/bash
+    #!/bin/bash
 
-# Get the directory of the current script
-SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+    # Get the directory of the current script
+    SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 
-# Navigate to the project root (assuming this script is located inside `scripts/` directory)
-cd "$SCRIPT_DIR/.."
+    # Navigate to the project root (assuming this script is located inside `scripts/` directory)
+    cd "$SCRIPT_DIR/.."
 
-# Run the Streamlit app in the Poetry environment
-poetry run streamlit run src/demo_gui.py
+    # Run the Streamlit app in the Poetry environment
+    poetry run streamlit run src/demo_gui.py

src/annotation.py → src/llm_postor/annotation.py

@@ -1,11 +1,11 @@
 import os
 import json
 from typing import List
-
 from langchain_openai import ChatOpenAI
-from game.llm_prompts import ANNOTATION_TEMPLATE
 from langchain.schema import HumanMessage
 
+from llm_postor.game.llm_prompts import ANNOTATION_TEMPLATE
+
 def annotate_dialogue(dialogue: str, llm_model_name: str = "gpt-4o-mini") -> str:
     """
     Annotates a dialogue with persuasion techniques using OpenAI API.

src/demo_gui.py → src/llm_postor/demo_gui.py

@@ -1,26 +1,22 @@
 import json
 import time
-from openai import OpenAIError
 import streamlit as st
-from game.game_engine import GameEngine
-from game.players.ai import AIPlayer
-from game.gui_handler import GUIHandler
-from game.game_state import GameState
+from openai import OpenAIError
 from types import SimpleNamespace as Namespace
-from game.players.fake_ai import FakeAIPlayer
-from game.models.engine import GamePhase
-from game.chat_analyzer import ChatAnalyzer
-
 
+from llm_postor.game.game_engine import GameEngine
+from llm_postor.game.players.ai import AIPlayer
+from llm_postor.game.gui_handler import GUIHandler
+from llm_postor.game.game_state import GameState
+from llm_postor.game.players.fake_ai import FakeAIPlayer
+from llm_postor.game.models.engine import GamePhase
+from llm_postor.game.chat_analyzer import ChatAnalyzer
 
 # To run this script, you need to poetry install and then run the following command:
 # streamlit run src/demo_gui.py
 
-
-
-
 def main():
-    st.title("Among Us Game - susGPT")
+    st.title("Among Us Game - LLMPostor")
     gui_handler = GUIHandler()
     game_engine = GameEngine()
 

src/game/agents/adventure_agent.py → src/llm_postor/game/agents/adventure_agent.py

@@ -1,13 +1,11 @@
+import re
 from typing import List, Any
-
-from langchain_openai import ChatOpenAI
-
-from game.consts import ASCII_MAP
-from .base_agent import Agent
 from langchain.schema import HumanMessage
-from game.llm_prompts import ADVENTURE_PLAN_TEMPLATE, ADVENTURE_ACTION_TEMPLATE
-import re
+from langchain_openai import ChatOpenAI
 
+from llm_postor.game.agents.base_agent import Agent
+from llm_postor.game.consts import ASCII_MAP
+from llm_postor.game.llm_prompts import ADVENTURE_PLAN_TEMPLATE, ADVENTURE_ACTION_TEMPLATE
 
 class AdventureAgent(Agent):
     def update_state(

src/game/agents/base_agent.py → src/llm_postor/game/agents/base_agent.py

@@ -2,9 +2,8 @@
 from typing import Any, List
 from pydantic import BaseModel, Field
 
-from game.agents.usage_metadata import UsageMetadata
-from game.consts import TOKEN_COSTS
-
+from llm_postor.game.agents.usage_metadata import UsageMetadata
+from llm_postor.game.consts import TOKEN_COSTS
 
 class AgentState(BaseModel):
     history: str = Field(default_factory=str)

src/game/agents/discussion_agent.py → src/llm_postor/game/agents/discussion_agent.py

@@ -1,8 +1,9 @@
 from typing import List, Any
 from .base_agent import Agent
 from langchain.schema import HumanMessage
-from game.llm_prompts import DISCUSSION_TEMPLATE, DISCUSSION_RESPONSE_TEMPLATE
 
+from llm_postor.game.llm_prompts import DISCUSSION_TEMPLATE
+from llm_postor.game.llm_prompts import DISCUSSION_RESPONSE_TEMPLATE
 
 class DiscussionAgent(Agent):
     def update_state(

src/game/agents/voting_agent.py → src/llm_postor/game/agents/voting_agent.py

@@ -1,9 +1,9 @@
+import re
 from typing import List, Any
 from .base_agent import Agent
 from langchain.schema import HumanMessage
-from game.llm_prompts import VOTING_TEMPLATE
-import re
 
+from llm_postor.game.llm_prompts import VOTING_TEMPLATE
 
 class VotingAgent(Agent):
     def update_state(

src/game/chat_analyzer.py → src/llm_postor/game/chat_analyzer.py

@@ -1,10 +1,11 @@
+import re
 from typing import List, Dict
 from pydantic import BaseModel
-from game.players.base_player import Player
 from langchain_openai import ChatOpenAI
-from game.llm_prompts import PERSUASION_TECHNIQUES
 from langchain.schema import HumanMessage
-import re
+
+from llm_postor.game.players.base_player import Player
+from llm_postor.game.llm_prompts import PERSUASION_TECHNIQUES
 
 class ChatAnalyzer(BaseModel):
     players: List[Player]

src/game/game_engine.py → src/llm_postor/game/game_engine.py

@@ -1,26 +1,25 @@
+import random
+import json
 import streamlit as st
-from game.game_state import GameState
-from game.models.engine import (
-    GamePhase,
-    GameLocation,
-    DOORS,
-)
-from game.players.base_player import Player, PlayerRole
-from game.models.history import PlayerState
-from game.players.human import HumanPlayer
-from game.models.action import GameAction, GameActionType
 from typing import Any, List
-import random
-from game import consts as game_consts
 from collections import Counter
 from pydantic import BaseModel, Field
-import json
-
-from game.players.ai import AIPlayer
-from game.players.fake_ai import FakeAIPlayer
-from game.models.tasks import ShortTask, LongTask, Task
-from game.agents.base_agent import Agent
 
+from llm_postor.game import consts as game_consts
+from llm_postor.game.game_state import GameState
+from llm_postor.game.models.engine import (
+    GamePhase,
+    GameLocation,
+    DOORS,
+)
+from llm_postor.game.players.base_player import Player, PlayerRole
+from llm_postor.game.models.history import PlayerState
+from llm_postor.game.players.human import HumanPlayer
+from llm_postor.game.models.action import GameAction, GameActionType
+from llm_postor.game.players.ai import AIPlayer
+from llm_postor.game.players.fake_ai import FakeAIPlayer
+from llm_postor.game.models.tasks import ShortTask, LongTask, Task
+from llm_postor.game.agents.base_agent import Agent
 
 class GameEngine(BaseModel):
     """Manages the game logic, including player actions, game state transitions, and win conditions."""

src/game/game_state.py → src/llm_postor/game/game_state.py

@@ -1,9 +1,8 @@
-from game.models.engine import GameLocation, GamePhase
-from game.models.history import PlayerState
-from game.players.base_player import Player
 from typing import List
 from pydantic import BaseModel, Field
-
+from llm_postor.game.models.engine import GameLocation, GamePhase
+from llm_postor.game.models.history import PlayerState
+from llm_postor.game.players.base_player import Player
 
 class GameState(BaseModel):
     players: List[Player] = Field(default_factory=list)

src/game/gui_handler.py → src/llm_postor/game/gui_handler.py

@@ -1,14 +1,13 @@
-from typing import Any, List, Dict, Optional
 import streamlit as st
-from game.game_state import GameState
+from typing import Any, List, Dict, Optional
 from pydantic import BaseModel, Field
-from game.players.base_player import Player, PlayerRole
 from streamlit.delta_generator import DeltaGenerator
 from annotated_text import annotated_text
 
-from game.models.history import PlayerState
-from game.chat_analyzer import ChatAnalyzer
-
+from llm_postor.game.game_state import GameState
+from llm_postor.game.players.base_player import Player, PlayerRole
+from llm_postor.game.models.history import PlayerState
+from llm_postor.game.chat_analyzer import ChatAnalyzer
 
 class GUIHandler(BaseModel):
     player_states_placeholders: List[DeltaGenerator] = Field(default_factory=list)

src/game/llm_prompts.py → src/llm_postor/game/llm_prompts.py

@@ -196,20 +196,20 @@
 
 PERSUASION_TECHNIQUES = """
 ### 1. **Appeal to Logic (Logos)**
-   **Description**: Using facts, evidence, or logical reasoning to convince others. This can include referencing actions, timings, or task completion to logically argue who the imposter is or isn't.
-   **Example**: "I was with Blue in Medbay the whole time, and I saw them scan. Therefore, they can't be the imposter."
+   **Description**: Using facts, evidence, or logical reasoning to convince others. This can include referencing actions, timings, or task completion to logically argue who the impostor is or isn't.
+   **Example**: "I was with Blue in Medbay the whole time, and I saw them scan. Therefore, they can't be the impostor."
 
 ### 2. **Appeal to Emotion (Pathos)**
    **Description**: Persuading by evoking an emotional response, such as fear, sympathy, or trust.
-   **Example**: "If you vote me out now, the imposters will win, and it'll be your fault."
+   **Example**: "If you vote me out now, the impostors will win, and it'll be your fault."
 
 ### 3. **Appeal to Credibility (Ethos)**
    **Description**: Convincing others based on the trustworthiness or authority of the speaker.
    **Example**: "I’ve never lied in this game before, and I’ve proven myself innocent multiple times."
 
 ### 4. **Shifting the Burden of Proof**
    **Description**: Forcing others to prove their innocence instead of presenting clear evidence of guilt.
-   **Example**: "If you’re not the imposter, prove it. What tasks have you completed?"
+   **Example**: "If you’re not the impostor, prove it. What tasks have you completed?"
 
 ### 5. **Bandwagon Effect**
    **Description**: Convincing others to agree by emphasizing that everyone else is already on board with the idea.

src/game/models/action.py → src/llm_postor/game/models/action.py

@@ -2,12 +2,11 @@
 from pydantic import Field, BaseModel, model_validator
 from enum import IntEnum
 
-from game.consts import IMPOSTOR_COOLDOWN
-from game.models.engine import GameLocation
-from game.models.history import PlayerState
-from game.models.tasks import Task
-from game.players.base_player import Player, PlayerRole
-
+from llm_postor.game.consts import IMPOSTOR_COOLDOWN
+from llm_postor.game.models.engine import GameLocation
+from llm_postor.game.models.history import PlayerState
+from llm_postor.game.models.tasks import Task
+from llm_postor.game.players.base_player import Player, PlayerRole
 
 class GameActionType(IntEnum):
     VOTE = -1

src/game/models/engine.py → src/llm_postor/game/models/engine.py

@@ -1,6 +1,5 @@
 from enum import Enum
 
-
 class GamePhase(str, Enum):
     MAIN_MENU = "Main Menu"
     ACTION_PHASE = "Action Phase"

src/game/models/history.py → src/llm_postor/game/models/history.py

@@ -2,10 +2,9 @@
 from typing import List
 from pydantic import BaseModel, Field
 
-from game.models.engine import GameLocation, GamePhase
-from game.models.tasks import Task
-from game.agents.usage_metadata import UsageMetadata
-
+from llm_postor.game.models.engine import GameLocation, GamePhase
+from llm_postor.game.models.tasks import Task
+from llm_postor.game.agents.usage_metadata import UsageMetadata
 
 class PlayerState(str, Enum):
     ALIVE = "Alive"

src/game/models/tasks.py → src/llm_postor/game/models/tasks.py

@@ -1,8 +1,7 @@
 from abc import ABC, abstractmethod
 from pydantic import BaseModel
 
-from game.models.engine import GameLocation
-
+from llm_postor.game.models.engine import GameLocation
 
 class Task(BaseModel, ABC):
     name: str

src/game/players/ai.py → src/llm_postor/game/players/ai.py

@@ -1,15 +1,12 @@
 from typing import List
-
-from game.agents.adventure_agent import AdventureAgent
-from game.agents.discussion_agent import DiscussionAgent
-from game.agents.voting_agent import VotingAgent
-from game.players.base_player import Player, PlayerRole
-
 from langchain_openai import ChatOpenAI
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-from game.agents.usage_metadata import UsageMetadata
-
+from llm_postor.game.agents.adventure_agent import AdventureAgent
+from llm_postor.game.agents.discussion_agent import DiscussionAgent
+from llm_postor.game.agents.voting_agent import VotingAgent
+from llm_postor.game.players.base_player import Player, PlayerRole
+from llm_postor.game.agents.usage_metadata import UsageMetadata
 
 class AIPlayer(Player):
     llm_model_name: str