Update app.py

app.py

@@ -1,5 +1,7 @@
 """ Basic Agent Evaluation Runner"""
 import os
+from dotenv import load_dotenv
+from supabase.client import create_client
 import inspect
 import gradio as gr
 import requests
@@ -198,12 +200,32 @@ with gr.Blocks() as demo:
         outputs=[status_output, results_table]
     )
 
+def test_supabase_connection():
+    load_dotenv()
+    
+    try:
+        supabase = create_client(
+            os.environ.get("SUPABASE_URL"),
+            os.environ.get("SUPABASE_SERVICE_KEY")
+        )
+        
+        # Test query
+        result = supabase.table('documents').select("*").limit(1).execute()
+        print(f"Connection successful! Retrieved {len(result.data)} documents.")
+        return True
+        
+    except Exception as e:
+        print(f"Connection failed: {e}")
+        return False
+
+
 if __name__ == "__main__":
     print("\n" + "-"*30 + " App Starting " + "-"*30)
     # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")
     space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
-
+    test_supabase_connection()
+    
     if space_host_startup:
         print(f"✅ SPACE_HOST found: {space_host_startup}")
         print(f"   Runtime URL should be: https://{space_host_startup}.hf.space")