Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, Tool, tool,PythonInterpreterTool, VisitWebpageTool, FinalAnswerTool,  load_tool
+from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool,Tool,tool,PythonInterpreterTool,VisitWebpageTool,FinalAnswerTool,load_tool
 
 
 # (Keep Constants as is)
@@ -75,19 +75,22 @@ class BasicAgent:
             """
             if file_path.endswith(".csv"):
                 df = pd.read_csv(file_path)
-            elif file_path.edswith(".pdf"):
-                df = pd.read_pdf(file_path)
             else: 
                 df = pd.read_excel(file_path)
             return df.to_string()
-
-        #image_analysis_tool = Tool.from_space(
-            #"nvidia/LocateAnything",
-            #name="image_analyser",
-            #description="Analyses images to locate and describe objects in them.",
-            #trust_remote_code=True,
-        #)
-        
+            
+        @tool
+        def read_pdf(file_path: str) -> str:
+            """
+            Extracts text from a PDF file.
+            Args: 
+                 file_path: Local path to the PDF file.
+            Returns: 
+                 The extracted text content
+                 """
+            from pypdf import PdfReader
+            reader = PdfReader(file_path)
+            return "\n".join(page.extract_text() or "" for page in reader.pages)
         #pdf_tool = load_tool(
             #"matterattetatte/pdf-upload-extractor-tool", 
            # trust_remote_code = True
@@ -100,10 +103,12 @@ class BasicAgent:
                 FinalAnswerTool(),
                 #pdf_tool,
                 fetch_task_file,
+                read_pdf,
                 read_spreadsheet,
                 get_youtube_transcript,
                 ],
             model= InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"),
+            additional_authorized_imports=["pandas", "requests", "re"],
             #instructions = "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.",
             max_steps=20,
         )