Update app.py

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
+import re
 
 # 🔹 NOVO: imports do smolagents
 from smolagents import CodeAgent, DuckDuckGoSearchTool, InferenceClientModel
@@ -17,35 +18,42 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # =========================================================
 def clean_answer(text: str) -> str:
     """
-    Limpa a saída do modelo para ficar mais adequada ao EXACT MATCH.
-    Remove prefixos como 'Answer:', 'Final answer:' etc.,
-    aspas externas e ponto final solto.
+    Cleans the answer returned by the model:
+    - removes newlines
+    - removes phrases like 'final answer', 'answer:'
+    - removes quotes
+    - trims spaces
+    But DOES NOT delete useful content.
     """
-    if text is None:
+
+    if not text:
         return ""
 
-    ans = str(text).strip()
+    text = str(text).strip()
 
-    # remove prefixos comuns
-    prefixes = [
-        "answer:", "resposta:", "final answer:", "final:", "ans:", "a:",
-        "the answer is", "the final answer is",
+    # Remover frases proibidas
+    patterns_to_remove = [
+        r"(?i)final answer[:\- ]*",
+        r"(?i)answer[:\- ]*",
+        r"(?i)the answer is[:\- ]*",
+        r"(?i)my answer is[:\- ]*",
     ]
-    lower = ans.lower()
-    for p in prefixes:
-        if lower.startswith(p):
-            ans = ans[len(p):].strip()
-            break
+    for p in patterns_to_remove:
+        text = re.sub(p, "", text).strip()
+
+    # Remover quebras de linha
+    text = text.replace("\n", " ").strip()
 
-    # remove ponto final se não parecer número decimal
-    if ans.endswith(".") and not ans.replace(".", "", 1).isdigit():
-        ans = ans[:-1].strip()
+    # Remover aspas externas (se existirem)
+    if len(text) >= 2 and text.startswith('"') and text.endswith('"'):
+        text = text[1:-1].strip()
+    if len(text) >= 2 and text.startswith("'") and text.endswith("'"):
+        text = text[1:-1].strip()
 
-    # remove aspas externas
-    if len(ans) > 1 and ans[0] == ans[-1] and ans[0] in ["'", '"']:
-        ans = ans[1:-1].strip()
+    # Remover espaços múltiplos
+    text = re.sub(r"\s+", " ", text)
 
-    return ans
+    return text.strip()
 
 
 # =========================================================