Update app.py

app.py

@@ -228,13 +228,13 @@ def agent(prompt_in,history,mod=2):
         in_data[4]=outph
         print(outph)
         history=history+[{'role':'assistant','content':str(outph)}]
-        yield [{'role':'assistant','content':str(outph)}]
+        yield history
         role="MANAGER"
         outp=generate(prompt,history,mod,128,seed,role,in_data)
         outp0=list(outp)[0].split('<|im_end|>')[0]
         #outp0 = re.sub('[^a-zA-Z0-9\s.,?!%()]', '', outpp)
         history=history+[{'role':'assistant','content':str(outp0)}]
-        yield [{'role':'assistant','content':str(outp0)}]
+        yield history
         for line in outp0.split("\n"):
             if "action:" in line:
                 try:
@@ -254,6 +254,7 @@ def agent(prompt_in,history,mod=2):
                     ret1,ret2 = parse_json(out_w[2].split('<|im_end|>')[0])
                     build_space('test1',ret1,ret2)
                     history+=[{'role':'system','content':f'We just successfully build the file: {ret1}'}]
+                    yield history
                 elif 'IMAGE' in fn:
                     print('IMAGE called')
                     #out_im=gen_im(prompt,seed)