Update app.py

app.py

@@ -147,6 +147,19 @@ class BasicAgent:
             )
             risposta = self._ask_gpt4o(prompt_python)
 
+        elif file_info.endswith(".xlsx"):
+            print_coso("Excel file detected")
+            excel_text = self._load_excel_as_text(file_info)
+            print_coso(f"Excel before prompt: {excel_text}")
+            prompt = (
+                "The following is the text extracted from an Excel spreadsheet.\n"
+                "Please use it to answer the question that follows:\n\n"
+                f"{excel_text}\n\n"
+                f"Question: {question}\n"
+                "Provide only the final answer. If it is a number, format it as USD with two decimal places if relevant. Do not explain."
+            )
+            risposta = self._ask_gpt4o(prompt)
+
         elif file_info.endswith(".txt"):
             print("coso Text file detected")
             text_content = self._load_text(file_info)
@@ -336,6 +349,27 @@ def get_or_download_code(file_name: str) -> str:
 
     with open(file_path, "r") as f:
         return f.read()
+
+
+def _load_excel_as_text(self, file_name: str) -> str:
+    import pandas as pd
+    import os
+
+    file_path = os.path.join("data", file_name)
+    try:
+        xl = pd.ExcelFile(file_path)
+        sheets = xl.sheet_names
+        all_text = ""
+        for sheet in sheets:
+            df = xl.parse(sheet)
+            all_text += f"\nSheet: {sheet}\n"
+            all_text += df.to_string(index=False)
+        return all_text
+    except Exception as e:
+        print_coso(f"[ERRORE] Impossibile leggere il file Excel: {e}")
+        return "ERROR: Could not read the Excel file."
+
+
     
     '''
     base_url = "https://huggingface.co/datasets/gaia-benchmark/GAIA/resolve"
@@ -372,19 +406,22 @@ def create_mock_questions():
             "Level": "1",
             "file_name": ""
         }
-    '''  
-
-
-
-    
-    
-    return [  
-       {
+         {
           "task_id":"f918266a-b3e0-4914-865d-4faa564f1aef",
           "question":"What is the final numeric output from the attached Python code?",
           "Level":"1",
           "file_name":"f918266a-b3e0-4914-865d-4faa564f1aef.py"
        }
+    '''  
+
+
+    return [    
+        {
+            "task_id":"7bd855d8-463d-4ed5-93ca-5fe35145f733",
+            "question":"The attached Excel file contains the sales of menu items for a local fast-food chain. What were the total sales that the chain made from food (not including drinks)? Express your answer in USD with two decimal places.",
+            "Level":"1",
+            "file_name":"7bd855d8-463d-4ed5-93ca-5fe35145f733.xlsx"
+        }
     ]
 
 
@@ -543,7 +580,6 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         print(f"coso final_status: {final_status} - results_df: {results_df}")
         return final_status, results_df
         
-        #return "mock1", None
 
         
     except requests.exceptions.HTTPError as e: