Update main.py

main.py

@@ -255,13 +255,12 @@ def new_verify_otp(
     }
     collection.update_one({"user_id": user_id}, {"$set": update_doc}, upsert=True)
 
-def get_verify_otp(user_id):
+
+ def get_stored_otp(user_id: int):
     user = collection.find_one({"user_id": user_id})
-    if user:
-        otp_code = user.get("otp_code")
-        return otp_code
-    else:
-        return None
+    if user and "otp_code" in user:
+        return user["otp_code"]
+    return None       
 
 def generate_otp(length=6):
     """Generate a random OTP"""
@@ -275,7 +274,7 @@ class OTPCallbackPayload(BaseModel):
 @app.get("/api/get-verify", response_model=SuccessResponse, responses={422: {"model": SuccessResponse}})
 def get_verify_otp(user_id: int):
     try:
-        get_otp_code = get_verify_otp(user_id)
+        get_otp_code = get_stored_otp(user_id)
         if get_otp_code:
             return SuccessResponse(
                 status="True",