Update data_loader.py

data_loader.py

@@ -5,18 +5,20 @@ from supabase import create_client, Client
 
 #Aqui van las credenciales, conectar las credenciales de Supabase en "Secrets"
 
-#SUPABASE_URL = 
-#SUPABASE_KEY = 
+Class DataLoader():
 
-def load_data():
-    """fertility, geo data, labor, population y predictions """
-    try:
-        response = supabase.from_("LabourForecast").execute()
-        if response.error:
-            st.error(f"Error fetching data: {response.error}")
+    #SUPABASE_URL = 
+    #SUPABASE_KEY = 
+
+    def load_data():
+        """fertility, geo data, labor, population y predictions """
+        try:
+            response = supabase.from_("LabourForecast").execute()
+            if response.error:
+                st.error(f"Error fetching data: {response.error}")
+                return pd.DataFrame()
+            else:
+                return pd.DataFrame(response.data)
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
             return pd.DataFrame()
-        else:
-            return pd.DataFrame(response.data)
-    except Exception as e:
-        st.error(f"An error occurred: {e}")
-        return pd.DataFrame()