Update app/noi_tu.py

app/noi_tu.py

@@ -1,30 +1,49 @@
-# Thêm vào đầu file main.py (sau các import hiện có)
 import httpx
-from cachetools import TTLCache
-from fastapi import Querẻy
+import random
+from typing import Optional
 
-word_cache = TTLCache(maxsize=1, ttl=86400)
+class NoiTuApp:
+    def __init__(self):
+        self.url = "https://github.com/NNBnh/noi-tu/raw/refs/heads/main/words/words.txt"
+        self.words = set()
 
-async def load_noi_tu_words():
-    if "words" in word_cache:
-        return word_cache["words"]
-    
-    url = "https://github.com/NNBnh/noi-tu/raw/refs/heads/main/words/words.txt"
-    
-    try:
-        async with httpx.AsyncClient() as client:
-            resp = await client.get(url, timeout=10.0)
-            resp.raise_for_status()
-            lines = resp.text.strip().splitlines()
-            # Filter chỉ giữ từ có đúng 2 âm tiết (split() == 2)
-            filtered = []
-            for line in lines:
-                tu = line.strip().lower()
-                if tu and len(tu.split()) == 2:  # chỉ 2 âm tiết
-                    filtered.append(tu)
-            word_cache["words"] = set(filtered)  # dùng set để check nhanh
-            print(f"Loaded {len(word_cache['words'])} từ ghép 2 âm tiết từ NNBnh/noi-tu")
-            return word_cache["words"]
-    except Exception as e:
-        print(f"Error loading words: {e}")
-        return set()
+    async def start(self):
+        """Tải dữ liệu từ vựng khi khởi động server"""
+        try:
+            async with httpx.AsyncClient(timeout=30.0) as client:
+                response = await client.get(self.url)
+                if response.status_code == 200:
+                    # Tách từ theo dòng, chuyển về chữ thường và strip khoảng trắng
+                    self.words = {line.strip().lower() for line in response.text.splitlines() if line.strip()}
+                    print(f"Loaded {len(self.words)} words for NoiTuApp")
+                else:
+                    print(f"Failed to load words: {response.status_code}")
+        except Exception as e:
+            print(f"Error loading words: {e}")
+
+    async def stop(self):
+        self.words.clear()
+
+    def get_next_word(self, current_word: str) -> Optional[str]:
+        """Tìm một từ bắt đầu bằng tiếng cuối của từ hiện tại"""
+        current_word = current_word.lower().strip()
+        parts = current_word.split()
+        
+        # Lấy tiếng cuối cùng (ví dụ: "học sinh" -> "sinh")
+        last_syllable = parts[-1]
+        
+        # Tìm các từ trong danh sách bắt đầu bằng last_syllable
+        # Và phải là từ có ít nhất 2 tiếng (để game tiếp tục được)
+        candidates = [
+            w for w in self.words 
+            if w.startswith(last_syllable + " ") and w != current_word
+        ]
+        
+        if not candidates:
+            return None
+            
+        return random.choice(candidates)
+
+    def is_valid_word(self, word: str) -> bool:
+        """Kiểm tra từ có trong từ điển không"""
+        return word.lower().strip() in self.words