file handling

agent_utils.py

@@ -6,7 +6,7 @@ from tavily import TavilyClient
 import requests
 from PIL import Image
 from io import BytesIO
-
+import pandas as pd
 
 def wiki_ret(question: str) -> str:
     """ Retrieve docs from wikipedia """
@@ -68,7 +68,6 @@ def tavily_ret(query: str, max_results: int = 3) -> str:
 
 
 def handle_file_tool(input: dict) -> str:
-    """Processes file attached to a question"""
     question = input.get("question", "")
     file_url = input.get("file", "")
 
@@ -79,16 +78,30 @@ def handle_file_tool(input: dict) -> str:
     if not response.ok:
         return f"Failed to download file: {file_url}"
 
-    # Handle image files
-    if file_url.endswith((".png", ".jpg", ".jpeg")):
+    file_ext = file_url.split("?")[0].split(".")[-1].lower()
+
+    # IMAGE
+    if file_ext in ("png", "jpg", "jpeg"):
         image = Image.open(BytesIO(response.content))
-        return f"Received image for question: '{question}'. (Add vision model here.)"
+        return f"Image file received for question: '{question}'. (Insert vision model here.)"
 
-    # Handle PDFs
-    elif file_url.endswith(".pdf"):
+    # PDF
+    elif file_ext == "pdf":
         with open("temp.pdf", "wb") as f:
             f.write(response.content)
-        return f"Received PDF for question: '{question}'. (Add PDF parser here.)"
+        return f"PDF file received. (Insert PDF parsing logic here.)"
+
+    # EXCEL
+    elif file_ext in ("xlsx", "xls"):
+        df = pd.read_excel(BytesIO(response.content))
+        summary = f"Excel data preview:\n{df.head().to_string(index=False)}"
+        return f"Question: {question}\n{summary}"
+
+    # PYTHON CODE
+    elif file_ext == "py":
+        code = response.content.decode("utf-8")
+        return f"Question: {question}\nPython code received:\n{code[:500]}..."  # Limit preview
 
     else:
-        return f"Unsupported file type: {file_url}"
+        return f"Unsupported file type: .{file_ext}"
+

requirements.txt

@@ -7,4 +7,5 @@ wikipedia
 langgraph
 arxiv
 tavily-python
-Pillow
+Pillow
+pandas