Update app.py

app.py

@@ -15,19 +15,42 @@ def load_file(file):
     """Load a CSV or Excel file into a pandas DataFrame."""
     if file is None:
         return None
-    file_name = file.name.lower()
-    try:
-        if file_name.endswith('.csv'):
-            df = pd.read_csv(file)
-        elif file_name.endswith('.xlsx'):
-            df = pd.read_excel(file, engine='openpyxl')
-        elif file_name.endswith('.xls'):
-            df = pd.read_excel(file, engine='xlrd')
-        else:
+
+    # If file is provided as a dictionary (common in Gradio on Spaces)
+    if isinstance(file, dict):
+        file_name = file.get("name", "").lower()
+        file_path = file.get("data", None)
+        if file_path is None:
             return None
-    except Exception:
-        return None
-    return df
+        try:
+            if file_name.endswith('.csv'):
+                df = pd.read_csv(file_path)
+            elif file_name.endswith('.xlsx'):
+                df = pd.read_excel(file_path, engine='openpyxl')
+            elif file_name.endswith('.xls'):
+                df = pd.read_excel(file_path, engine='xlrd')
+            else:
+                return None
+        except Exception as e:
+            print(e)
+            return None
+        return df
+    else:
+        # Assume file is a file-like object (as on your local machine)
+        file_name = file.name.lower()
+        try:
+            if file_name.endswith('.csv'):
+                df = pd.read_csv(file)
+            elif file_name.endswith('.xlsx'):
+                df = pd.read_excel(file, engine='openpyxl')
+            elif file_name.endswith('.xls'):
+                df = pd.read_excel(file, engine='xlrd')
+            else:
+                return None
+        except Exception:
+            return None
+        return df
+
 
 def preview_file(file):
     """Return the DataFrame for preview."""