Update app.py

app.py

@@ -33,7 +33,7 @@ viru_model = AutoModelForSeq2SeqLM.from_pretrained(
 )
 
 # =========================
-# WIKTIONARY LOOKUP
+# WIKTIONARY API
 # =========================
 
 def get_russian_info(word):
@@ -41,52 +41,81 @@ def get_russian_info(word):
     try:
 
         url = (
-            "https://ru.wiktionary.org/w/api.php"
+            "https://ru.wiktionary.org/api/rest_v1/page/definition/"
+            + word
         )
 
-        params = {
-            "action": "query",
-            "prop": "extracts",
-            "exintro": True,
-            "explaintext": True,
-            "titles": word,
-            "format": "json"
-        }
-
         response = requests.get(
             url,
-            params=params,
             timeout=10
         )
 
         data = response.json()
 
-        pages = data["query"]["pages"]
+        if "ru" not in data:
 
-        page = next(iter(pages.values()))
+            return (
+                "Không tìm thấy định nghĩa",
+                "Không có ví dụ"
+            )
 
-        extract = page.get(
-            "extract",
-            "Không tìm thấy định nghĩa"
-        )
+        entries = data["ru"]
+
+        definitions = []
+        examples = []
+
+        for entry in entries:
+
+            senses = entry.get(
+                "definitions",
+                []
+            )
+
+            for sense in senses:
+
+                definition = sense.get(
+                    "definition",
+                    ""
+                )
 
-        lines = extract.split("\n")
+                if definition:
+                    definitions.append(
+                        definition
+                    )
 
-        definition = (
-            lines[0]
-            if len(lines) > 0
-            else "Không tìm thấy"
+                exs = sense.get(
+                    "examples",
+                    []
+                )
+
+                for ex in exs:
+
+                    text = ex.get(
+                        "text",
+                        ""
+                    )
+
+                    if text:
+                        examples.append(text)
+
+        definition_text = (
+            "\n\n".join(definitions[:5])
+            if definitions
+            else "Không tìm thấy định nghĩa"
         )
 
-        example = (
-            lines[1]
-            if len(lines) > 1
+        example_text = (
+            "\n\n".join(examples[:5])
+            if examples
             else "Không có ví dụ"
         )
 
-        return definition, example
+        return (
+            definition_text,
+            example_text
+        )
 
-    except:
+    except Exception:
 
         return (
             "Không tìm thấy định nghĩa",