cleaner

app.py

@@ -471,23 +471,41 @@ class SlpMultiAgent:
         if result is None:
             return "I apologize, but I'm currently experiencing technical difficulties. Please try again later."
         
-        
-        # Extract clean answer from result
+        # --- Robust post-processing to extract final_answer ---
+        import re
+        def strip_code_blocks(text):
+            # Remove triple backtick code blocks
+            text = re.sub(r"```[\s\S]*?```", "", text)
+            # Remove <code>...</code> blocks
+            text = re.sub(r"<code>[\s\S]*?</code>", "", text)
+            # Remove markdown headers and print statements
+            text = re.sub(r"^#+.*$", "", text, flags=re.MULTILINE)
+            text = re.sub(r"print\(.*?\)", "", text)
+            return text.strip()
+        def extract_final_answer(text):
+            # Try to find final_answer('...') or final_answer("...")
+            match = re.search(r"final_answer\(['\"](.*?)['\"]\)", text, re.DOTALL)
+            if match:
+                return match.group(1).strip()
+            # Try to find final_answer(...) with any content
+            match = re.search(r"final_answer\((.*?)\)", text, re.DOTALL)
+            if match:
+                return match.group(1).strip().strip("'\"")
+            # As a last resort, return the first non-empty line
+            for line in text.splitlines():
+                if line.strip():
+                    return line.strip()
+            return text.strip()
+        # Post-process the result to remove code blocks and extract the answer
         if result and isinstance(result, str):
-            # Look for final_answer pattern
-            import re
-            final_answer_match = re.search(r'final_answer\(["\']([^"\']*)["\'\)]', result)  # Fixed regex
-            if final_answer_match:
-                clean_answer = final_answer_match.group(1)
-                return clean_answer
-            
-            # If no final_answer found, try to extract the last meaningful line
-            lines = result.strip().split('\n')
-            for line in reversed(lines):
-                line = line.strip()
-                if line and not line.startswith('#') and not line.startswith('###') and len(line) < 200:
-                    return line
-        
+            cleaned = strip_code_blocks(result)
+            answer = extract_final_answer(cleaned)
+            if answer:
+                return answer
+            # Fallback: try to extract from the original result
+            answer = extract_final_answer(result)
+            if answer:
+                return answer
         # Return the result from the agent
         return result if result else "Unable to determine answer."