Update main.py

main.py

@@ -50,6 +50,10 @@ API_KEY = os.getenv("API_KEY")
 START_TIME = time.time()
 order_cache = TTLCache(maxsize=100, ttl=86400)
 
+class NoiTuRequest(BaseModel):
+    tu_truoc: str
+    used: Optional[List[str]] = None
+
 class ProductItem(BaseModel):
     name: str
     price: str
@@ -138,6 +142,45 @@ def system_status():
         "total_ram_mb": round(memory.total / 1024 / 1024, 2)
     }
 
+@app.post("/api/v1/noitu/2-word")
+async def noi_tu_2_am_tiet(request: NoiTuRequest):
+    tudien = await load_noi_tu_words()
+    if not tudien:
+        raise HTTPException(status_code=503, detail="Không load được từ điển, thử lại sau")
+    
+    tu_truoc = request.tu_truoc.strip().lower()
+    if len(tu_truoc.split()) != 2:
+        raise HTTPException(status_code=400, detail="Từ trước phải đúng 2 âm tiết (cách nhau bằng space)")
+    
+    # Lấy âm tiết cuối
+    am_cuoi = tu_truoc.split()[-1]
+    
+    used_set = set(t.lower() for t in (request.used or []))
+    
+    candidates = []
+    for tu in tudien:
+        if tu not in used_set:
+            am_dau = tu.split()[0]
+            if am_dau == am_cuoi:
+                candidates.append(tu)
+    
+    if not candidates:
+        return {
+            "ok": False,
+            "message": "Hết từ hợp lệ rồi! Bạn thắng hoặc... hết vốn từ :("
+        }
+    
+    # Chọn random 1 từ (hoặc trả hết nếu muốn client random)
+    import random
+    tu_moi = random.choice(candidates)
+    
+    return {
+        "ok": True,
+        "tu_moi": tu_moi,
+        "am_tiet_noi": am_cuoi,
+        "so_tu_con_lai_uoc_tinh": len(candidates)  # bonus info
+    }
+
 @app.get("/api/v1/invites/discord")
 async def get_new_invite():
     res = await invite_app.get_invite()