Update app.py

app.py

@@ -96,7 +96,7 @@ def generate(prompt,history,mod=2,tok=4000,seed=1,role="ASSISTANT",data=None):
         #prompt=f"We just completed role:{role}, now choose the next tool to complete the task:{prompt}, or COMPLETE"
         for response in stream:
             output += response.token.text
-            #print(output)
+        print(output)
         yield output
         yield history
         yield prompt
@@ -196,7 +196,7 @@ def agent(prompt_in,history,mod=2):
                     fn="NONE"
                 if 'CREATE_FILE' in fn:
                     print('CREATE_FILE called')
-                    out_w =generate(com,history,mod=mod,tok=None,seed=seed,role="CREATE_FILE")
+                    out_w =generate(com,history,mod=mod,tok=None,seed=seed,role="CREATE_FILE",data=in_data)
                     build_space(out_w[0],out_w[1],out_w[2])
                 elif 'IMAGE' in fn:
                     print('IMAGE called')