Update app.py

Testing connection to supabase

app.py

@@ -58,7 +58,7 @@ def load_data():
         return pd.DataFrame()
 
 
-
+data = load_data()
 # Pendiente las Keys, dependiendo del modelo que escojamos
 
 model_name = "google/flan-t5-small"  # Probando modelos
@@ -72,19 +72,12 @@ generate_button = st.button("Generate")
 
 if generate_button and user_input:
     
-    if not SUPABASE_URL or not SUPABASE_KEY:
-        print("Error: Supabase URL and/or Key not found in environment variables.")
-    
+    if data.empty and supabase is not None:
+        st.warning("Successfully connected to Supabase, but no data was loaded (either the table is empty or there was a query issue). Check the error message above if any.")
+    elif not data.empty:
+        st.success("Successfully connected to Supabase and loaded data!")
+        st.dataframe(data.head()) # Display a small sample of the data
+    elif supabase is None:
+        st.error("Failed to initialize Supabase client. Check environment variables in Settings.")
     else:
-        try:
-            
-            response = supabase.table("LabourForecast").select("*").limit(1).execute()
-    
-            if response.error:
-                print(f"Supabase connection test failed with error: {response.error}")
-            else:
-                print("Supabase connection test successful!")
-                print("First row from 'your_test_table':", response.data)
-    
-        except Exception as e:
-            print(f"An unexpected error occurred during Supabase connection test: {e}")
+        st.info("Attempted to load data. Check for any error messages above.")