Update app.py

app.py

@@ -209,22 +209,22 @@ class BasicAgent:
         while not answer or answer in wrong_answers or answer.lower().startswith("error"):
             try:
                 alfred = build_agent(chat1)
-                answer1 = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
             except Exception as e:
                 alfred = build_agent(chat3)
-                answer1 = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content            
-        if answer1:
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content            
+        if answer:
             alfred_super = build_agent(chat1)
-            answer = alfred_super.invoke(
-                input={"messages": [f"Verify the answer to the following question. QUESTION: {message}, ANSWER: {answer1}"]},config={"recursion_limit": 6})['messages'][-1].content
-            answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))    
-            answer = answer.replace('FINAL ANSWER:', '')
-            answer = answer.replace('FINAL ANSWER', '')
-            answer = answer.replace('YOUR ', '')
-            answer = answer.replace('*', '')
-            print(f"Agent returning fixed answer: {answer}")        
+            answer_fin = alfred_super.invoke(
+                input={"messages": [f"Verify the answer to the following question. QUESTION: {message}, ANSWER: {answer}"]},config={"recursion_limit": 6})['messages'][-1].content
+            answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))    
+            answer_fin = answer_fin.replace('FINAL ANSWER:', '')
+            answer_fin = answer_fin.replace('FINAL ANSWER', '')
+            answer_fin = answer_fin.replace('YOUR ', '')
+            answer_fin = answer_fin.replace('*', '')
+            print(f"Agent returning fixed answer: {answer_fin}")        
     
-            return answer
+        return answer_fin
 
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):