Update main.py

main.py

@@ -5,7 +5,6 @@ from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from typing import Optional, List
 import httpx
-from supabase import create_client, Client
 from datetime import datetime
 import uuid
 
@@ -13,7 +12,7 @@ app = FastAPI(title="AI Team Chat API")
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],  # Replace with your frontend URL in production
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -22,15 +21,32 @@ app.add_middleware(
 # ─────────────────────────────────────────────
 # ENV VARS (set in HuggingFace Space secrets)
 # ─────────────────────────────────────────────
-GROQ_API_KEY = os.getenv("GROQ_API_KEY", "")
+GROQ_API_KEY   = os.getenv("GROQ_API_KEY", "")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY", "")
-SUPABASE_URL = os.getenv("SUPABASE_URL", "")
-SUPABASE_KEY = os.getenv("SUPABASE_KEY", "")
+SUPABASE_URL   = os.getenv("SUPABASE_URL", "")
+SUPABASE_KEY   = os.getenv("SUPABASE_KEY", "")
 
-# Supabase client (optional - graceful fallback if not configured)
-supabase: Optional[Client] = None
-if SUPABASE_URL and SUPABASE_KEY:
-    supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
+# ─────────────────────────────────────────────
+# SUPABASE — lazy init (never crashes startup)
+# ─────────────────────────────────────────────
+_supabase_client = None
+
+def get_supabase():
+    global _supabase_client
+    if _supabase_client is not None:
+        return _supabase_client
+    url = os.getenv("SUPABASE_URL", "")
+    key = os.getenv("SUPABASE_KEY", "")
+    if not url or not key:
+        return None
+    try:
+        from supabase import create_client
+        _supabase_client = create_client(url, key)
+        print("Supabase connected successfully.")
+        return _supabase_client
+    except Exception as e:
+        print(f"Supabase init error (non-fatal): {e}")
+        return None
 
 
 # ─────────────────────────────────────────────
@@ -38,9 +54,9 @@ if SUPABASE_URL and SUPABASE_KEY:
 # ─────────────────────────────────────────────
 class ChatRequest(BaseModel):
     message: str
-    provider: str = "groq"           # "groq" or "openai"
-    religion: Optional[str] = None   # for Spiritual Coach
-    session_id: Optional[str] = None # for Supabase history
+    provider: str = "groq"
+    religion: Optional[str] = None
+    session_id: Optional[str] = None
     conversation_history: Optional[List[dict]] = []
 
 
@@ -63,9 +79,6 @@ class ChatResponse(BaseModel):
 # LLM WRAPPER
 # ─────────────────────────────────────────────
 async def call_llm(provider: str, system_prompt: str, user_message: str, temperature: float = 0.7) -> str:
-    """
-    Unified LLM wrapper. Supports Groq and OpenAI.
-    """
     if provider == "groq":
         return await call_groq(system_prompt, user_message, temperature)
     elif provider == "openai":
@@ -75,14 +88,15 @@ async def call_llm(provider: str, system_prompt: str, user_message: str, tempera
 
 
 async def call_groq(system_prompt: str, user_message: str, temperature: float) -> str:
-    if not GROQ_API_KEY:
-        raise HTTPException(status_code=500, detail="GROQ_API_KEY not set")
+    key = os.getenv("GROQ_API_KEY", "")
+    if not key:
+        raise HTTPException(status_code=500, detail="GROQ_API_KEY not set in environment")
 
     async with httpx.AsyncClient(timeout=30) as client:
         response = await client.post(
             "https://api.groq.com/openai/v1/chat/completions",
             headers={
-                "Authorization": f"Bearer {GROQ_API_KEY}",
+                "Authorization": f"Bearer {key}",
                 "Content-Type": "application/json",
             },
             json={
@@ -101,14 +115,15 @@ async def call_groq(system_prompt: str, user_message: str, temperature: float) -
 
 
 async def call_openai(system_prompt: str, user_message: str, temperature: float) -> str:
-    if not OPENAI_API_KEY:
-        raise HTTPException(status_code=500, detail="OPENAI_API_KEY not set")
+    key = os.getenv("OPENAI_API_KEY", "")
+    if not key:
+        raise HTTPException(status_code=500, detail="OPENAI_API_KEY not set in environment")
 
     async with httpx.AsyncClient(timeout=30) as client:
         response = await client.post(
             "https://api.openai.com/v1/chat/completions",
             headers={
-                "Authorization": f"Bearer {OPENAI_API_KEY}",
+                "Authorization": f"Bearer {key}",
                 "Content-Type": "application/json",
             },
             json={
@@ -222,9 +237,9 @@ Question: <your single question here>"""
 # SUPABASE HELPERS
 # ─────────────────────────────────────────────
 async def save_to_supabase(session_id: str, user_message: str, agent_responses: list, summary: str, question: str):
-    if not supabase:
+    db = get_supabase()
+    if not db:
         return
-
     try:
         record = {
             "session_id": session_id,
@@ -234,18 +249,18 @@ async def save_to_supabase(session_id: str, user_message: str, agent_responses:
             "question": question,
             "created_at": datetime.utcnow().isoformat(),
         }
-        supabase.table("chat_history").insert(record).execute()
+        db.table("chat_history").insert(record).execute()
     except Exception as e:
-        print(f"Supabase save error: {e}")
+        print(f"Supabase save error (non-fatal): {e}")
 
 
 async def get_session_history(session_id: str) -> list:
-    if not supabase or not session_id:
+    db = get_supabase()
+    if not db or not session_id:
         return []
-
     try:
         result = (
-            supabase.table("chat_history")
+            db.table("chat_history")
             .select("*")
             .eq("session_id", session_id)
             .order("created_at", desc=False)
@@ -254,7 +269,7 @@ async def get_session_history(session_id: str) -> list:
         )
         return result.data or []
     except Exception as e:
-        print(f"Supabase fetch error: {e}")
+        print(f"Supabase fetch error (non-fatal): {e}")
         return []
 
 
@@ -270,7 +285,7 @@ async def chat(request: ChatRequest):
     history_context = ""
     if request.conversation_history:
         history_lines = []
-        for turn in request.conversation_history[-6:]:  # last 3 exchanges
+        for turn in request.conversation_history[-6:]:
             history_lines.append(f"User: {turn.get('user', '')}")
             if turn.get("question"):
                 history_lines.append(f"Team Question: {turn.get('question', '')}")
@@ -325,7 +340,7 @@ async def chat(request: ChatRequest):
     if not question:
         question = "Can you share more details so the team can help you better?"
 
-    # ── Step 3: Save to Supabase ──
+    # ── Step 3: Save to Supabase (non-blocking) ──
     agent_data = [r.dict() for r in agent_results]
     asyncio.create_task(
         save_to_supabase(session_id, request.message, agent_data, summary, question)
@@ -347,9 +362,10 @@ async def get_history(session_id: str):
 
 @app.get("/health")
 async def health():
+    db = get_supabase()
     return {
         "status": "ok",
-        "groq_configured": bool(GROQ_API_KEY),
-        "openai_configured": bool(OPENAI_API_KEY),
-        "supabase_configured": bool(supabase),
+        "groq_configured": bool(os.getenv("GROQ_API_KEY")),
+        "openai_configured": bool(os.getenv("OPENAI_API_KEY")),
+        "supabase_configured": bool(db),
     }