Update app.py

app.py

@@ -28,79 +28,6 @@ CHATWOOT_ACCOUNT_ID = os.getenv("CHATWOOT_ACCOUNT_ID")       # Account ID (integ
 # Add at the top after imports:
 stop_reply_conversations = set()
 
-# @app.post("/ask")
-# async def ask(request: Request):
-#     payload = await request.json()
-#     print("📥 Incoming payload:", json.dumps(payload, indent=2))
-
-#     message_type = payload.get("message_type", "").lower()
-#     account_id = payload.get("account", {}).get("id")        # bot's account id
-#     sender_id = payload.get("sender", {}).get("id")          # sender id of this message
-
-#     # Only respond to incoming messages
-#     if message_type != "incoming":
-#         print("⚠️ Ignoring non-incoming messages")
-#         return {"status": "ignored"}
-
-#     # Ignore messages sent by bot itself (to avoid loops)
-#     if sender_id == account_id:
-#         print("⚠️ Ignoring message from bot itself to prevent loop.")
-#         return {"status": "ignored"}
-
-#     # Extract user question and conversation id
-#     user_question = payload.get("content", "")
-#     conversation_id = payload.get("conversation", {}).get("id")
-#     if not user_question or not conversation_id or not CHATWOOT_ACCOUNT_ID:
-#         print("❌ Missing message, conversation ID, or account ID")
-#         return {"status": "invalid payload"}
-
-#     messages = [
-#         {"role": "system", "content": system_prompt},
-#         {"role": "user", "content": user_question},
-#     ]
-
-#     try:
-#         response = client.chat.completions.create(
-#             model="gpt-4o-mini",
-#             messages=messages,
-#             temperature=0,
-#             max_tokens=200,
-#         )
-#         answer = response.choices[0].message.content.strip()
-#         print("✅ OpenAI Answer:", answer)
-#     except Exception as e:
-#         print("❌ OpenAI Error:", e)
-#         answer = "Sorry, I'm having trouble answering right now."
-
-#     message_payload = {
-#         "content": answer,
-#         "message_type": "outgoing",
-#         "private": False,
-#         "content_type": "text",
-#         "content_attributes": {}
-#     }
-
-#     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))
-
-#             resp = await http.post(
-#                 chatwoot_url,
-#                 headers={
-#                     "Content-Type": "application/json",
-#                     "api_access_token": CHATWOOT_API_KEY,
-#                 },
-#                 json=message_payload,
-#             )
-#             print("📬 Chatwoot Response:", resp.status_code, resp.text)
-
-#     except Exception as e:
-#         print("❌ Chatwoot Send Error:", e)
-
-#     return {"status": "ok"}
-
 @app.post("/ask")
 async def ask(request: Request):
     payload = await request.json()
@@ -110,6 +37,8 @@ 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"
+    message_content = payload.get("content", "").strip()
 
     # Only respond to incoming messages
     if message_type != "incoming":
@@ -121,19 +50,35 @@ async def ask(request: Request):
         print("⚠️ Ignoring message from bot itself")
         return {"status": "ignored"}
 
-    # If conversation is blacklisted (bot should stop replying)
+    # Handle human agent takeover: if agent sends message, stop AI replies
+    if sender_role == "agent":
+        # If agent sends special command to resume bot
+        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.")
+                # Optionally send confirmation message back to chatwoot
+                await send_chatwoot_message(conversation_id, "Bot resumed and will reply to users now.")
+            return {"status": "human takeover - bot resume command"}
+
+        # Otherwise, mark conversation as blacklisted 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"}
+
+    # If conversation is blacklisted, ignore user messages (don't reply)
     if conversation_id in stop_reply_conversations:
-        print(f"⚠️ Conversation {conversation_id} is blacklisted, no reply sent.")
-        return {"status": "ignored"}
+        print(f"⚠️ Conversation {conversation_id} is blacklisted, no AI reply sent.")
+        return {"status": "ignored due to human takeover"}
 
-    user_question = payload.get("content", "")
-    if not user_question or not conversation_id or not CHATWOOT_ACCOUNT_ID:
-        print("❌ Missing message, conversation ID, or account ID")
+    if not message_content or not conversation_id or not CHATWOOT_ACCOUNT_ID:
+        print("❌ Missing message content, conversation ID, or account ID")
         return {"status": "invalid payload"}
 
+    # Prepare messages for OpenAI chat completion
     messages = [
         {"role": "system", "content": system_prompt},
-        {"role": "user", "content": user_question},
+        {"role": "user", "content": message_content},
     ]
 
     try:
@@ -149,19 +94,26 @@ async def ask(request: Request):
         print("❌ OpenAI Error:", e)
         answer = "Sorry, I'm having trouble answering right now."
 
-    # If answer is "I'm not sure about that..." message, blacklist this conversation
-    if answer == "I'm not sure about that. Let me connect you with a human agent.":
+    # If fallback answer detected, blacklist conversation to stop further AI replies
+    fallback_msg = "I'm not sure about that. Let me connect you with a human agent."
+    if answer == fallback_msg:
         stop_reply_conversations.add(conversation_id)
-        print(f"⚠️ Added conversation {conversation_id} to stop reply blacklist.")
+        print(f"⚠️ Added conversation {conversation_id} to stop reply blacklist due to fallback answer.")
+
+    # Send AI reply back to Chatwoot
+    await send_chatwoot_message(conversation_id, answer)
 
+    return {"status": "ok"}
+
+
+async def send_chatwoot_message(conversation_id: str, content: str):
     message_payload = {
-        "content": answer,
+        "content": content,
         "message_type": "outgoing",
         "private": False,
         "content_type": "text",
         "content_attributes": {}
     }
-
     try:
         async with httpx.AsyncClient() as http:
             chatwoot_url = f"{CHATWOOT_BASE_URL}/api/v1/accounts/{CHATWOOT_ACCOUNT_ID}/conversations/{conversation_id}/messages"
@@ -177,8 +129,6 @@ async def ask(request: Request):
                 json=message_payload,
             )
             print("📬 Chatwoot Response:", resp.status_code, resp.text)
-
     except Exception as e:
         print("❌ Chatwoot Send Error:", e)
-
     return {"status": "ok"}