Spaces:
Runtime error
Runtime error
File size: 1,306 Bytes
28a47b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
from typing import List
from rich.console import Console
from rich.prompt import Prompt
from setting import Settings
from agent import SuspicionAgent
class Context:
def __init__(self, console: Console, settings: Settings, webcontext=None) -> None:
self.clock: int = 0
self.console: Console = console
self.agents: List[SuspicionAgent] = []
self.user_agent: SuspicionAgent = None
self.robot_agents: List[SuspicionAgent] = []
self.observations = ["Beginning of the day, people are living their lives."]
self.timewindow_size = 3
self.observations_size_history = []
self.settings = settings
self.webcontext = webcontext
def print(self, message: str, style: str = None):
if style:
self.console.print(message, style=style)
else:
self.console.print(message)
if self.webcontext:
self.webcontext.send_response(message)
def ask(self, message: str = "", choices: List[str] = None) -> str:
if self.webcontext:
return self.webcontext.ask_human(message, choices)
else:
if choices:
return Prompt.ask(message, choices=choices, default=choices[0])
else:
return Prompt.ask(message)
|