Update app.py

app.py

@@ -30,8 +30,19 @@ def trimMessages(messages):
         messages.append({"role": "assistant", "content": content})
     return messages
 
+def getChooseRole(msg):
+    start_index = msg.index("As an ") + len("As an ")
+    end_index = msg.index(",")
+    return msg[start_index:] + msg[:start_index]
+
+def getChooseClass(msg):
+    full = getChooseRole(msg)
+    start_index = full.index(" ") + len(" ")
+    end_index = msg.index(",")
+    return full[start_index:] + full[:start_index]
+
 def chatbot(input, messages):
-    messages = messages or [{"role": "system", "content": "i choose "+initMsg}]
+    messages = messages or [{"role": "system", "content": initMsg}]
     if len(messages) == 1:
         
         memory = trimMessages(messages)
@@ -49,7 +60,15 @@ def chatbot(input, messages):
             max_tokens=2048,n=1,temperature=0.5,
         )
         reply = chat.choices[0].message.content
-        messages.append({"role": "assistant", "content": reply})
+
+        ans = ""
+        if len(messages) == 3:
+            ans = "I choose "+ getChooseRole(reply) +" as my race." + classAsk
+            logging.info("ChooseRole:"+ans)
+        if len(messages) == 5:
+            ans = "My Race is ，I choose "+ getChooseClass(reply) +" as my class." + classAns
+            logging.info("ChooseClass:"+ans)
+        messages.append({"role": "assistant", "content": ans})
         return reply, messages
 
 def printMessages(messages):
@@ -59,6 +78,6 @@ def printMessages(messages):
     return msg_string
 
 app = gr.Interface(fn=chatbot, inputs=[gr.Textbox(lines=7, label="You ask and answer questions below"), "state"], 
-                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=role), "state"], title="DND Game",#, gr.Textbox(label="History"), 
+                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=roleAns), "state"], title="DND Game",#, gr.Textbox(label="History"), 
                    description="DND Game",theme="compact")
 app.launch(share=False)