Update crs_arena/arena.py

crs_arena/arena.py

@@ -161,7 +161,7 @@ def end_conversation(crs: CRSFighter, sentiment: str) -> None:
     st.rerun()
 
 
-def get_crs_response(crs: CRSFighter, message: str):
+def get_crs_response(crs: CRSFighter, message: str) -> str:
     """Gets the CRS response for the given message.
 
     This method sends a POST request to the CRS model including the history of
@@ -171,8 +171,8 @@ def get_crs_response(crs: CRSFighter, message: str):
         crs: CRS model.
         message: User's message.
 
-    Yields:
-        Words from the CRS response.
+    Returns:
+        CRS response.
     """
     history = deepcopy(st.session_state[f"messages_{crs.fighter_id}"])
     history = history[:-1]  # Remove the last message (user's message)
@@ -183,11 +183,10 @@ def get_crs_response(crs: CRSFighter, message: str):
         options_state=st.session_state.get(f"state_{crs.fighter_id}", []),
     )
     st.session_state[f"state_{crs.fighter_id}"] = state
-    # response = "CRS response for testing purposes."
-    for word in response.split():
-        yield f"{word} "
-        time.sleep(0.05)
-
+    # for word in response.split():
+    #    yield f"{word} "
+    #    time.sleep(0.05)
+    return response
 
 @st.dialog("Your vote has been submitted!")
 def feedback_dialog(row_id: int) -> None: