Update app.py

app.py

@@ -3,6 +3,7 @@ import datetime
 import requests
 import pytz
 import yaml
+import pandas as pd
 from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
@@ -29,6 +30,26 @@ def get_random_joke()-> str: #it's import to specify the return type
         # Return an error message if the request fails
         return f"Failed to retrieve joke. Status code: {response.status_code}"
 
+@tool
+def get_user_row(user_name: str) -> str:
+    """A tool that fetches more information about the user in a .csv file.
+    Args:
+        user_name: A string that's the user name.
+    """
+
+    # Load the CSV file into a DataFrame
+    df = pd.read_csv("data.csv", encoding="utf-8")
+
+    # Filter the DataFrame to find rows where the CLIENTE column matches the username
+    client_row = df[df["CLIENTE:"].str.strip().str.lower() == username.strip().lower()]
+
+    # If no rows are found, return None
+    if client_row.empty:
+        return f"Didn't find any information for CLIENTE: {user_name}"
+
+    # Return the row(s) as a dictionary (or DataFrame if preferred)
+    return client_row.to_dict(orient="records")
+
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
     """A tool that fetches the current local time in a specified timezone.
@@ -66,7 +87,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[get_random_joke,final_answer], ## add your tools here (don't remove final answer)
+    tools=[get_user_row,get_random_joke,final_answer], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,