added user input to app (non-functional)

src/app.py

@@ -18,7 +18,22 @@ def display_message(message):
     elif message["type"] == "debug":
         messages_container.markdown(f":orange[DEBUG: {message['content']}]")
 
+
 class StreamlitGame(Game):
+    @staticmethod
+    async def human_input(prompt: str) -> str:
+        _user_input = st.chat_input("Your message", key=f"user_input_{st.session_state.user_input_id}")
+        st.session_state.user_input_id += 1
+
+        while _user_input is None or _user_input == "":
+            sleep(0.1)
+
+        print(f"User input: {_user_input}")
+
+        response = AIMessage(content=_user_input)
+
+        return response
+
     def human_message(self, message: str):
         message = {"type": "game", "content": message}
         st.session_state["messages"].append(message)
@@ -39,8 +54,10 @@ class StreamlitGame(Game):
 
 if "messages" not in st.session_state:
     st.session_state["messages"] = []
-if "game" not in st.session_state:
-    st.session_state["game"] = StreamlitGame(verbose=True)
+if "game_started" not in st.session_state:
+    st.session_state["game_started"] = False
+if "user_input_id" not in st.session_state:
+    st.session_state["user_input_id"] = 0
 
 margin_size = 1
 center_size = 3
@@ -53,17 +70,24 @@ with title_center:
 left, center, right = st.columns([margin_size, center_size, margin_size])
 
 with center:
-    st.write("Welcome to Chameleon! A deduction and deception game powered by LLMs.")
+    messages_container = st.container()
+
+    messages_container.write("Welcome to Chameleon! A social deduction game powered by LLMs.")
+
+    messages_container.write("Enter your name to begin...")
+
+    if st.session_state.messages:
+        for message in st.session_state.messages:
+            display_message(message)
 
-    start_button = st.button("Start Game")
+    user_input = st.chat_input("Your message")
+    st.session_state.user_input_id += 1
 
-with right:
-    st.markdown("### Player Scores")
-    for player in st.session_state["game"].players:
-        st.write(f"{player.name}: {player.points}")
+if not st.session_state.game_started and user_input:
+    st.session_state.game_started = True
+    if "game" not in st.session_state:
+        st.session_state.game = StreamlitGame(human_name=user_input, verbose=True)
 
+    asyncio.run(st.session_state.game.start())
 
-messages_container = center.container()
 
-if start_button:
-    asyncio.run(st.session_state["game"].start())

src/game.py

@@ -131,7 +131,7 @@ class Game:
     # The following methods are used to broadcast messages to a human.
     # They are design so that they can be overridden by a subclass for a different player interface.
     @staticmethod
-    def human_input(prompt: str) -> str:
+    async def human_input(prompt: str) -> AIMessage:
         """Gets input from the human player."""
         response = AIMessage(content=input())
         return response

src/player.py

@@ -138,22 +138,22 @@ class Player:
     def is_ai(self):
         return not self.is_human()
 
-    def _generate(self, message: Message):
+    async def _generate(self, message: Message):
         """Entry point for the Runnable generating responses, automatically logs the message."""
         self.add_to_history(message)
 
         # AI's need to be fed the whole message history, but humans can just go back and look at it
         if self.controller_type == "human":
-            response = self.controller.invoke(message.content)
+            response = await self.controller.ainvoke(message.content)
         else:
             formatted_messages = [(message.langchain_type, message.content) for message in self.messages]
-            response = self.controller.invoke(formatted_messages)
+            response = await self.controller.ainvoke(formatted_messages)
 
         self.add_to_history(Message(type="player", content=response.content))
 
         return response
 
-    def _output_formatter(self, inputs: dict):
+    async def _output_formatter(self, inputs: dict):
         """Formats the output of the response."""
         output_format: BaseModel = inputs["output_format"]
 
@@ -167,6 +167,6 @@ class Player:
 
         message = Message(type="player", content=prompt.text)
 
-        response = self.generate.invoke(message)
+        response = await self.generate.ainvoke(message)
 
-        return parser.invoke(response)
+        return await parser.ainvoke(response)