Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 import inspect
 import pandas as pd
 from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, Tool, tool, VisitWebpageTool
+from smolagents.prompts import CODE_SYSTEM_PROMPT
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -98,16 +99,22 @@ class BasicAgent:
                 "to inspect it before answering. Use DuckDuckGoSearchTool only "
                 "when the question needs current or external information."
             ),"""
-
+        pdf_tool = load_tool(
+            "matterattetatte/pdf-upload-extractor-tool", 
+            trust_remote_code = True
+        )
+        modified_system_message = CODE_SYSTEM_PROMPT + "If the question includes a task_id and mentions a file, call fetch_task_file first; route YouTube URLs to get_youtube_transcript, other URLs to visit_webpage, PDFs to pdf_tool, and spreadsheets to read_spreadsheet, using web_search only when no URL or file is given — then respond with only the exact final answer value, no explanation, no prefix.
         self.agent = CodeAgent(
             tools=[
                 DuckDuckGoSearchTool(),
                 VisitWebpageTool(),
+                pdf_tool,
                 fetch_task_file,
                 read_spreadsheet,
                 get_youtube_transcript,
                 #os.environ["HF_TOKEN"] = "hf_token",
                 #image_analysis_tool,
+                modified_system_message
             ],
             model=InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"),
             max_steps=20,