Update app.py

app.py

@@ -62,9 +62,9 @@ def getChooseClass(msg):
     else:
         return ""
 
-def chatbot(input, race, cls, localvar):    
+def chatbot(race, cls, input, localvar):    
     #messages = messages or [{"role": "system", "content": initMsg}]
-    localvar = localvar or {"messages": [{"role": "system", "content": "I choose as "+ race + " "+ cls +" in D&D world.\n" + initMsg}], "race": race, "class": cls}
+    localvar = localvar or {"messages": [{"role": "system", "content": "I choose as "+ race + " "+ cls +" in D&D world.\n" + initMsg}], "race": race, "class": cls, "status":""}
     messages = localvar['messages']
     #if len(messages) == 1:
         
@@ -77,13 +77,18 @@ def chatbot(input, race, cls, localvar):
         #logging.info("put memory return:"+chat.choices[0].message.content)
 
     if input:
-        messages.append({"role": "user", "content": input})
+        if len(messages) == 1:
+            messages.append({"role": "user", "content": input})
+        else:
+            messages.append({"role": "user", "content": defaultMsg + localvar['status'] + ". \n" + input})
         chat = openai.ChatCompletion.create(
             model="gpt-3.5-turbo", messages=messages[-2:],
             max_tokens=2048,n=1,temperature=0.5,
         )
         reply = chat.choices[0].message.content
         logging.info("AI Reply:"+reply)
+        if len(messages) == 1:
+            localvar['status'] = reply
         messages.append({"role": "assistant", "content": reply})
         localvar['messages'] = messages
         return reply, printMessages(messages), localvar
@@ -96,7 +101,7 @@ def printMessages(messages):
 
 app = gr.Interface(fn=chatbot, inputs=[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(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"), 
+                                       gr.Textbox(lines=7, label="You ask and answer questions below", placeholder="Type word "start game" to play"), "state"], 
+                   outputs=[gr.Textbox(label="DND Game Reply", placeholder="Waiting for you start..."), gr.Textbox(label="History"), "state"], title="DND Game",#, gr.Textbox(label="History"), 
                    description="DND Game",theme="compact")
 app.launch(share=False)