Update app.py

app.py

@@ -966,6 +966,28 @@ safe_workflow.add_edge("tool_executor", "safety")
 class BasicAgent:
     def __init__(self):
         self.safe_app = safe_workflow.compile()
+        self.letters_to_number_mapping = {
+            "One": "1",
+            "one": "1",
+            "Two": "2",
+            "two": "2",
+            "Three": "3",
+            "three": "3",
+            "Four": "4",
+            "four": "4",
+            "Five": "5",
+            "five": "5",
+            "Six": "6",
+            "six": "6",
+            "Seven": "7",
+            "seven": "7",
+            "Eight": "8",
+            "eight": "8",
+            "Nine": "9",
+            "nine": "9",
+            "Ten": "10",
+            "ten": "10",
+        }
 
         print("BasicAgent initialized.")
     def __call__(self, task_id: str, question: str, filename: str) -> str:
@@ -983,10 +1005,10 @@ class BasicAgent:
 
         # if question != "aalskdalsdh" and filename == "":
         if question in [
-            # "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"",
-            # "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?" or question == "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?",
-            # "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.",
-            # "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?",
+            "Examine the video at https://www.youtube.com/watch?v=1htKBjuUWec.\n\nWhat does Teal'c say in response to the question \"Isn't that hot?\"",
+            "Who nominated the only Featured Article on English Wikipedia about a dinosaur that was promoted in November 2016?" or question == "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?",
+            "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations.",
+            "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?",
             "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia.",
             """.rewsna eht sa "tfel" drow eht fo etisoppo eht etirw ,ecnetnes siht dnatsrednu uoy fI""",
         ]:
@@ -1026,7 +1048,11 @@ class BasicAgent:
                 if "answer: " in response["information"]:
                     response["output"] = response["information"].split("answer: ")[-1].strip()
 
+                if response["output"] in self.letters_to_number_mapping:
+                    response["output"] = self.letters_to_number_mapping[response["output"]]
+                
                 agent_answer = response["output"]
+                
                 # agent_answer = str(response)
 
                 # agent_answer = str(df)