Update app.py

app.py

@@ -16,9 +16,15 @@ class BasicAgent:
         self.agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel())
     def __call__(self, question: str) -> str:
         print(f"Agent received question (first 50 chars): {question[:50]}...")
-        agent_answer = self.agent.run(question)
-        print(f"Agent returning fixed answer: {fixed_answer}")
-        return fixed_answer
+        agent_answer = self.agent.run(f"There is GAIA test level 1 <question>{question}</question>. Provide the most possible answer inside <answer></answer> tag")
+        
+        print(f"Agent returning answer: {agent_answer}")
+        pattern = re.compile(r"<answer>(.*?)</answer>", re.DOTALL | re.IGNORECASE)
+
+        if many:
+            return pattern.findall(text)           # list[str]
+        m = pattern.search(text)
+        return m.group(1) if m else agent_answer
 
 def run_and_submit_all( profile: gr.OAuthProfile | None):
     """