new tools + timezone tool in agent

app.py

@@ -1,11 +1,15 @@
 from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
 import datetime
+import fitz
 import requests
+import spacy
+import pandas as pd
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
+nlp = spacy.load("en_core_web_sm")
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
@@ -18,6 +22,46 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
     """
     return "What magic will you build ?"
 
+@tool
+def extract_text_from_pdf(pdf_path):
+    """A tool that extracts text from a PDF file.
+    Args:
+        pdf_path: A string representing the path to the PDF file.
+    """
+    document = fitz.open(pdf_path)
+    text = ""
+
+    # Iterate through each page
+    for page_num in range(len(document)):
+        page = document.load_page(page_num)
+        text += page.get_text()
+
+    return text
+
+@tool
+def extract_entities(text, desired_entities):
+    # Process the text with spaCy
+    doc = nlp(text)
+
+    # Extract entities that match the desired types
+    entities = [(ent.text, ent.label_) for ent in doc.ents if ent.label_ in desired_entities]
+    return entities
+
+@tool
+def create_table_with_specific_entities(text, desired_entities):
+    # Extract specific entities
+    entities = extract_entities(text, desired_entities)
+
+    # Create a DataFrame with desired entities as columns
+    df = pd.DataFrame(columns=desired_entities)
+
+    # Populate the DataFrame with entity values
+    for entity, label in entities:
+        df.loc[len(df), label] = entity
+
+    return df
+
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -51,7 +95,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,