Update app.py

app.py

@@ -78,25 +78,27 @@ class BasicAgent:
             else: 
                 df = pd.read_excel(file_path)
             return df.to_string()
-            
-        @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)
-            print(read_pdf.description)
-            print(read_pdf.inputs)
+
+        class ReadPDFTool(Tool):
+            name = "read_pdf"
+            description = "Extracts text from a PDF file."
+            inputs = {
+                "file_path": {
+                    "type": "string",
+                    "description": "Local path to the PDF file.",
+                }
+            } 
+            output_type = "string"
+
+            def forward(self, file_path: str) -> str:
+                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
         #)
+        read_pdf = ReadPDFTool()
         self.agent = CodeAgent(
             tools=[
                 DuckDuckGoSearchTool(),
@@ -110,7 +112,7 @@ class BasicAgent:
                 get_youtube_transcript,
                 ],
             model= InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"),
-            additional_authorized_imports=["pandas", "requests", "re"],
+            #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,
         )