Sw1ft0 commited on
Commit
0b8b15e
·
1 Parent(s): 23901ad

Update upload_excel() to read all pages in a file instead of just 1

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -40,9 +40,13 @@ def check_occupancy(month: str) -> str:
40
  def upload_excel(file):
41
  if file is None:
42
  return "Please upload an Excel file."
43
- df = pd.read_excel(file.name)
44
- dataframes["df"] = df
45
- return f"✅ File loaded with {len(df)} rows and {len(df.columns)} columns."
 
 
 
 
46
 
47
  # -----------------------------
48
  # LangChain Agent Setup
 
40
  def upload_excel(file):
41
  if file is None:
42
  return "Please upload an Excel file."
43
+ dfs = pd.read_excel(file.name, sheet_name=None)
44
+ dataframes.clear()
45
+ dataframes.update(dfs)
46
+ sheet_info = []
47
+ for name, df in dfs.items():
48
+ sheet_info.append(f"'{name}': {len(df)} rows, {len(df.columns)} cols")
49
+ return f"✅ Loaded sheets: {', '.join(sheet_info)}"
50
 
51
  # -----------------------------
52
  # LangChain Agent Setup