Update app.py

app.py

@@ -15,9 +15,13 @@ defaultMsg = os.environ['defaultMsg']
 
 mod = 3;
 
-def localVar(lv):
-    lv = lv or {"race":"", "cls":""}
-    return lv
+def localRace(race):    
+    race = race
+    return race
+
+def localClass(cls):    
+    cls = cls
+    return cls
 
 def trimMessages(messages):
     # Assuming messages is an array of objects with "role" and "content" properties
@@ -58,8 +62,9 @@ def getChooseClass(msg):
     else:
         return ""
 
-def chatbot(input, messages):
+def chatbot(input, messages, localvar):    
     messages = messages or [{"role": "system", "content": initMsg}]
+    localvar = localvar or {"race": "", "class": ""}
     #if len(messages) == 1:
         
         #memory = trimMessages(messages)
@@ -80,19 +85,15 @@ def chatbot(input, messages):
         logging.info("AI Reply:"+reply)
         ans = ""
         if len(messages) == 2:
-            lv.race = getChooseRole(reply)
-            lv.cls = ""
-            lv = localVar(lv)
-            ans = "I choose "+ lv.race +" as my race." + classAsk
+            localvar['race'] = localRace(getChooseRole(reply))
+            ans = "I choose "+ localvar['race'] +" as my race." + classAsk
             logging.info("ChooseRole:"+ans)
         if len(messages) == 4:
-            lv = localVar(lv)
-            lv.cls = getChooseClass(reply)
-            lv = localVar(lv)
-            ans = "My Race is "+ lv.race +"，I choose "+ lv.cls +" as my class." + classAns
+            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})
-        return reply, printMessages(messages), messages
+        return reply, printMessages(messages), messages, localvar
 
 def printMessages(messages):
     delimiter = '\n'
@@ -100,7 +101,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"], 
-                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=roleAns), gr.Textbox(label="History"), "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", "state"], 
+                   outputs=[gr.Textbox(label="DND Game Reply", placeholder=roleAns), gr.Textbox(label="History"), "state", "state"], title="DND Game",#, gr.Textbox(label="History"), 
                    description="DND Game",theme="compact")
 app.launch(share=False)