Fix adding brain to whatsapp

whatsapp/webhook.py

@@ -65,9 +65,6 @@ async def send_whatsapp_message(to: str, text: str):
     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
@@ -124,17 +121,40 @@ def check_and_increment_quota() -> bool:
         
     return True
 
+# ── AI Brain Connection ──────────────────────────────────────────────────────
 async def process_and_reply(from_number: str, query: str, language: str):
     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
 
-    answer = f"Echoing your RAG query: {query}" 
-    sources = ["Source 1", "Source 2"]
+    print(f"🧠 Querying GovBridge AI: '{query}'")
+    
+    try:
+        # Internal API call to your existing RAG pipeline
+        async with httpx.AsyncClient(timeout=60.0) as client:
+            api_url = "http://127.0.0.1:7860/api/rag/query"
+            payload = {"question": query, "language": language}
+            
+            resp = await client.post(api_url, json=payload)
+            
+            if resp.status_code == 200:
+                answer = resp.text
+                # Extract sources from the headers you built in api.py
+                sources_raw = resp.headers.get("X-Sources", "")
+                sources = [s for s in sources_raw.split("|") if s.strip()]
+            else:
+                answer = "My AI brain is temporarily offline for upgrades. Please try again soon!"
+                sources = []
+                
+    except Exception as e:
+        print(f"❌ Internal AI Connection Error: {e}")
+        answer = "I am having a little trouble thinking right now. Please try again in a minute!"
+        sources = []
 
     sources_text = ""
     if sources:
-        sources_text = "\n\n📄 *Sources:*\n" + "\n".join([f"• {s}" for s in sources[:3]])
+        unique_sources = list(dict.fromkeys(sources))
+        sources_text = "\n\n📄 *Sources:*\n" + "\n".join([f"• {s}" for s in unique_sources[:3]])
         
     reply = f"🏛️ *GovBridge India*\n\n{answer}{sources_text}"
     await send_whatsapp_message(from_number, reply)