Update app.py

app.py

@@ -227,7 +227,8 @@ NAME_TO_FUNC = {
 def run_action(purpose, task, history, action_name, action_input):
     if action_name == "COMPLETE":
         print("Complete - Exiting")
-        exit(0) 
+        #exit(0) 
+        return "COMPLETE", None, history, task
 
     # compress the history when it is long
     if len(history.split("\n")) > MAX_HISTORY:
@@ -241,7 +242,7 @@ def run_action(purpose, task, history, action_name, action_input):
         print("RUN: ", action_name, action_input)
         return NAME_TO_FUNC[action_name](purpose, task, history, action_input)
     else:
-        history += "observation: The TOOL I tried to use returned an error, I need to select a tool from: (UPDATE-TASK, SEARCH, COMPLETE)"
+        history += "observation: The TOOL I tried to use returned an error, I need to select a tool from: (UPDATE-TASK, SEARCH, COMPLETE)\n"
 
         return "MAIN", None, history, task