Upload 2 files

app.py

@@ -1,22 +1,29 @@
 import os
 import time
+import logging
+import json
 from fastapi import FastAPI, Request, HTTPException, Depends, status
+from fastapi.exceptions import RequestValidationError
+from fastapi.responses import JSONResponse
 from fastapi.security import APIKeyHeader
-from typing import Dict, Any
+from typing import Dict, Any, Optional
 from datetime import datetime
 
 # LangGraph and Model Imports
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.checkpoint.base import BaseCheckpointSaver
-from agent import create_honeypot_graph, final_callback
+from agent import create_honeypot_graph
 from models import (
     HoneypotRequest, HoneypotResponse, 
     AgentState, ExtractedIntelligence, Message, EngagementMetrics
 )
 
+# --- Logging Configuration ---
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 # --- Configuration ---
 API_KEY_NAME = "x-api-key"
-# Default key for testing, should be set via environment variable in production
 API_KEY = os.environ.get("HONEYPOT_API_KEY", "sk_test_123456789")
 api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
 
@@ -24,13 +31,39 @@ api_key_header = APIKeyHeader(name=API_KEY_NAME, auto_error=False)
 app = FastAPI(
     title="Agentic Honey-Pot API",
     description="REST API for Scam Detection and Intelligence Extraction (Problem Statement 2).",
-    version="1.0.0"
+    version="1.2.0"
 )
 
 # Initialize LangGraph Checkpointer
 checkpointer: BaseCheckpointSaver = MemorySaver()
 honeypot_app = create_honeypot_graph(checkpointer)
 
+# --- Exception Handler for Diagnostic Logging ---
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request: Request, exc: RequestValidationError):
+    """
+    Captures and logs the exact payload that caused a 422 Unprocessable Entity error.
+    This is crucial for diagnosing why the tester is failing.
+    """
+    body = await request.body()
+    try:
+        payload = json.loads(body)
+    except:
+        payload = body.decode()
+    
+    logger.error(f"422 Unprocessable Entity Error!")
+    logger.error(f"Payload received: {payload}")
+    logger.error(f"Validation errors: {exc.errors()}")
+    
+    return JSONResponse(
+        status_code=status.HTTP_422_UNPROCESSABLE_ENTITY,
+        content={
+            "detail": exc.errors(),
+            "body": payload,
+            "message": "The request body does not match the Problem Statement 2 schema. Check logs for details."
+        }
+    )
+
 # --- Dependency for API Key Validation ---
 async def get_api_key(api_key_header: str = Depends(api_key_header)):
     if api_key_header is None or api_key_header != API_KEY:
@@ -59,7 +92,6 @@ async def honeypot_detection(
 
     if checkpoint and checkpoint.values:
         current_state_dict = checkpoint.values
-        # Ensure all required fields exist
         current_state_dict.setdefault("callbackSent", False)
         current_state_dict.setdefault("agentNotes", "")
         current_state_dict.setdefault("extractedIntelligence", ExtractedIntelligence())
@@ -92,7 +124,7 @@ async def honeypot_detection(
         
         engagement_duration = int(time.time() - start_time)
         
-        # Prepare response strictly matching PDF page 10/11
+        # Prepare response strictly matching PDF
         return {
             "status": "success",
             "scamDetected": final_state["scamDetected"],
@@ -105,7 +137,7 @@ async def honeypot_detection(
         }
 
     except Exception as e:
-        print(f"Error in Honey-Pot: {e}")
+        logger.error(f"Internal Error: {e}")
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=f"Internal server error: {str(e)}",

models.py

@@ -26,7 +26,7 @@ class HoneypotRequest(BaseModel):
 class ExtractedIntelligence(BaseModel):
     """Structured data to be extracted from the conversation."""
     bankAccounts: List[str] = Field(default_factory=list)
-    upilds: List[str] = Field(default_factory=list)
+    upilds: List[str] = Field(default_factory=list) # Strictly 'upilds' as per PDF
     phishingLinks: List[str] = Field(default_factory=list)
     phoneNumbers: List[str] = Field(default_factory=list)
     suspiciousKeywords: List[str] = Field(default_factory=list)