Update app.py

app.py

@@ -21,9 +21,9 @@ client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 client.base_url = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
 
 # Chatwoot config
-CHATWOOT_BASE_URL = os.getenv("CHATWOOT_BASE_URL")           # e.g., https://app.chatwoot.com
-CHATWOOT_API_KEY = os.getenv("CHATWOOT_API_KEY")             # API Access Token of bot
-CHATWOOT_ACCOUNT_ID = os.getenv("CHATWOOT_ACCOUNT_ID")       # Account ID (integer)
+CHATWOOT_BASE_URL = os.getenv("CHATWOOT_BASE_URL")
+CHATWOOT_API_KEY = os.getenv("CHATWOOT_API_KEY")
+CHATWOOT_ACCOUNT_ID = int(os.getenv("CHATWOOT_ACCOUNT_ID", "123911"))  # Defaulting to 123911
 
 # Global set to track conversations where AI should stop replying
 stop_reply_conversations = set()
@@ -37,44 +37,45 @@ async def ask(request: Request):
     account_id = payload.get("account", {}).get("id")
     sender_id = payload.get("sender", {}).get("id")
     conversation_id = payload.get("conversation", {}).get("id")
-    sender_role = payload.get("sender", {}).get("role", "").lower()  # e.g. "agent", "contact", "bot"
+    sender_role = payload.get("sender", {}).get("role", "").lower()
     message_content = payload.get("content", "").strip()
 
+    # Stop if message is not from allowed account ID
+    if account_id != CHATWOOT_ACCOUNT_ID:
+        print(f"❌ Ignoring message from account ID {account_id} (only {CHATWOOT_ACCOUNT_ID} is allowed)")
+        return {"status": "ignored: invalid account"}
+
     # Only respond to incoming messages
     if message_type != "incoming":
         print("⚠️ Ignoring non-incoming messages")
         return {"status": "ignored"}
 
-    # Ignore messages from bot itself to prevent loops
+    # Ignore messages from bot itself
     if sender_id == account_id:
         print("⚠️ Ignoring message from bot itself")
         return {"status": "ignored"}
 
-    # If human agent sends a message, stop AI replies for this conversation
+    # Human agent message: disable AI for this conversation
     if sender_role == "agent":
-        # If agent sends command to resume bot replies
         if message_content.lower() == "/botresume":
-            if conversation_id in stop_reply_conversations:
-                stop_reply_conversations.remove(conversation_id)
-                print(f"ℹ️ Bot resumed for conversation {conversation_id} by human agent command.")
-                await send_chatwoot_message(conversation_id, "Bot resumed and will reply to users now.")
-            return {"status": "human takeover - bot resume command"}
-
-        # Otherwise blacklist conversation to stop AI replies
-        stop_reply_conversations.add(conversation_id)
-        print(f"⚠️ Human agent message detected. Conversation {conversation_id} blacklisted for AI replies.")
-        return {"status": "human takeover activated"}
+            stop_reply_conversations.discard(conversation_id)
+            print(f"ℹ️ Bot resumed for conversation {conversation_id}")
+            await send_chatwoot_message(conversation_id, "Bot resumed and will reply to users now.")
+            return {"status": "bot resumed"}
+        else:
+            stop_reply_conversations.add(conversation_id)
+            print(f"🚫 Human agent takeover. AI disabled for conversation {conversation_id}")
+            return {"status": "human takeover"}
 
-    # If conversation is blacklisted, ignore user messages
     if conversation_id in stop_reply_conversations:
-        print(f"⚠️ Conversation {conversation_id} is blacklisted, no AI reply sent.")
-        return {"status": "ignored due to human takeover"}
+        print(f"🚫 Conversation {conversation_id} blacklisted. No AI reply.")
+        return {"status": "ignored: blacklisted"}
 
-    if not message_content or not conversation_id or not CHATWOOT_ACCOUNT_ID:
-        print("❌ Missing message content, conversation ID, or account ID")
+    if not message_content or not conversation_id:
+        print("❌ Missing essential data")
         return {"status": "invalid payload"}
 
-    # Prepare OpenAI messages
+    # Prepare and send to OpenAI
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": message_content},
@@ -93,14 +94,12 @@ async def ask(request: Request):
         print("❌ OpenAI Error:", e)
         answer = "Sorry, I'm having trouble answering right now."
 
-    fallback_msg = "I'm not sure about that. Let me connect you with a human agent."
-    if answer == fallback_msg:
+    fallback = "I'm not sure about that. Let me connect you with a human agent."
+    if answer == fallback:
         stop_reply_conversations.add(conversation_id)
-        print(f"⚠️ Added conversation {conversation_id} to stop reply blacklist due to fallback answer.")
+        print(f"🚫 Fallback triggered. Conversation {conversation_id} blacklisted.")
 
-    # Send AI reply back to Chatwoot
     await send_chatwoot_message(conversation_id, answer)
-
     return {"status": "ok"}
 
 
@@ -114,12 +113,12 @@ async def send_chatwoot_message(conversation_id: str, content: str):
     }
     try:
         async with httpx.AsyncClient() as http:
-            chatwoot_url = f"{CHATWOOT_BASE_URL}/api/v1/accounts/{CHATWOOT_ACCOUNT_ID}/conversations/{conversation_id}/messages"
-            print("📤 Sending to Chatwoot:", chatwoot_url)
-            print("📦 Payload to Chatwoot:", json.dumps(message_payload, indent=2))
+            url = f"{CHATWOOT_BASE_URL}/api/v1/accounts/{CHATWOOT_ACCOUNT_ID}/conversations/{conversation_id}/messages"
+            print("📤 Sending to Chatwoot:", url)
+            print("📦 Payload:", json.dumps(message_payload, indent=2))
 
             resp = await http.post(
-                chatwoot_url,
+                url,
                 headers={
                     "Content-Type": "application/json",
                     "api_access_token": CHATWOOT_API_KEY,