Update app.py

app.py

@@ -143,6 +143,8 @@ def call_main(purpose, task, history, directory, action_input):
             history += "{}\n".format(line)
             if "COMPLETE" in action_name or "COMPLETE" in action_input:
                 task = "END"
+                action_name, action_input, history, task = end_fn(purpose, task, history, directory, action_input)
+
                 return action_name, action_input, history, task
             else:
                 return action_name, action_input, history, task