File size: 690 Bytes
284f14a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
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}"