Update tools.py

tools.py

@@ -7,7 +7,48 @@ from langchain_community.document_loaders import ArxivLoader
 
 
 @tool
-def extract_text(img_path: str) -> str:
+def read_excel(file_path: str) -> str:
+    """
+    Extract readable text from an Excel file (.xlsx or .xls).
+
+    Args:
+        file_path: Path to the Excel file.
+
+    Returns:
+        A string representation of all sheets and their content.
+    """
+    try:
+        df_dict = pd.read_excel(file_path, sheet_name=None)  # Read all sheets
+        result = []
+        for sheet_name, sheet_df in df_dict.items():
+            sheet_text = sheet_df.to_string(index=False)
+            result.append(f"Sheet: {sheet_name}\n{sheet_text}")
+        return "\n\n".join(result)
+
+    except Exception as e:
+        return f"Error reading Excel file: {str(e)}"
+
+
+@tool
+def read_python(file_path: str) -> str:
+    """
+    Extract source code from a Python (.py) file.
+
+    Args:
+        file_path: Path to the Python file.
+
+    Returns:
+        A string containing the full source code of the file.
+    """
+    try:
+        with open(file_path, "r", encoding="utf-8") as f:
+            return f.read()
+    except Exception as e:
+        return f"Error reading Python file: {str(e)}"
+        
+        
+@tool
+def extract_text_from_image(img_path: str) -> str:
     """
     Extract text from an image file using a multimodal model.