Add debug regex

app.py

@@ -1,3 +1,5 @@
+from time import sleep
+
 import httpx
 from fastapi import FastAPI
 from fastapi.responses import JSONResponse, FileResponse
@@ -74,6 +76,30 @@ def predict(model: InputModel) -> OutputModel:
 
     print(f"[{sender}] {text}")
 
+    # Debug sleep
+    pattern = r"^Sent from your Twilio trial account - sleep (\d+)$"
+    match = re.search(pattern, text)
+
+    if match:
+        number_str = match.group(1)
+        sleep_duration = int(number_str)
+        sleep(sleep_duration)
+        return OutputModel(action=ActionModel.JUNK, sub_action=SubActionModel.NONE)
+
+    # Debug category
+    pattern = r"^Sent from your Twilio trial account - (junk|transaction|promotion)$"
+    match = re.search(pattern, text)
+
+    if match:
+        category_str = match.group(1)
+        match category_str:
+            case 'junk':
+                return OutputModel(action=ActionModel.JUNK, sub_action=SubActionModel.NONE)
+            case 'transaction':
+                return OutputModel(action=ActionModel.TRANSACTION, sub_action=SubActionModel.NONE)
+            case 'promotion':
+                return OutputModel(action=ActionModel.PROMOTION, sub_action=SubActionModel.NONE)
+
     result = pipe(text)
     label = result[0]['label']
     score = result[0]['score']