Fix ghost variables and timeout

whatsapp/webhook.py

@@ -3,28 +3,23 @@ import hashlib
 import os
 import httpx
 import datetime
+import asyncio
 from fastapi import APIRouter, Request, HTTPException, BackgroundTasks
 from fastapi.responses import PlainTextResponse
 from supabase import create_client
 
 router = APIRouter()
 
-VERIFY_TOKEN = os.environ.get("WHATSAPP_VERIFY_TOKEN")
-WA_TOKEN = os.environ.get("WHATSAPP_TOKEN")
-PHONE_NUMBER_ID = os.environ.get("WHATSAPP_PHONE_NUMBER_ID")
-APP_SECRET = os.environ.get("WHATSAPP_APP_SECRET")  # Fixed signature secret
-
-SUPABASE_URL = os.environ.get("SUPABASE_URL")
-SUPABASE_KEY = os.environ.get("SUPABASE_KEY")
-supabase = create_client(SUPABASE_URL, SUPABASE_KEY)
-
-WA_API_URL = f"https://graph.facebook.com/v19.0/{PHONE_NUMBER_ID}/messages"
+# ── Dynamic Secret Loader ────────────────────────────────────────────────────
+def get_secret(key: str) -> str:
+    """Dynamically fetches secrets to prevent Hugging Face soft-reboot caching."""
+    return os.environ.get(key, "").strip()
 
 # ── Webhook Verification (GET) ────────────────────────────────────────────────
 @router.get("/webhook/whatsapp")
 async def verify_webhook(request: Request):
     params = request.query_params
-    if params.get("hub.mode") == "subscribe" and params.get("hub.verify_token") == VERIFY_TOKEN:
+    if params.get("hub.mode") == "subscribe" and params.get("hub.verify_token") == get_secret("WHATSAPP_VERIFY_TOKEN"):
         print("✅ WhatsApp webhook verified")
         return PlainTextResponse(content=params.get("hub.challenge"))
     raise HTTPException(status_code=403, detail="Verification failed")
@@ -32,11 +27,11 @@ async def verify_webhook(request: Request):
 # ── Incoming Message Handler (POST) ──────────────────────────────────────────
 @router.post("/webhook/whatsapp")
 async def receive_message(request: Request, background_tasks: BackgroundTasks):
-    # Verify HMAC signature using APP_SECRET
     signature = request.headers.get("X-Hub-Signature-256", "")
     body = await request.body()
+    app_secret = get_secret("WHATSAPP_APP_SECRET")
     
-    expected = "sha256=" + hmac.new(APP_SECRET.encode(), body, hashlib.sha256).hexdigest()
+    expected = "sha256=" + hmac.new(app_secret.encode(), body, hashlib.sha256).hexdigest()
     if not hmac.compare_digest(signature, expected):
         raise HTTPException(status_code=403, detail="Invalid signature")
 
@@ -44,7 +39,7 @@ async def receive_message(request: Request, background_tasks: BackgroundTasks):
     try:
         entry = data["entry"][0]["changes"][0]["value"]
         if "messages" not in entry:
-            return {"status": "ok"} # Ignore status updates (read receipts, etc.)
+            return {"status": "ok"} 
             
         message = entry["messages"][0]
         from_number = message["from"]
@@ -62,27 +57,61 @@ async def receive_message(request: Request, background_tasks: BackgroundTasks):
         
     return {"status": "ok"}
 
+# ── Outgoing Message Pipeline ────────────────────────────────────────────────
 async def send_whatsapp_message(to: str, text: str):
-    """Send a text reply via WhatsApp Cloud API."""
     if len(text) > 4096:
         text = text[:4000] + "\n\n_[Reply truncated. Visit govbridge.in for full answer]_"
         
+    clean_phone_id = get_secret("WHATSAPP_PHONE_NUMBER_ID")
+    clean_token = get_secret("WHATSAPP_TOKEN")
+    
+    print(f"\n🚀 Initiating reply to {to}...")
+    print(f"🔑 Keys Loaded -> Phone ID: '{clean_phone_id}' | Token Length: {len(clean_token)}")
+    
+    if not clean_phone_id or not clean_token:
+        print("🚨 FATAL ERROR: API Keys are completely empty. Hugging Face failed to load secrets!")
+        return
+        
+    api_url = f"https://graph.facebook.com/v25.0/{clean_phone_id}/messages"
     payload = {
         "messaging_product": "whatsapp",
         "to": to,
         "type": "text",
         "text": {"body": text}
     }
-    async with httpx.AsyncClient() as client:
-        await client.post(
-            WA_API_URL, json=payload, headers={"Authorization": f"Bearer {WA_TOKEN}"}
-        )
+    headers = {
+        "Authorization": f"Bearer {clean_token}", 
+        "Content-Type": "application/json"
+    }
+    
+    try:
+        # Primary Attempt: Fast Asynchronous Request
+        async with httpx.AsyncClient(timeout=15.0) as client:
+            resp = await client.post(api_url, json=payload, headers=headers)
+            print(f"📤 Meta Send Response (ASYNC): {resp.status_code} - {resp.text}")
+    except httpx.ConnectTimeout:
+        print("⚠️ ConnectTimeout triggered! Falling back to synchronous network request...")
+        # Backup Attempt: Bypasses async IPv6 network glitches 
+        try:
+            resp = await asyncio.to_thread(
+                lambda: httpx.Client().post(api_url, json=payload, headers=headers, timeout=15.0)
+            )
+            print(f"📤 Meta Send Response (SYNC FALLBACK): {resp.status_code} - {resp.text}")
+        except Exception as fallback_e:
+            print(f"❌ Fallback completely failed: {fallback_e}")
+    except Exception as e:
+        print(f"❌ Unknown send error: {e}")
 
 def check_and_increment_quota() -> bool:
-    """Returns True if within quota, False if limit reached."""
     current_month = datetime.datetime.utcnow().strftime("%Y-%m")
+    supabase_url = get_secret("SUPABASE_URL")
+    supabase_key = get_secret("SUPABASE_KEY")
     
-    # Supabase atomic upsert via RPC or standard select/update
+    if not supabase_url or not supabase_key:
+        print("⚠️ Supabase keys missing, skipping quota check.")
+        return True
+        
+    supabase = create_client(supabase_url, supabase_key)
     res = supabase.table("whatsapp_quota").select("conversation_count").eq("month", current_month).execute()
     
     if res.data:
@@ -96,16 +125,10 @@ def check_and_increment_quota() -> bool:
     return True
 
 async def process_and_reply(from_number: str, query: str, language: str):
-    """Full RAG pipeline for WhatsApp."""
     if not check_and_increment_quota():
         await send_whatsapp_message(from_number, "⚠️ Our WhatsApp service has reached its monthly limit. Please visit govbridge.in for full access.")
         return
 
-    # TODO: Import your actual run_rag_pipeline function here once refactored
-    # from api import run_rag_pipeline  
-    # answer, sources = await run_rag_pipeline(query, language)
-    
-    # Mocking response until Step 4 refactor is complete
     answer = f"Echoing your RAG query: {query}" 
     sources = ["Source 1", "Source 2"]
 
@@ -117,20 +140,25 @@ async def process_and_reply(from_number: str, query: str, language: str):
     await send_whatsapp_message(from_number, reply)
 
 async def process_voice_and_reply(from_number: str, audio_id: str):
-    """Download voice message and transcribe via Whisper."""
+    clean_token = get_secret("WHATSAPP_TOKEN")
+    groq_key = get_secret("GROQ_API_KEY")
+    
     async with httpx.AsyncClient() as client:
         meta_resp = await client.get(
-            f"https://graph.facebook.com/v19.0/{audio_id}",
-            headers={"Authorization": f"Bearer {WA_TOKEN}"}
+            f"https://graph.facebook.com/v25.0/{audio_id}",
+            headers={"Authorization": f"Bearer {clean_token}"}
         )
+        if meta_resp.status_code != 200:
+            print(f"❌ Failed to get audio metadata: {meta_resp.text}")
+            return
+            
         audio_url = meta_resp.json()["url"]
-        
-        audio_resp = await client.get(audio_url, headers={"Authorization": f"Bearer {WA_TOKEN}"})
+        audio_resp = await client.get(audio_url, headers={"Authorization": f"Bearer {clean_token}"})
         audio_bytes = audio_resp.content
 
         transcription_resp = await client.post(
             "https://api.groq.com/openai/v1/audio/transcriptions",
-            headers={"Authorization": f"Bearer {os.environ['GROQ_API_KEY']}"},
+            headers={"Authorization": f"Bearer {groq_key}"},
             files={"file": ("audio.ogg", audio_bytes, "audio/ogg")},
             data={"model": "whisper-large-v3-turbo", "language": "hi"}
         )
@@ -139,4 +167,4 @@ async def process_voice_and_reply(from_number: str, audio_id: str):
     if transcript:
         await process_and_reply(from_number, transcript, "hindi")
     else:
-        await send_whatsapp_message(from_number, "🙏 Sorry, I couldn't understand the voice message. Please type your question.")
+        await send_whatsapp_message(from_number, "🙏 Sorry, I couldn't understand the voice message. Please type your question.")