Add graceful handling for missing db connection

agent/backend.py

@@ -10,10 +10,12 @@ GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_KEY = os.getenv("SUPABASE_KEY")
 
-if not all([GROQ_API_KEY, SUPABASE_URL, SUPABASE_KEY]):
-    raise ValueError("Missing required environment variables: GROQ_API_KEY, SUPABASE_URL, SUPABASE_KEY")
-
-supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
+supabase = None
+if all([GROQ_API_KEY, SUPABASE_URL, SUPABASE_KEY]):
+    try:
+        supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
+    except Exception as e:
+        print(f"WARNING: Could not initialize Supabase: {e}")
 
 
 # ─────────────────────────────────────────
@@ -21,26 +23,38 @@ supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
 # ─────────────────────────────────────────
 
 def get_patient_by_name(name: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("patients").select("*").ilike("name", f"%{name}%").execute()
     return res.data if res.data else {"error": "Patient not found"}
 
 def get_patients_by_doctor(doctor: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("patients").select("*").ilike("doctor", f"%{doctor}%").execute()
     return res.data if res.data else {"error": "No patients found for this doctor"}
 
 def get_admitted_patients():
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("patients").select("*").eq("status", "admitted").execute()
     return res.data if res.data else {"message": "No admitted patients currently"}
 
 def get_appointments_by_date(date: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("appointments").select("*").eq("date", date).execute()
     return res.data if res.data else {"message": f"No appointments on {date}"}
 
 def get_patient_appointments(patient_name: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("appointments").select("*").ilike("patient_name", f"%{patient_name}%").execute()
     return res.data if res.data else {"error": "No appointments found for this patient"}
 
 def add_patient(name: str, age: int, gender: str, disease: str, doctor: str, admission_date: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     data = {
         "name": name,
         "age": age,
@@ -54,16 +68,22 @@ def add_patient(name: str, age: int, gender: str, disease: str, doctor: str, adm
     return {"success": True, "patient": res.data[0]} if res.data else {"error": "Failed to add patient"}
 
 def discharge_patient(patient_id: int):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("patients").update({"status": "discharged"}).eq("id", patient_id).execute()
     return {"success": True, "message": f"Patient {patient_id} discharged"} if res.data else {"error": "Patient not found"}
 
 def update_appointment_status(appointment_id: int, status: str):
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     if status not in ["scheduled", "completed", "cancelled"]:
         return {"error": "Invalid status. Use: scheduled, completed, cancelled"}
     res = supabase.table("appointments").update({"status": status}).eq("id", appointment_id).execute()
     return {"success": True, "message": f"Appointment {appointment_id} updated to {status}"} if res.data else {"error": "Appointment not found"}
 
 def get_disease_statistics():
+    if not supabase:
+        return {"error": "Database not connected. Check SUPABASE_URL and SUPABASE_KEY."}
     res = supabase.table("patients").select("disease").execute()
     if not res.data:
         return {"message": "No data available"}
@@ -227,6 +247,9 @@ Format responses cleanly with bullet points for multiple records.
 """
 
 async def process_agent_chat(user_message: str):
+    if not GROQ_API_KEY:
+        return {"response": "Error: GROQ_API_KEY not configured. Please set it in Space Variables."}
+    
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"