Upload app.py

app.py

@@ -1,6 +1,8 @@
 from fastapi import FastAPI, Request, Depends, HTTPException, status
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
 import random
+from pydantic import BaseModel
+
 
 app = FastAPI()
 security = HTTPBearer()
@@ -66,10 +68,37 @@ async def book_ticket(request: Request):
         "message": f"Ticket booked for '{movie_name}'. Username: {username}, Time: {time}, Price: {price}"
     }
 
+class AgentActionContent(BaseModel):
+    email: str = None
+    from_number: str = None
+    to_number: str
+
+
+class AgentAction(BaseModel):
+    name: str
+    content: AgentActionContent
+
+class Payload(BaseModel):
+    agent_action_id: str
+    agent_action: AgentAction
+
 @app.get('/generate_otp')
-async def generate_otp(token: str = Depends(verify_token)):
+async def generate_otp(payload: Payload, token: str = Depends(verify_token)):
     otp = random.randint(10000, 99999)
-    return {
-        "status": "success",
-        "otp": otp
-    }
+
+    if payload.agent_action.content.from_number is None:
+        print(f"[AMSService][generate_otp] Generating OTP for email: {payload.agent_action.content.email}")
+        result = {
+            "status": "success",
+            "verification_code": otp,
+            "message": f"OTP generated for email: {payload.agent_action.content.email}"
+        }
+    else:
+        print(f"[AMSService][generate_otp] Generating OTP for phone number: {payload.agent_action.content.from_number}")
+        result = {
+            "status": "success",
+            "verification_code": otp,
+            "message": f"OTP generated for phone number: {payload.agent_action.content.from_number}"
+        }
+
+    return result