Upload model.py

model.py

@@ -0,0 +1,45 @@
+from typing import TypedDict
+
+class SafetyResult(TypedDict):
+    safe: bool
+    reason: str
+
+UNSAFE_KEYWORDS = [
+    "đốt",
+    "cháy",
+    "nổ",
+    "phá hủy",
+    "phá hoại",
+    "đập vỡ",
+    "làm đau",
+    "tấn công",
+    "giết",
+    "cầm dao",
+    "tắt báo cháy",
+]
+
+def check_safety(command: str) -> SafetyResult:
+    """Check whether a Vietnamese command is safe for a home robot to execute."""
+    text = command.lower()
+
+    for word in UNSAFE_KEYWORDS:
+        if word in text:
+            return {
+                "safe": False,
+                "reason": f"Command contains unsafe keyword: '{word}'",
+            }
+
+    # default: safe
+    return {
+        "safe": True,
+        "reason": "No unsafe patterns detected",
+    }
+
+if __name__ == "__main__":
+    tests = [
+        "Bật đèn phòng khách",
+        "Đốt rèm cửa đi",
+        "Mang dao vào phòng khách",
+    ]
+    for t in tests:
+        print(t, "->", check_safety(t))