Update app.py

app.py

@@ -13,7 +13,7 @@ import pandas as pd
 # from langchain.llms.base import LLM
 # from huggingface_hub import InferenceClient
 from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel
-
+from smolagent.tools import Tool
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -30,6 +30,17 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
+# --- Excel Reading Tool ---
+def read_excel_tool(file_path: str) -> str:
+    try:
+        if file_path.endswith('.csv'):
+            df = pd.read_csv(file_path)
+        else:
+            df = pd.read_excel(file_path)
+        return df.head(10).to_string(index=False)
+    except Exception as e:
+        return f"Error reading file: {str(e)}"
+        
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -40,17 +51,24 @@ class BasicAgent:
             "Always return a direct answer, no extra text or explanation. "
             "If a question asks for a count, return only the number."
         )
+
+        # Register Excel tool
+        read_excel = Tool.from_function(
+            func=read_excel_tool,
+            name="read_excel",
+            description="Reads Excel or CSV file and returns the first 10 rows as a string."
+        )
         
         model = InferenceClientModel(
             model_id="gpt-4o",
             token=openaikey,
-            provider="openai",
-            temperature=0.0 # 0.7 by default (not sure)
+            provider="openai"#,
+            # temperature=0.0 # 0.7 by default (not sure)
         )
 
 
         self.agent = ToolCallingAgent(
-            tools=[DuckDuckGoSearchTool()],
+            tools=[DuckDuckGoSearchTool(), read_excel],
             model=model
         )