strip

app.py

@@ -70,7 +70,7 @@ def main():
     role = target(question)
     max_steps = 10
     for step, _ in enumerate(range(max_steps), start=1):
-        with st.spinner(f"{step}/{max_steps} Asking {role}..."):
+        with st.spinner(f"({step}/{max_steps}) Asking {role}..."):
             actor = Actor[role]
             answer = ask(actor.model, actor.system_prompt, actor.pre_prompt, question)
             st.write(f":blue[{actor.role} says:] {answer}")
@@ -79,8 +79,7 @@ def main():
 
 
 def target(question):
-    return re.split(r'\s|,|:', question)[0]
-
+    return re.split(r'\s|,|:', question.strip())[0].strip()
 
 def sanitize(question):
     return re.sub(r"\([^)]*\)", "", question)

test_sanitize.py

@@ -1,4 +1,4 @@
-from app import sanitize
+from app import sanitize, target
 
 
 def test_sanitize():
@@ -10,3 +10,7 @@ def test_sanitize():
     assert 'qwertyui  poiuy' == sanitize('qwertyui (abcdef) poiuy')
 
     assert '"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" ' == sanitize('"Teacher, I heard about a fascinating story from someone who lives in the city. Can you guess who it might be?" (I want to find out if the Teacher can make any connections between my statement and their own information.)')
+
+def test_target():
+    assert 'Priest' == target("Priest, your task is to figure out their names and where they live. Do not ask directly, they must not realize what information you are after!")
+    assert 'Priest' == target(sanitize("(I'm thinking hmm) Priest, .."))