initial

honeypot_api.py

@@ -124,83 +124,85 @@ def generate_agent_reply(session_id):
     history = conversation_store[session_id]
     scammer_msgs = [m for m in history if m["sender"] == "scammer"]
     turn = len(scammer_msgs)
-
     last_text = scammer_msgs[-1]["text"].lower()
 
-    # Escalation Phases
+    # Track asked categories
+    if "asked_categories" not in session_meta[session_id]:
+        session_meta[session_id]["asked_categories"] = set()
+
+    asked = session_meta[session_id]["asked_categories"]
+
+    # Escalation
     if turn <= 2:
-        phase = "confused"
+        tone = "confused"
     elif turn <= 5:
-        phase = "concerned"
+        tone = "concerned"
     elif turn <= 8:
-        phase = "skeptical"
+        tone = "skeptical"
     else:
-        phase = "firm"
+        tone = "firm"
 
     emotional_map = {
-        "confused": [
-            "I am not fully understanding this.",
-            "This is confusing to me."
-        ],
-        "concerned": [
-            "I am worried about my account.",
-            "This situation feels risky."
-        ],
-        "skeptical": [
-            "Something does not feel right here.",
-            "I am starting to doubt this."
-        ],
-        "firm": [
-            "Before I proceed, I need proper proof.",
-            "I will not share anything without verification."
-        ]
+        "confused": ["I am not fully understanding this."],
+        "concerned": ["I am worried about my account."],
+        "skeptical": ["Something does not feel right here."],
+        "firm": ["I will not share anything without proper verification."]
     }
 
+    opener = random.choice(emotional_map[tone])
+
+    # Aggressive elicitation order
+    elicitation_priority = [
+        ("phoneNumbers", "Can you provide your direct official contact number?"),
+        ("emailAddresses", "What is your official company email address?"),
+        ("bankAccounts", "Please resend the full bank account number clearly."),
+        ("upiIds", "Can you resend the exact UPI ID?"),
+        ("caseIds", "What is the official case reference number?"),
+        ("policyNumbers", "What is the policy number linked to this?"),
+        ("orderNumbers", "Is there any transaction or order ID?")
+    ]
+
+    intel = intelligence_store[session_id]
+
+    question = None
+
+    for key, q in elicitation_priority:
+        if not intel.get(key) and key not in asked:
+            question = q
+            asked.add(key)
+            break
+
+    if not question:
+        fallback_questions = [
+            "Which branch are you calling from?",
+            "What is your registered company name?",
+            "Can you share the official website?",
+            "Can you provide your employee ID again?"
+        ]
+        question = random.choice(fallback_questions)
+
+    # Red flag statements
     red_flags = []
     if "otp" in last_text:
         red_flags.append("You are asking for my OTP which is extremely sensitive.")
     if "urgent" in last_text:
         red_flags.append("You are creating urgency which is suspicious.")
     if "fee" in last_text:
-        red_flags.append("Why is there a fee before resolving this?")
+        red_flags.append("Why is there a fee involved?")
     if "link" in last_text:
         red_flags.append("The link you shared looks suspicious.")
-    if "upi" in last_text:
-        red_flags.append("I am unsure about this UPI ID.")
-
-    opener = random.choice(emotional_map[phase])
-    flag_statement = random.choice(red_flags) if red_flags else ""
-
-    investigative_questions = [
-        "Can you provide your official employee ID?",
-        "What is your branch location?",
-        "Can you share your direct contact number?",
-        "Is there an official website I can verify?",
-        "What is the reference or case ID?",
-        "Please resend the full bank account details clearly.",
-        "What is the registered company name?"
-    ]
 
-    question = random.choice(investigative_questions)
-
-    structure_type = random.choice(["short", "medium", "long"])
-
-    if structure_type == "short":
-        reply = f"{opener} {question}"
-    elif structure_type == "medium":
-        reply = f"{opener} {flag_statement} {question}"
-    else:
-        reply = f"{opener} {flag_statement} If this is genuine, why is this different from standard procedure? {question}"
+    flag = random.choice(red_flags) if red_flags else ""
 
+    reply = f"{opener} {flag} {question}"
     reply = re.sub(r"\s+", " ", reply).strip()
 
     if not reply.endswith("?"):
         reply += "?"
 
-    time.sleep(random.uniform(0.3, 0.8))
+    time.sleep(random.uniform(0.3, 0.7))
 
     return reply
-
 # ======================================================
 # FINAL OUTPUT SUBMISSION
 # ======================================================