Update app.py

app.py

@@ -62,9 +62,10 @@ def getChooseClass(msg):
     else:
         return ""
 
-def chatbot(input, messages, localvar):    
-    messages = messages or [{"role": "system", "content": initMsg}]
-    localvar = localvar or {"race": "", "class": ""}
+def chatbot(input, localvar):    
+    #messages = messages or [{"role": "system", "content": initMsg}]
+    localvar = localvar or {"messages": [{"role": "system", "content": initMsg}], "race": "", "class": ""}
+    messages = localvar['messages']
     #if len(messages) == 1:
         
         #memory = trimMessages(messages)
@@ -93,7 +94,8 @@ def chatbot(input, messages, localvar):
             ans = "My Race is "+ localvar['race'] +"，I choose "+ localvar['class'] +" as my class." + classAns
             logging.info("ChooseClass:"+ans)
         messages.append({"role": "assistant", "content": ans})
-        return reply, printMessages(messages), messages, localvar
+        localvar['messages'] = messages
+        return reply, printMessages(messages), localvar
 
 def printMessages(messages):
     delimiter = '\n'
@@ -101,7 +103,7 @@ def printMessages(messages):
     logging.info("messages:"+msg_string)
     return msg_string
 
-app = gr.Interface(fn=chatbot, inputs=[gr.Textbox(lines=7, label="You ask and answer questions below"), "state", "state"], 
-                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=roleAns), gr.Textbox(label="History"), "state", "state"], title="DND Game",#, gr.Textbox(label="History"), 
+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"), 
                    description="DND Game",theme="compact")
 app.launch(share=False)