Update main.py

main.py

@@ -53,8 +53,9 @@ START_TIME = time.time()
 order_cache = TTLCache(maxsize=100, ttl=86400)
 
 class NoiTuRequest(BaseModel):
-    tu_truoc: str
-    used: Optional[List[str]] = None
+    before: str
+    answer: str
+    list: List[str]
 
 class ProductItem(BaseModel):
     name: str
@@ -144,40 +145,19 @@ def system_status():
         "total_ram_mb": round(memory.total / 1024 / 1024, 2)
     }
 
-@app.get("/api/v1/noitu/2-word")
-async def api_noitu(word: str = Query(..., description="Nhập từ 2 tiếng")):
-    word = word.lower().strip()
-    
-    # 1. Kiểm tra định dạng (phải là 2 tiếng)
-    if len(word.split()) != 2:
-        return {
-            "ok": False,
-            "error": "Vui lòng chỉ nhập từ có 2 tiếng (Ví dụ: 'ân sư')."
-        }
-
-    # 2. Kiểm tra từ có trong từ điển không
-    if not noitu_app.is_valid_word(word):
-        return {
-            "ok": False,
-            "error": f"Từ '{word}' không có trong hệ thống từ điển."
-        }
-    
-    # 3. Bot tìm từ nối
-    bot_word = noitu_app.get_next_word(word)
-    
-    if not bot_word:
-        return {
-            "ok": True,
-            "bot_word": None,
-            "message": "Bạn thắng! Bot không tìm được từ nối tiếp."
-        }
+@app.post("/api/v1/noitu/check")
+async def api_noitu_check(
+    data: NoiTuRequest, 
+    apikey: Optional[str] = Query(None), 
+    x_api_key: Optional[str] = Header(None, alias="X-API-Key")
+):
+    # Kiểm tra API KEY
+    key = x_api_key or apikey
+    if not API_KEY or key != API_KEY:
+        raise HTTPException(status_code=403, detail="API key invalid. Go touch grass")
     
-    return {
-        "ok": True,
-        "input": word,
-        "bot_word": bot_word
-    }
-
+    # Trả về kết quả check từ trọng tài
+    return noitu_app.validate_turn(data.before, data.answer, data.list)
 
 @app.get("/api/v1/invites/discord")
 async def get_new_invite():