Yashrajnpai's picture
Create tools_excel.py
284f14a verified
raw
history blame contribute delete
690 Bytes
import pandas as pd
import re
def excel_answer(file_path, question):
df = pd.read_excel(file_path)
# Example: sum for food sales, or other logic based on question
if 'food' in question.lower():
# Try to find a column with 'food' or a category column
if 'Category' in df.columns and 'Sales' in df.columns:
total = df[df['Category'].str.lower() == 'food']['Sales'].sum()
return f"{total:.2f}"
# Fallback: sum last column
total = df.iloc[:, -1].sum()
return f"{total:.2f}"
# Add more logic for average, etc., if needed
# Fallback: sum last column
total = df.iloc[:, -1].sum()
return f"{total:.2f}"