Update app.py

app.py

@@ -15,6 +15,10 @@ defaultMsg = os.environ['defaultMsg']
 
 mod = 3;
 
+def localVar(lv):
+    lv = lv or {"race":"", "cls":""}
+    return lv
+
 def trimMessages(messages):
     # Assuming messages is an array of objects with "role" and "content" properties
     content = {}
@@ -32,14 +36,22 @@ def trimMessages(messages):
 
 def getChooseRole(msg):
     start_index = msg.index("As a") + len("As a")
-    end_index = msg.index(",")
-    return msg[start_index:] + msg[:end_index]
+    if start_index >= 0:
+        end_index = msg.index(",")
+        str = trim(msg[start_index:] + msg[:end_index])
+        if str.lower() = "dm"
+            return ""
+        else:
+            return str
+    else:
+        return ""
 
 def getChooseClass(msg):
     full = getChooseRole(msg)
+    if 
     start_index = full.index(" ") + len(" ")
     end_index = msg.index(",")
-    return full[start_index:] + full[:end_index]
+    return trim(full[start_index:] + full[:end_index])
 
 def chatbot(input, messages):
     messages = messages or [{"role": "system", "content": initMsg}]
@@ -63,10 +75,16 @@ def chatbot(input, messages):
         logging.info("AI Reply:"+reply)
         ans = ""
         if len(messages) == 2:
-            ans = "I choose "+ getChooseRole(reply) +" as my race." + classAsk
+            lv.race = getChooseRole(reply)
+            lv.cls = ""
+            lv = localVar(lv)
+            ans = "I choose "+ lv.race +" as my race." + classAsk
             logging.info("ChooseRole:"+ans)
         if len(messages) == 4:
-            ans = "My Race is ，I choose "+ getChooseClass(reply) +" as my class." + classAns
+            lv = localVar(lv)
+            lv.cls = getChooseClass(reply)
+            lv = localVar(lv)
+            ans = "My Race is "+ lv.race +"，I choose "+ lv.cls +" as my class." + classAns
             logging.info("ChooseClass:"+ans)
         messages.append({"role": "assistant", "content": ans})
         return reply, printMessages(messages), messages