Update app.py

app.py

@@ -31,15 +31,15 @@ def trimMessages(messages):
     return messages
 
 def getChooseRole(msg):
-    start_index = msg.index("As an ") + len("As an ")
+    start_index = msg.index("As a") + len("As a")
     end_index = msg.index(",")
-    return msg[start_index:] + msg[:start_index]
+    return msg[start_index:] + msg[:end_index]
 
 def getChooseClass(msg):
     full = getChooseRole(msg)
     start_index = full.index(" ") + len(" ")
     end_index = msg.index(",")
-    return full[start_index:] + full[:start_index]
+    return full[start_index:] + full[:end_index]
 
 def chatbot(input, messages):
     messages = messages or [{"role": "system", "content": initMsg}]