Update app.py

app.py

@@ -6,9 +6,10 @@ import operator
 import requests
 import pandas as pd
 import gradio as gr
+
 try:
     import spaces
- 
+
     @spaces.GPU
     def _zerogpu_warmup():
         # Dummy function so this ZeroGPU Space passes its startup check.
@@ -18,6 +19,7 @@ try:
 except ImportError:
     # Not running on a ZeroGPU Space / package not installed locally — fine to skip.
     pass
+
 from typing import TypedDict, Annotated
 from langgraph.graph import StateGraph, END
 from langgraph.prebuilt import ToolNode
@@ -115,7 +117,10 @@ TOOLS = [calculator, search_tool, wiki_tool, python_tool, read_downloaded_file]
 # MODEL
 # =========================================================
 # Requires GROQ_API_KEY set as a Space secret (free — get one at console.groq.com).
-llm = ChatGroq(model="llama-3.3-70b-versatile", temperature=0)
+# NOTE: llama-3.3-70b-versatile was deprecated by Groq and produces unreliable
+# tool calls. openai/gpt-oss-120b is Groq's recommended replacement and is
+# built for reliable structured tool calling.
+llm = ChatGroq(model="openai/gpt-oss-120b", temperature=0)
 llm_with_tools = llm.bind_tools(TOOLS)
 
 SYSTEM_PROMPT = """You are a general AI assistant answering benchmark questions that will be
@@ -159,9 +164,11 @@ def verify_answer(state: AgentState):
     draft = state["messages"][-1].content
     verify_prompt = [
         SystemMessage(content=(
-            "Extract ONLY the final answer from the text below, with no extra words, "
-            "no explanation, no punctuation unless part of the answer itself, and no "
-            "'FINAL ANSWER:' prefix. Return just the answer."
+            "Extract ONLY the final answer from the text below. Strip ALL extra words, "
+            "units, labels, explanations, and surrounding context — return the bare answer only. "
+            "If it's a number, return digits only (e.g. '277' not '277 species', '3' not "
+            "'three albums'). If it's a name, return just the name. No punctuation unless it is "
+            "part of the answer itself. No 'FINAL ANSWER:' prefix. No quotes around the answer."
         )),
         HumanMessage(content=str(draft)),
     ]