Update app.py

app.py

@@ -1,63 +1,97 @@
 import os
 import json
-from fastapi import FastAPI, Request, Header, HTTPException
+import httpx
+from fastapi import FastAPI, Request
+
 from openai import OpenAI
 
 app = FastAPI()
 
-# Load KB from kb.json
+# Load knowledge base
 with open("kb.json") as f:
     kb = json.load(f)
 
-# Prepare system prompt with KB FAQs
+# Prepare system prompt
 system_prompt = "You are a helpful assistant. Only answer questions based on the following knowledge base:\n\n"
 for question, answer in kb.items():
     system_prompt += f"Q: {question}\nA: {answer}\n\n"
 system_prompt += (
     "If the question is not in the knowledge base, respond with: "
-    "'I'm not sure about that. Let me connect you with a human agent.'"
+    "'I'm not sure about that. I've transferred you to a human agent.'"
 )
 
-# Initialize OpenAI client with your API key
+# OpenAI client
 client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
-# Set custom base URL if provided, else default to official endpoint
 client.base_url = os.getenv("OPENAI_API_BASE", "https://fast.typegpt.net/v1")
 
-# Optional: Secret token for webhook auth, set in environment variables
-WEBHOOK_SECRET_TOKEN = os.getenv("WEBHOOK_SECRET_TOKEN")
+# Track conversations transferred
+transferred_conversations = set()
+
+# Chatwoot configs
+CHATWOOT_BASE_URL = os.getenv("CHATWOOT_BASE_URL")  # e.g., https://app.chatwoot.com
+CHATWOOT_API_TOKEN = os.getenv("CHATWOOT_API_TOKEN")  # from profile
+CHATWOOT_AGENT_ID = os.getenv("CHATWOOT_AGENT_ID")  # numeric ID of your agent
 
 @app.post("/ask")
-async def ask(
-    request: Request,
-    authorization: str | None = Header(default=None)
-):
-    # Verify auth token if set
-    if WEBHOOK_SECRET_TOKEN:
-        if authorization != f"Bearer {WEBHOOK_SECRET_TOKEN}":
-            raise HTTPException(status_code=401, detail="Unauthorized")
-
-    data = await request.json()
-
-    # Extract message text from Chatwoot payload (support both keys)
-    user_question = data.get("message") or data.get("content") or ""
-    if not user_question:
-        return {"reply": "Please ask a question."}
+async def ask(request: Request):
+    payload = await request.json()
+
+    user_msg = payload.get("content") or payload.get("message") or ""
+    conv_id = payload.get("conversation", {}).get("id")
+    inbox_id = payload.get("conversation", {}).get("inbox_id")
+
+    if not user_msg or not conv_id:
+        return {"status": "ignored"}
+
+    if str(conv_id) in transferred_conversations:
+        return {"status": "already transferred"}
 
+    # Compose OpenAI prompt
     messages = [
         {"role": "system", "content": system_prompt},
-        {"role": "user", "content": user_question},
+        {"role": "user", "content": user_msg},
     ]
 
     try:
         response = client.chat.completions.create(
-            model="gpt-4o-mini",  # Use your desired OpenAI model here
+            model="gpt-4o-mini",
             messages=messages,
             temperature=0.0,
             max_tokens=200,
         )
         answer = response.choices[0].message.content.strip()
     except Exception as e:
-        print(f"OpenAI API error: {e}")
+        print(f"OpenAI error: {e}")
         answer = "Sorry, I'm having trouble answering right now."
 
-    return {"reply": answer}
+    # If fallback response, stop future replies
+    if "transferred you to a human agent" in answer:
+        transferred_conversations.add(str(conv_id))
+
+    # Send reply using your agent
+    await send_chatwoot_reply(
+        conversation_id=conv_id,
+        message=answer,
+        agent_id=CHATWOOT_AGENT_ID,
+        inbox_id=inbox_id
+    )
+
+    return {"status": "replied"}
+
+
+async def send_chatwoot_reply(conversation_id, message, agent_id, inbox_id):
+    url = f"{CHATWOOT_BASE_URL}/api/v1/conversations/{conversation_id}/messages"
+    headers = {
+        "Content-Type": "application/json",
+        "api_access_token": CHATWOOT_API_TOKEN
+    }
+    payload = {
+        "content": message,
+        "message_type": "outgoing",
+        "private": False,
+        "sender_type": "AgentBot",
+        "sender_id": agent_id,
+        "inbox_id": inbox_id
+    }
+    async with httpx.AsyncClient() as client:
+        await client.post(url, json=payload, headers=headers)