Update app.py

app.py

@@ -430,7 +430,7 @@ class SlpMultiAgent:
             description="Answer questions using Google search, Wikipedia, and math.",
             additional_authorized_imports=["re", "math", "requests"],
             verbosity_level=0,
-            max_steps=4
+            max_steps=6
         )
         
         # Create a task for the agent run with retry mechanism for rate limits
@@ -445,20 +445,22 @@ class SlpMultiAgent:
                     lambda: manager_agent.run(f"""
                     Question: {short_question}
                     
-                    Instructions:
-                    1. For Wikipedia questions, use wikipedia_search() first
-                    2. For other factual questions, use google_search()
-                    3. For math problems, use math_calculator()
-                    4. Give a specific, direct answer
+                    Answer this question by:
+                    1. Using wikipedia_search() for Wikipedia questions
+                    2. Using google_search() for other factual questions  
+                    3. Using math_calculator() for calculations
+                    4. Extracting the specific answer from search results
+                    5. Calling final_answer() with ONLY the answer
                     
-                    Format:
-                    Thoughts: [brief reasoning]
+                    Example:
+                    Thoughts: Need to search for information about X
                     <code>
-                    # Search for information
-                    result = wikipedia_search("search term") # or google_search()
-                    # Extract the specific answer from results
-                    final_answer("specific answer only")
+                    search_result = wikipedia_search("search term")
+                    # Extract specific answer from search_result
+                    final_answer("just the answer")
                     </code>
+                    
+                    IMPORTANT: final_answer() should contain ONLY the direct answer, not explanations.
                     """)
                 )
                 break  # Success, exit retry loop
@@ -484,6 +486,8 @@ class SlpMultiAgent:
         if result and isinstance(result, str):
             # Look for final_answer pattern - more flexible regex
             import re
+            
+            # First try to find final_answer calls
             final_answer_patterns = [
                 r'final_answer\(["\']([^"\']*)["\'\)]',
                 r'final_answer\("([^"]*)",?\)',
@@ -495,12 +499,27 @@ class SlpMultiAgent:
                 match = re.search(pattern, result, re.IGNORECASE | re.DOTALL)
                 if match:
                     clean_answer = match.group(1).strip('"\'')
-                    # Clean up common artifacts
                     clean_answer = re.sub(r'^[\s\n]*', '', clean_answer)
                     clean_answer = re.sub(r'[\s\n]*$', '', clean_answer)
-                    if clean_answer and not clean_answer.startswith('If you have'):
+                    if clean_answer and len(clean_answer) > 0 and not clean_answer.startswith('If you have'):
                         return clean_answer
             
+            # If no final_answer found, try to extract from the end of the result
+            if result:
+                # Remove code blocks and common artifacts
+                cleaned = re.sub(r'</code>.*$', '', result, flags=re.DOTALL)
+                cleaned = re.sub(r'<code>.*?</code>', '', cleaned, flags=re.DOTALL)
+                cleaned = re.sub(r'```.*?```', '', cleaned, flags=re.DOTALL)
+                
+                # Get the last meaningful line
+                lines = cleaned.strip().split('\n')
+                for line in reversed(lines):
+                    line = line.strip()
+                    if (line and not line.startswith('#') and not line.startswith('Thoughts:') 
+                        and not line.startswith('•') and len(line) < 200 
+                        and not line.startswith('If you have')):
+                        return line
+            
             # If no final_answer found, try to extract the last meaningful line
             lines = result.strip().split('\n')
             for line in reversed(lines):