Update app.py

app.py

@@ -258,69 +258,6 @@ def call_set_task(purpose, task, history, action_input):
     return "MAIN", None, history, task
 
 
-NAME_TO_FUNC = {
-    "MAIN": call_main,
-    "UPDATE-TASK": call_set_task,
-    "SEARCH_ENGINE": find_all,
-    "WEBSITE_SCRAPE": find_all,
-    
-
-}
-
-
-def run_action(purpose, task, history, action_name, action_input):
-    if action_name == "COMPLETE":
-        print("Complete - Exiting")
-        #exit(0) 
-        return "COMPLETE", None, history, task
-
-    # compress the history when it is long
-    if len(history.split("\n")) > MAX_HISTORY:
-        if VERBOSE:
-            print("COMPRESSING HISTORY")
-        history = compress_history(purpose, task, history)
-    if action_name in NAME_TO_FUNC:
-        
-        assert action_name in NAME_TO_FUNC
-
-        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)\n"
-
-        return "MAIN", None, history, task
-
-def run(purpose,history):
-    task=None
-    history = ""
-    #if not history:
-    #    history = []
-    action_name = "SEARCH" if task is None else "MAIN"
-    action_input = None
-    while True:
-        print("")
-        print("")
-        print("---")
-        #print("purpose:", purpose)
-        print("task:", task)
-        print("---")
-        #print(history)
-        print("---")
-
-        action_name, action_input, history, task = run_action(
-            purpose,
-            task,
-            history,
-            action_name,
-            action_input,
-        )
-        yield history
-        if action_name == "COMPLETE":
-            return history
-
-
-
-
 
 ###########################################################
 def search_all(url):
@@ -423,6 +360,69 @@ def find_it2(url):
         return e
 #################################
 
+NAME_TO_FUNC = {
+    "MAIN": call_main,
+    "UPDATE-TASK": call_set_task,
+    "SEARCH_ENGINE": find_all,
+    "WEBSITE_SCRAPE": find_all,
+    
+
+}
+
+
+def run_action(purpose, task, history, action_name, action_input):
+    if action_name == "COMPLETE":
+        print("Complete - Exiting")
+        #exit(0) 
+        return "COMPLETE", None, history, task
+
+    # compress the history when it is long
+    if len(history.split("\n")) > MAX_HISTORY:
+        if VERBOSE:
+            print("COMPRESSING HISTORY")
+        history = compress_history(purpose, task, history)
+    if action_name in NAME_TO_FUNC:
+        
+        assert action_name in NAME_TO_FUNC
+
+        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)\n"
+
+        return "MAIN", None, history, task
+
+def run(purpose,history):
+    task=None
+    history = ""
+    #if not history:
+    #    history = []
+    action_name = "SEARCH" if task is None else "MAIN"
+    action_input = None
+    while True:
+        print("")
+        print("")
+        print("---")
+        #print("purpose:", purpose)
+        print("task:", task)
+        print("---")
+        #print(history)
+        print("---")
+
+        action_name, action_input, history, task = run_action(
+            purpose,
+            task,
+            history,
+            action_name,
+            action_input,
+        )
+        yield history
+        if action_name == "COMPLETE":
+            return history
+
+
+
+
 examples =[
     "find the most popular model that I can use to generate an image by providing a text prompt",
     "return the top 10 models that I can use to identify objects in images",