Update app.py

app.py

@@ -1,139 +1,3 @@
-# import os
-# import json
-# import httpx
-# from fastapi import FastAPI, Request
-# from openai import OpenAI
-
-# app = FastAPI()
-
-# # Load KB
-# with open("kb.json") as f:
-#     kb = json.load(f)
-
-# # Build system prompt
-# system_prompt = "You are a helpful assistant. Only answer questions based on the following knowledge base:\n\n"
-# for q, a in kb.items():
-#     system_prompt += f"Q: {q}\nA: {a}\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.'"
-
-# # OpenAI setup
-# 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")
-# CHATWOOT_API_KEY = os.getenv("CHATWOOT_API_KEY")
-# CHATWOOT_ACCOUNT_ID = int(os.getenv("CHATWOOT_ACCOUNT_ID"))  # e.g., 123911
-
-# # Track conversations where AI should stop replying
-# stop_reply_conversations = set()
-
-# @app.post("/ask")
-# async def ask(request: Request):
-#     payload = await request.json()
-#     print("📥 Incoming payload:", json.dumps(payload, indent=2))
-
-#     account_id = payload.get("account", {}).get("id")
-#     conversation_id = str(payload.get("conversation", {}).get("id"))
-#     sender = payload.get("sender") or {}
-#     sender_id = sender.get("id")
-#     sender_role = (sender.get("role") or "").lower()
-#     message_type = payload.get("message_type", "").lower()
-#     message_content = payload.get("content", "").strip()
-
-#     print(f"🧾 sender_id: {sender_id}, sender_role: {sender_role}, account_id: {account_id}")
-
-#     # Step 1: Detect agent message via Slack and disable AI for that conversation
-#     if message_type != "incoming":
-#         messages = payload.get("conversation", {}).get("messages", [])
-#         if messages:
-#             msg = messages[0]
-#             external_ids = msg.get("external_source_ids", {})
-#             if "slack" in external_ids:
-#                 stop_reply_conversations.add(conversation_id)
-#                 print(f"🛑 Human intervened via Slack in conversation {conversation_id}. Disabling AI.")
-#                 return {"status": "AI disabled due to Slack intervention"}
-        
-#         print("⚠️ Ignoring non-incoming message")
-#         return {"status": "ignored"}
-
-#     # Bot must not reply to itself
-#     if sender_id == account_id:
-#         print("⚠️ Ignoring bot's own message")
-#         return {"status": "ignored"}
-
-#     # Handle special bot resume command
-#     if sender_role == "agent" and message_content.lower() == "#botresume":
-#         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"}
-
-#     # Check if AI is blacklisted for this conversation
-#     if conversation_id in stop_reply_conversations:
-#         print(f"🚫 AI is disabled for conversation {conversation_id}")
-#         return {"status": "ignored: human takeover"}
-
-#     # Ensure all data is present
-#     if not message_content or not conversation_id:
-#         print("❌ Missing content or conversation ID")
-#         return {"status": "invalid payload"}
-
-#     # Build messages for GPT
-#     messages = [
-#         {"role": "system", "content": system_prompt},
-#         {"role": "user", "content": message_content},
-#     ]
-
-#     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("✅ GPT Answer:", answer)
-#     except Exception as e:
-#         print("❌ OpenAI Error:", e)
-#         answer = "Sorry, I'm having trouble answering right now."
-
-#     if answer == "I'm not sure about that. Let me connect you with a human agent.":
-#         stop_reply_conversations.add(conversation_id)
-#         print(f"🚫 Fallback answer, disabling AI for conversation {conversation_id}")
-
-#     await send_chatwoot_message(conversation_id, answer)
-
-#     return {"status": "ok"}
-
-
-# async def send_chatwoot_message(conversation_id: str, content: str):
-#     message_payload = {
-#         "content": content,
-#         "message_type": "outgoing",
-#         "private": False,
-#         "content_type": "text",
-#         "content_attributes": {}
-#     }
-
-#     try:
-#         async with httpx.AsyncClient() as http:
-#             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(
-#                 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)
-
 import os
 import json
 import httpx
@@ -142,7 +6,7 @@ from openai import OpenAI
 
 app = FastAPI()
 
-# Load knowledge base
+# Load KB
 with open("kb.json") as f:
     kb = json.load(f)
 
@@ -159,10 +23,10 @@ client.base_url = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1")
 # Chatwoot config
 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"))
+CHATWOOT_ACCOUNT_ID = int(os.getenv("CHATWOOT_ACCOUNT_ID"))  # e.g., 123911
 
-# Track temporarily blocked sessions (session_key = sender_id + last_seen_at)
-stop_reply_sessions = set()
+# Track conversations where AI should stop replying
+stop_reply_conversations = set()
 
 @app.post("/ask")
 async def ask(request: Request):
@@ -170,56 +34,52 @@ async def ask(request: Request):
     print("📥 Incoming payload:", json.dumps(payload, indent=2))
 
     account_id = payload.get("account", {}).get("id")
-    conversation = payload.get("conversation", {})
-    conversation_id = str(conversation.get("id"))
-    sender = payload.get("sender", {}) or {}
+    conversation_id = str(payload.get("conversation", {}).get("id"))
+    sender = payload.get("sender") or {}
     sender_id = sender.get("id")
     sender_role = (sender.get("role") or "").lower()
     message_type = payload.get("message_type", "").lower()
     message_content = payload.get("content", "").strip()
-    sender_last_seen = sender.get("last_seen_at") or "unknown"
-
-    # Session key: changes when user reloads/returns
-    session_key = f"{sender_id}_{sender_last_seen}"
 
-    print(f"🧾 session_key: {session_key}, sender_role: {sender_role}, account_id: {account_id}")
+    print(f"🧾 sender_id: {sender_id}, sender_role: {sender_role}, account_id: {account_id}")
 
-    # Step 1: Detect agent message via Slack and disable AI for current session
+    # Step 1: Detect agent message via Slack and disable AI for that conversation
     if message_type != "incoming":
-        messages = conversation.get("messages", [])
+        messages = payload.get("conversation", {}).get("messages", [])
         if messages:
             msg = messages[0]
             external_ids = msg.get("external_source_ids", {})
             if "slack" in external_ids:
-                stop_reply_sessions.add(session_key)
-                print(f"🛑 Human intervened via Slack. Disabling AI for session {session_key}.")
+                stop_reply_conversations.add(conversation_id)
+                print(f"🛑 Human intervened via Slack in conversation {conversation_id}. Disabling AI.")
                 return {"status": "AI disabled due to Slack intervention"}
+        
         print("⚠️ Ignoring non-incoming message")
         return {"status": "ignored"}
 
-    # Don't reply to own messages
+    # Bot must not reply to itself
     if sender_id == account_id:
         print("⚠️ Ignoring bot's own message")
         return {"status": "ignored"}
 
-    # Handle manual resume
+    # Handle special bot resume command
     if sender_role == "agent" and message_content.lower() == "#botresume":
-        stop_reply_sessions.discard(session_key)
-        print(f"ℹ️ Bot resumed for session {session_key}")
+        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"}
 
-    # Check if AI is disabled for current session
-    if session_key in stop_reply_sessions:
-        print(f"🚫 AI is disabled for session {session_key}")
+    # Check if AI is blacklisted for this conversation
+    if conversation_id in stop_reply_conversations:
+        print(f"🚫 AI is disabled for conversation {conversation_id}")
         return {"status": "ignored: human takeover"}
 
-    # Validate data
+    # Ensure all data is present
     if not message_content or not conversation_id:
         print("❌ Missing content or conversation ID")
         return {"status": "invalid payload"}
 
-    # Prepare messages for GPT
+    # Build messages for GPT
     messages = [
         {"role": "system", "content": system_prompt},
         {"role": "user", "content": message_content},
@@ -239,10 +99,11 @@ async def ask(request: Request):
         answer = "Sorry, I'm having trouble answering right now."
 
     if answer == "I'm not sure about that. Let me connect you with a human agent.":
-        stop_reply_sessions.add(session_key)
-        print(f"🚫 Fallback answer, disabling AI for session {session_key}")
+        stop_reply_conversations.add(conversation_id)
+        print(f"🚫 Fallback answer, disabling AI for conversation {conversation_id}")
 
     await send_chatwoot_message(conversation_id, answer)
+
     return {"status": "ok"}
 
 
@@ -260,6 +121,7 @@ async def send_chatwoot_message(conversation_id: str, content: str):
             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(
                 url,
                 headers={