Update app.py

app.py

@@ -62,9 +62,9 @@ def getChooseClass(msg):
     else:
         return ""
 
-def chatbot(input, localvar):    
+def chatbot(input, race, cls, localvar):    
     #messages = messages or [{"role": "system", "content": initMsg}]
-    localvar = localvar or {"messages": [{"role": "system", "content": initMsg}], "race": "", "class": ""}
+    localvar = localvar or {"messages": [{"role": "system", "content": "I choose as "+ race + " "+ cls +" in D&D world.\n" + initMsg}], "race": race, "class": cls}
     messages = localvar['messages']
     #if len(messages) == 1:
         
@@ -84,16 +84,7 @@ def chatbot(input, localvar):
         )
         reply = chat.choices[0].message.content
         logging.info("AI Reply:"+reply)
-        ans = ""
-        if len(messages) == 2:
-            localvar['race'] = localRace(getChooseRole(reply))
-            ans = "I choose "+ localvar['race'] +" as my race." + classAsk
-            logging.info("ChooseRole:"+ans)
-        if len(messages) == 4:
-            localvar['class'] = getChooseClass(reply)
-            ans = "My Race is "+ localvar['race'] +"，I choose "+ localvar['class'] +" as my class." + classAns
-            logging.info("ChooseClass:"+ans)
-        messages.append({"role": "assistant", "content": ans})
+        messages.append({"role": "assistant", "content": reply})
         localvar['messages'] = messages
         return reply, printMessages(messages), localvar
 
@@ -104,6 +95,8 @@ 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=roleAns), gr.Textbox(label="History"), "state"], title="DND Game",#, gr.Textbox(label="History"), 
+                   outputs=[gr.Dropdown(["Orc", "Human", "Elf", "Dwarf", "Halfling", "Goliath", "Dragonborn"], label="Race", info="please choose your race:", 
+                                        gr.Dropdown(["Warrior", "Mage", "Priest", "Assassin", "Thief", "Paladin", "Archer", "Guardian"], label="Class", info="please choose the name of the class you wish to play as:",
+                                        gr.Textbox(label="DND Game Reply", placeholder=roleAns), gr.Textbox(label="History"), "state"], title="DND Game",#, gr.Textbox(label="History"), 
                    description="DND Game",theme="compact")
 app.launch(share=False)