update app.py

app.py

@@ -17,14 +17,20 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 from openai import OpenAI
 
 class BasicAgent:
+
     def __init__(self):
-        print("Advanced GAIA Agent initialized")
+        print("File-aware GAIA agent initialized")
         self.client = OpenAI()
 
-    # calculator tool
-    def calculate(self, expression):
+    def download_file(self, url):
+
+        if not url:
+            return None
+
         try:
-            return str(eval(expression))
+            r = requests.get(url, timeout=20)
+            r.raise_for_status()
+            return r.text[:4000]
         except:
             return None
 
@@ -35,21 +41,32 @@ class BasicAgent:
         text = re.sub(r"(?i)final answer[:\- ]*", "", text)
         text = re.sub(r"(?i)answer[:\- ]*", "", text)
 
-        # take last line
         text = text.split("\n")[-1]
 
         return text.strip()
 
-    def ask_llm(self, question):
+    def ask_llm(self, question, file_content=None):
 
-        prompt = f"""
-You are solving a GAIA benchmark problem.
+        if file_content:
+            prompt = f"""
+You are solving a GAIA benchmark question.
 
-Instructions:
-- Think step by step
-- Use math reasoning when necessary
-- Return ONLY the final answer
-- No explanation
+A file is provided.
+
+File content:
+{file_content}
+
+Question:
+{question}
+
+Return ONLY the final answer.
+No explanation.
+"""
+        else:
+            prompt = f"""
+Solve the following GAIA question.
+
+Return ONLY the final answer.
 
 Question:
 {question}
@@ -66,37 +83,20 @@ Question:
 
         return response.choices[0].message.content.strip()
 
-    def try_math(self, question):
-
-        # detect simple math expressions
-        matches = re.findall(r"[\d\.\+\-\*\/\(\) ]+", question)
-
-        for m in matches:
-            m = m.strip()
-            if len(m) > 3:
-                result = self.calculate(m)
-                if result:
-                    return result
-
-        return None
-
-    def __call__(self, question: str) -> str:
+    def __call__(self, question, file_url=None):
 
         print("Question:", question)
 
-        # try math tool first
-        math_result = self.try_math(question)
+        file_content = self.download_file(file_url)
 
-        if math_result:
-            print("Calculator used:", math_result)
-            return math_result
+        if file_content:
+            print("File detected and loaded")
 
-        # otherwise ask LLM
-        raw_answer = self.ask_llm(question)
+        raw_answer = self.ask_llm(question, file_content)
 
         final_answer = self.clean_answer(raw_answer)
 
-        print("Final answer:", final_answer)
+        print("Final Answer:", final_answer)
 
         return final_answer
 
@@ -161,7 +161,9 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             print(f"Skipping item with missing task_id or question: {item}")
             continue
         try:
-            submitted_answer = agent(question_text)
+     #       submitted_answer = agent(question_text) //vineet
+             file_url = item.get("file", None)
+             submitted_answer = agent(question_text, file_url)
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e: