Update app.py

app.py

@@ -28,10 +28,19 @@ client = InferenceClient(
 )
 
 def parse_action(string: str):
+    print("PARSING:")
+    print(string)
     assert string.startswith("action:")
     idx = string.find("action_input=")
+    print(idx)
     if idx == -1:
+        print ("idx == -1")
+        print (string[8:])
         return string[8:], None
+
+    print ("last return:")
+    print (string[8 : idx - 1])
+    print (string[idx + 13 :].strip("'").strip('"'))
     return string[8 : idx - 1], string[idx + 13 :].strip("'").strip('"')
 
 
@@ -89,7 +98,7 @@ def call_search(purpose, task, history, action_input):
                 print("compressing...")
                 return_list = compress_data(rl,purpose,task,return_list)
             history = "observation: the search results are:\n {}\n".format(return_list)
-            return "COMPLETE", None, history, task
+            return "MAIN", None, history, task
         else: 
             history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input=SEARCH_QUERY\n"
             return "UPDATE-TASK", None, history, task
@@ -309,11 +318,11 @@ def find_all(purpose,task,history, url):
             task = "complete?"
             return "MAIN", None, history, task
         else: 
-            history += "observation: I need to trigger a search using the following syntax:\naction: WEBSITE_SCRAPE action_input=SEARCH_QUERY\n"
+            history += "observation: I need to trigger a search using the following syntax:\naction: SCRAPE_WEBSITE action_input=SEARCH_QUERY\n"
             return "MAIN", None, history, task
     except Exception as e:
         print (e)
-        history += "observation: I need to trigger a search using the following syntax:\naction: WEBSITE_SCRAPE action_input=SEARCH_QUERY\n"
+        history += "observation: I need to trigger a search using the following syntax:\naction: SCRAPE_WEBSITE action_input=SEARCH_QUERY\n"
         return "MAIN", None, history, task
 
         #else: